无需学习率衰减,也能高效训练大模型。
Through the River: Understanding the Benefit of Schedule-Free Methods for Language Model Training
- 用无调度方法自动适应训练进程,省去复杂调参。
- 在大批次下表现更稳,比传统方法更快收敛。
- 适合持续扩展的超大规模语言模型训练。
随着模型和数据集规模迅速扩大,传统的固定计算预算预训练策略(如余弦学习率调度)在大规模训练中日益不足。近期替代方案如暖身-稳定-衰减(WSD)调度和权重平均虽更具灵活性,但前者依赖显式衰减阶段,后者需额外内存。为寻找更严谨且可扩展的方案,我们重新审视了近期提出的无调度(Schedule-Free, SF)方法 [Defazio et al., 2024],其在多种场景中已表现出优异性能。我们发现,SF-AdamW 能有效穿越损失曲面的“河流”结构,无需衰减阶段或辅助平均。通过理论与实证分析,揭示其隐式实现权重平均而无内存开销。基于此,我们提出改进的SF变体,增强对动量的鲁棒性,并在大批次下表现更优,解决了原始方法的关键缺陷。这些结果确立了SF作为语言模型训练中实用、可扩展且理论扎实的方法。
原文摘要 · Abstract (English)
As both model and dataset sizes continue to scale rapidly, conventional pretraining strategies with fixed compute budgets-such as cosine learning rate schedules-are increasingly inadequate for large-scale training. Recent alternatives, including warmup-stable-decay (WSD) schedules and weight averaging, offer greater flexibility. However, WSD relies on explicit decay phases to track progress, while weight averaging addresses this limitation at the cost of additional memory. In search of a more principled and scalable alternative, we revisit the Schedule-Free (SF) method [Defazio et al., 2024], which has shown strong empirical performance across diverse settings. We show that SF-AdamW effectively navigates the "river" structure of the loss landscape without decay phases or auxiliary averaging, making it particularly suitable for continuously scaling training workloads. To understand this behavior, we conduct a theoretical and empirical analysis of SF dynamics, revealing that it implicitly performs weight averaging without memory overhead. Guided by this analysis, we propose a refined variant of SF that improves robustness to momentum and performs better under large batch sizes, addressing key limitations of the original method. Together, these results establish SF as a practical, scalable, and theoretically grounded approach for language model training.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。