arXiv:2605.06654cs.LGcs.AI2026-05被引 4

用和预训练相同的优化器微调,能更好记住旧知识且性能不降。

Optimizer-Model Consistency: Full Finetuning with the Same Optimizer as Pretraining Forgets Less

  • 微调时保持与预训练一致的优化器,可减少知识遗忘。
  • 相同优化器使权重更新更契合模型当前状态,降低遗忘率。
  • 适合关注知识保留的场景,如推理类任务微调。

在训练大语言模型时,优化器在预训练和微调阶段均起关键作用。本文观察到:在监督微调(SFT)阶段,使用与预训练相同优化器进行全参数微调,相比其他优化器甚至LoRA,能在保持或提升新任务性能的同时,实现更优的学习-遗忘权衡。我们称此现象为优化器-模型一致性。通过受控实验与理论分析发现:1)优化器通过对激活值施加正则化,影响预训练检查点附近的损失曲面结构;2)为降低预训练知识遗忘,微调中的权重更新需遵循特定结构,而使用相同优化器可自然获得该结构。进一步对比全程使用Muon与AdamW的情况发现,前者在推理任务微调中表现较差。合成语言建模实验表明,这可能源于Muon对机械记忆的强倾向,在小数据量的SFT中不利于模式学习。

原文摘要 · Abstract (English)

Optimizers play an important role in both pretraining and finetuning stages when training large language models (LLMs). In this paper, we present an observation that full finetuning with the same optimizer as in pretraining achieves a better learning-forgetting tradeoff, i.e., forgetting less while achieving the same or better performance on the new task, than other optimizers and, possibly surprisingly, LoRA, during the supervised finetuning (SFT) stage. We term this phenomenon optimizer-model consistency. To better understand it, through controlled experiments and theoretical analysis, we show that: 1) optimizers can shape the models by having regularization effects on the activations, leading to different landscapes around the pretrained checkpoints; 2) in response to this regularization effect, the weight update in SFT should follow some specific structures to lower forgetting of the knowledge learned in pretraining, which can be obtained by using the same optimizer. Moreover, we specifically compare Muon and AdamW when they are employed throughout the pretraining and SFT stages and find that Muon performs worse when finetuned for reasoning tasks. With a synthetic language modeling experiment, we demonstrate that this can come from Muon's strong tendency towards rote memorization, which may hurt pattern acquisition with a small amount of data, as for SFT.

微调优化器知识保留大模型

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。