arXiv:2605.22297cs.LGcs.AI2026-05被引 3

根据层间权重分布差异,动态调整学习率以加速大模型训练。

One LR Doesn't Fit All: Heavy-Tail Guided Layerwise Learning Rates for LLMs

  • 依据权重相关矩阵的重尾特性,为每层分配不同学习率。
  • 在1B~3B模型上实现最高1.5倍训练提速,零样本准确率提升1.9~2.03个百分点。
  • 无需额外调参即可复用统一学习率基线设置,适合大规模模型训练者。

学习率配置是现代深度学习的基础环节。当前普遍对所有层采用统一学习率的做法忽略了Transformer结构的异质性,可能限制其作为大语言模型骨干的有效性。本文提出分层学习率(LLR)方法,基于重尾自正则化(HT-SR)理论,通过量化权重相关矩阵的经验谱密度(ESD)来评估各层的重尾程度:重尾性较弱的层赋予更大学习率以加速训练,重尾性较强的层则给予更小学习率。该策略促进各层训练平衡,实现更快收敛与更好泛化。在从LLaMA到GPT-nano等多种架构、AdamW与Muon等优化器、60M至3B参数规模及最多100B训练令牌的广泛实验中,LLR表现优异,最高可实现1.5倍训练加速,并持续优于统一学习率基线。特别地,1B模型的平均零样本准确率从47.09%提升至49.02%,3B模型从48.58%升至50.61%。关键优势在于低调参成本:几乎可直接将统一学习率基线的最优设置迁移至本方法。代码已开源。

原文摘要 · Abstract (English)

Learning rate configuration is a fundamental aspect of modern deep learning. The prevailing practice of applying a uniform learning rate across all layers overlooks the structural heterogeneity of Transformers, potentially limiting their effectiveness as the backbone of Large Language Models (LLMs). In this paper, we introduce Layerwise Learning Rate (LLR), an adaptive scheme that assigns distinct learning rates to individual Transformer layers. Our method is grounded in Heavy-Tailed Self-Regularization (HT-SR) theory, which characterizes the empirical spectral density (ESD) of weight correlation matrices to quantify heavy-tailedness. Layers with weaker heavy-tailedness are assigned larger learning rates to accelerate training, while layers with stronger heavy-tailedness receive smaller learning rates. By tailoring learning rates in this manner, LLR promotes more balanced training across layers, leading to faster convergence and improved generalization. Extensive experiments across architectures ranging from LLaMA to GPT-nano, optimizers including AdamW and Muon, and model scales from 60M to 3B parameters with up to 100B training tokens demonstrate the effectiveness of LLR. LLR achieves up to 1.5x training speedup and consistently outperforms uniform-learning-rate baselines. In particular, it improves the average zero-shot accuracy of 1B models from 47.09% to 49.02%, and that of 3B models from 48.58% to 50.61%. A key advantage of LLR is its low tuning overhead: it can transfer nearly optimal learning-rate settings directly from the uniform baseline. Code is available at https://github.com/hed-ucas/Layer-wise-Learning-Rate.

大模型训练学习率调度Transformer

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