arXiv:2602.05298stat.MLcs.LG2026-02被引 4

通过动态调整优化器参数,显著提升大模型训练效率

Logarithmic-time Schedules for Scaling Language Models with Momentum

  • 采用对数时间调度,让优化器记忆随训练时长增长
  • 在45M到2.6B参数模型上,比调优后的AdamW节省40%算力
  • 适合大规模语言模型训练,尤其关注算力效率的研究者

实践中,AdamW的超参数(β₁, β₂)和权重衰减λ通常保持固定。我们发现,在大规模语言模型训练中,改变这一做法有充分理由:利用语言数据的幂律结构,设计随时间变化的(β₁, β₂, λ)调度策略可带来显著性能提升。本文研究对数时间调度,使优化器的梯度记忆时长随训练时间增长。尽管原始变体不稳定,但适当的阻尼机制可恢复稳定性并保留长记忆优势。基于此,提出ADANA——一种类AdamW优化器,将对数时间调度与显式阻尼结合,平衡稳定性和性能。我们在45M至2.6B参数的Transformer模型上评估ADANA,对比AdamW、Muon和AdEMAMix。调优后,ADANA相比调优后的AdamW最高可实现40%的计算效率提升,且随着模型规模增大,优势不减反增。进一步表明,对AdEMAMix应用对数时间调度亦有类似收益,且仅对数时间权重衰减即可带来显著改进。最后,提出多种变体以缓解潜在失败模式,增强鲁棒性。

原文摘要 · Abstract (English)

In practice, the hyperparameters $(β_1, β_2)$ and weight-decay $λ$ in AdamW are typically kept at fixed values. Is there any reason to do otherwise? We show that for large-scale language model training, the answer is yes: by exploiting the power-law structure of language data, one can design time-varying schedules for $(β_1, β_2, λ)$ that deliver substantial performance gains. We study logarithmic-time scheduling, in which the optimizer's gradient memory horizon grows with training time. Although naive variants of this are unstable, we show that suitable damping mechanisms restore stability while preserving the benefits of longer memory. Based on this, we present ADANA, an AdamW-like optimizer that couples log-time schedules with explicit damping to balance stability and performance. We empirically evaluate ADANA across transformer scalings (45M to 2.6B parameters), comparing against AdamW, Muon, and AdEMAMix. When properly tuned, ADANA achieves up to 40% compute efficiency relative to a tuned AdamW, with gains that persist--and even improve--as model scale increases. We further show that similar benefits arise when applying logarithmic-time scheduling to AdEMAMix, and that logarithmic-time weight-decay alone can yield significant improvements. Finally, we present variants of ADANA that mitigate potential failure modes and improve robustness.

优化器大模型训练效率对数调度

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