arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2605.20005cs.LG
收藏

通过损失自适应学习率实现无遗忘的微调

Fine-Tuning Without Forgetting via Loss-Adaptive Learning Rates

University of California San Diego(加州大学圣迭戈分校)

Parjanya Prajakta Prashant, Jiongli Zhu, Aldan Creo, Babak Salimi

AI总结:

本文提出了一种损失自适应学习率调度方法FINCH,通过动态调整学习率来减少微调过程中的遗忘现象,同时保持任务性能,从而在知识获取、科学和低资源语言适应等基准测试中显著提升了模型表现。

摘要:

在新数据上微调大型语言模型可以提高任务性能,但会损害预训练期间学到的能力,这种现象称为灾难性遗忘。现有方法通过修改微调目标来抑制高损失的token或序列,但这些token对于学习新任务尤其重要,尤其是那些预训练覆盖不足的任务。在这种情况下,硬token仍应有助于学习,因此必须在不抑制它们的情况下控制遗忘。我们发现了一个简单的机制:每步的遗忘受学习率和当前训练损失平方根的乘积限制。这表明高损失批次尤其容易引发遗忘。受此启发,我们引入了FINCH,一种损失自适应的学习率调度方法,它在高损失批次上降低学习率,在模型收敛时增加学习率,同时保持微调目标不变。在知识获取、科学和低资源语言适应基准测试中,FINCH平均减少了93%的遗忘,同时保持与标准微调相当的任务性能。在Qwen3-4B知识获取任务中,FINCH将TruthfulQA的退化减少了5倍,并逆转了HaluEval的退化,同时更好地保持了置信度校准。总体而言,我们的结果表明,学习率调度是微调过程中塑造模型行为的有效工具,而不仅仅是为了目标任务优化。

英文摘要:

Fine-tuning large language models on new data improves task performance but degrades capabilities learned during pretraining, a phenomenon known as catastrophic forgetting. Existing methods mitigate this by modifying the fine-tuning objective to suppress high-loss tokens or sequences, but these tokens are essential for learning new tasks, especially those with poor pretraining coverage. In such settings, hard tokens should still contribute to learning, so forgetting must be controlled without suppressing them. We identify a simple mechanism for doing so: per-step forgetting is bounded by the product of the learning rate and the square root of the current training loss. This suggests that high-loss batches are especially prone to inducing forgetting. Motivated by this observation, we introduce FINCH, a loss-adaptive learning-rate schedule that reduces the learning rate on high-loss batches and increases it as the model converges, while leaving the fine-tuning objective unchanged. Across knowledge acquisition, science, and low-resource language adaptation benchmarks, FINCH reduces forgetting by 93% on average while matching the task performance of standard fine-tuning. On Qwen3-4B knowledge acquisition, FINCH cuts TruthfulQA degradation by 5x and reverses HaluEval degradation, while better preserving confidence calibration. Overall, our results show that learning-rate schedules are an effective tool to shape model behavior during fine-tuning, beyond just target-task optimization.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑