【论文速递】Pretraining Data Scale Reverses the Muon-AdamW Fine-Tuning Gap

英文标题：Pretraining Data Scale Reverses the Muon-AdamW Fine-Tuning Gap

arXiv编号：2611.03633
作者：Catherine Li, Jingchu Gai, Andrej Risteski
来源：arXiv
原文链接：https://arxiv.org/abs/2611.03633

中文摘要：
Muon has recently emerged as a promising optimizer for pretraining large language models, offering a speedup over AdamW in minimizing pretraining loss. However, it remains unclear whether Muon's advantage persists after fine-tuning. In this paper, we study the downstream behavior of Muon and AdamW and uncover a surprising trend: beyond a sufficiently large pretraining data budget, AdamW outperforms Muon after fine-tuning. Moreover, as model size increases, AdamW overtakes Muon at progressively smaller pretraining token-parameter ratios. To explain this reversal, we view fine-tuning as a perturbation of the outputs and introduce the margin-to-spread ratio, which measures how far the correct logit lies above the wrong ones relative to how widely the wrong logits are spread. In language models, Muon has the larger margin-to-spread ratio at small data budgets and AdamW at large ones, mirroring the fine-tuning trend. In a linear associative memory, we prove that after one update the margin-to-spread ratio of gradient descent gains on that of Muon as the data budget grows, and simulations show that this trend persists after extended training.

本文由华人空间 AI 论文助手自动抓取翻译，仅供学术交流。