AMUSE结合快速更新与稳定平均,实现无需学习率调度的高效训练。
AMUSE: Anytime Muon with Stable Gradient Evaluation

- 通过时变插值融合Muon快速进展与平均化稳定性。
- 在视觉任务和大模型预训练中超越AdamW与Muon的性能-迭代权衡。
- 支持任意时间训练,无需手动调学习率,适合追求效率的场景。
现代深度学习普遍依赖带固定学习率调度的AdamW,但近期工作挑战了这两个组件:Schedule-Free优化通过迭代平均去除了显式调度,Muon则通过正交化矩阵参数的动量改善更新几何。尽管Muon表现优异,其内在机制仍不清晰。我们通过河流-山谷损失景观分析发现,有效训练沿低曲率主干区域(河流)进行,而高曲率主导方向形成陡峭谷壁引发振荡。实证表明,Muon虽加速主干进展,但也放大主导方向噪声导致振荡。为此,我们提出任意时间正交梯度评估(AMUSE),融合Muon的快速进展与Schedule-Free平均的稳定性。AMUSE采用时变插值系数,初期近似快速Muon序列以快速适应,随后逐步转向稳定平均序列抑制谷壁振荡。结果表明,AMUSE无需学习率调度,支持任意时间训练,在视觉任务与大语言模型预训练中持续优于(Schedule-Free)AdamW与Muon的性能-迭代前沿。
原文摘要 · Abstract (English)
Modern deep learning commonly relies on AdamW with prescribed learning rate schedules, but recent works challenge both components: Schedule-Free optimization removes explicit schedules via iterate averaging, and Muon improves the update geometry by orthogonalizing momentum for matrix parameters. Despite Muon's strong empirical performance, its underlying mechanism remains partially understood. We study Muon through the river-valley loss landscape, where useful training progress occurs along a flat, low-curvature bulk subspace (the river), while high-curvature dominant directions form steep valley walls that induce oscillations. We empirically show that while Muon's orthogonalization accelerates river progress by increasing the bulk component, it also amplifies dominant-direction noise, causing oscillatory trajectories. Building on this, we propose Anytime MUon with Stable gradient Evaluation (AMUSE), which integrates Muon's rapid bulk progress with the stabilizing effect of Schedule-Free averaging. AMUSE uses a time-varying interpolation coefficient that initially evaluates gradients near the fast Muon sequence for rapid adaptation, then gradually shifts toward the stable averaged sequence to suppress valley-wall oscillations. As a result, AMUSE requires no learning rate schedules and supports anytime training. Across vision tasks and large language model pretraining, AMUSE consistently improves the performance-iteration Pareto frontier over (Schedule-Free) AdamW and Muon.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。