Paper Overview
Field: Machine Learning Authors: Yuxing Liu, Jianyu Wang, Tong Zhang Published: 2026-05-07 arXiv: 2605.06654
Abstract (Translated)
Optimizers play an important role in both the pretraining and finetuning stages of large language model (LLM) training. In this paper, we make an observation: during the supervised finetuning (SFT) stage, full finetuning with the same optimizer used in pretraining achieves a better learning-forgetting tradeoff—forgetting less while attaining equal or better performance on new tasks—compared to other optimizers, and sometimes surprisingly outperforms LoRA. We call this phenomenon optimizer-model consistency.
To better understand it, through controlled experiments and theoretical analysis, we demonstrate:
1. Optimizers can shape models through a regularization effect on activations, leading to different landscapes around the pretraining checkpoint. 2. In response to this regularization effect, weight updates during SFT should follow certain specific structures to reduce forgetting of knowledge learned in pretraining, which can be achieved by using the same optimizer.
Additionally, we specifically compare Muon and AdamW across both the pretraining and SFT stages, finding that Muon performs worse when finetuning on reasoning tasks. Through a synthetic language modeling experiment, we show that this may stem from Muon's strong tendency toward rote memorization, which can harm pattern acquisition in small-data regimes—as is the case in SFT.
---
*Auto-collected on 2026-05-10*