arXiv:2602.20937cs.LG2026-02

提出通用方法,让多种优化器在不同模型规模间直接复用最佳学习率。

Extending $μ$P: Spectral Conditions for Feature Learning Across Optimizers

  • 基于谱条件替代张量编程,推导出多类优化器的μP规则。
  • 在多个模型上实现零样本学习率跨宽度迁移,验证有效性。
  • 适合大规模模型训练者,尤其关注超参调优效率的研究者。

多种自适应一阶和二阶优化方法被提出以加速和扩展大语言模型的训练。这些优化器的性能对超参数(HPs)极为敏感,而为大规模模型调优超参数计算成本高昂。最大更新参数化(μP)是一组缩放规则,旨在使最优超参数与模型大小无关,从而允许在较小(计算成本更低)模型上调优的超参数转移到更大目标模型的训练中。尽管在SGD和Adam上取得良好效果,但为其他优化器推导μP仍具挑战性,因为其底层张量编程方法难以理解。基于最近引入的谱条件作为张量程序的替代方案,我们提出一种新框架,推导出更广泛优化器类别的μP,包括AdamW、ADOPT、LAMB、Sophia、Shampoo和Muon。我们在多个基准模型上实现了这些优化器的μP推导,并展示了上述优化器在模型宽度增加时的零样本学习率迁移。此外,我们还提供了这些优化器在深度缩放参数化方面的实证洞察。

原文摘要 · Abstract (English)

Several variations of adaptive first-order and second-order optimization methods have been proposed to accelerate and scale the training of large language models. The performance of these optimization routines is highly sensitive to the choice of hyperparameters (HPs), which are computationally expensive to tune for large-scale models. Maximal update parameterization $(μ$P$)$ is a set of scaling rules which aims to make the optimal HPs independent of the model size, thereby allowing the HPs tuned on a smaller (computationally cheaper) model to be transferred to train a larger, target model. Despite promising results for SGD and Adam, deriving $μ$P for other optimizers is challenging because the underlying tensor programming approach is difficult to grasp. Building on recent work that introduced spectral conditions as an alternative to tensor programs, we propose a novel framework to derive $μ$P for a broader class of optimizers, including AdamW, ADOPT, LAMB, Sophia, Shampoo and Muon. We implement our $μ$P derivations on multiple benchmark models and demonstrate zero-shot learning rate transfer across increasing model width for the above optimizers. Further, we provide empirical insights into depth-scaling parameterization for these optimizers.

优化器超参迁移大模型训练μP

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