Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

HFT: Half Fine-Tuning for Large Language Models

About

Large language models (LLMs) with one or more fine-tuning phases have become a necessary step to unlock various capabilities, enabling LLMs to follow natural language instructions or align with human preferences. However, it carries the risk of catastrophic forgetting during sequential training, the parametric knowledge or the ability learned in previous stages may be overwhelmed by incoming training data. In this paper, we find that by regularly resetting partial parameters, LLMs can restore some of the original knowledge. Inspired by this, we introduce Half Fine-Tuning (HFT) for LLMs, as a substitute for full fine-tuning (FFT), to mitigate the forgetting issues, where half of the parameters are selected to learn new tasks while the other half are frozen to remain previous knowledge. We provide a feasibility analysis from the perspective of optimization and interpret the parameter selection operation as a regularization term. Without changing the model architecture, HFT could be seamlessly integrated into existing fine-tuning frameworks. Extensive experiments and analysis on supervised fine-tuning, direct preference optimization, and continual learning consistently demonstrate the effectiveness, robustness, and efficiency of HFT. Compared with FFT, HFT not only significantly alleviates the forgetting problem, but also achieves the best performance in a series of downstream benchmarks, with an approximately 30% reduction in training time.

Tingfeng Hui, Zhenyu Zhang, Shuohuan Wang, Weiran Xu, Yu Sun, Hua Wu• 2024

Related benchmarks

TaskDatasetResultRank
Instruction FollowingIFEval--
292
General ReasoningMMLU
MMLU Accuracy65.8
126
ChatAlpacaEval 2.0 (test)--
46
ChatMT-Bench
MT-Bench Score3.92
30
SafetyT3
T3 Score82.6
21
Machine TranslationFLORES-200 Target language
MT Score33.7
16
General ReasoningGlobal MMLU
MMLU35.5
16
Machine Reading ComprehensionBELEBELE Target Language
MRC Score47.6
16
Machine TranslationFLORES-200 Source language en
MT Score45.2
16
Machine Reading ComprehensionBelebele Source language en
MRC Score89.4
16
Showing 10 of 12 rows

Other info

Follow for update