通过遗忘无用信息,提升大模型微调效果
Forgetting: A New Mechanism Towards Better Large Language Model Fine-tuning
- 将训练数据分为有用和无用词元,无用部分主动遗忘
- 在多个基准上验证,模型性能显著优于基线
- 适合需要高质量微调的场景,如专业领域应用
监督微调(SFT)对预训练大语言模型至关重要,能有效提升其领域知识获取能力,同时保持或增强通用能力。然而,SFT的效果高度依赖数据质量与数量,否则可能仅带来有限收益,甚至性能劣化。为此,我们提出将语料库中的词元分为正向(有益)与负向(无益)两类:正向词元按常规方式训练,负向词元则需显式遗忘——这些词元可能语义不全或具有误导性。该分类机制使模型学会忽略低信息量内容,遗忘过程则引导模型更精准地学习关键信息。我们在多种主流模型架构和广泛基准上开展实验,结果表明该遗忘机制能有效提升模型性能。
原文摘要 · Abstract (English)
Supervised fine-tuning (SFT) plays a critical role for pretrained large language models (LLMs), notably enhancing their capacity to acquire domain-specific knowledge while preserving or potentially augmenting their general-purpose capabilities. However, the efficacy of SFT hinges on data quality as well as data volume, otherwise it may result in limited performance gains or even degradation relative to the associated baselines. To mitigate such reliance, we suggest categorizing tokens within each corpus into two parts -- positive and negative tokens -- based on whether they are useful to improve model performance. Positive tokens can be trained in common ways, whereas negative tokens, which may lack essential semantics or be misleading, should be explicitly forgotten. Overall, the token categorization facilitates the model to learn less informative messages, and the forgetting guides the model on what information to learn more precisely. We conduct experiments across diverse and well-established benchmarks using various model architectures, demonstrating that this forgetting mechanism enhances model performance.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。