arXiv:2510.14009cs.LG2025-10被引 3

为深度神经网络训练设计自适应学习率,提升几何感知优化效率。

Noise-Adaptive Layerwise Learning Rates: Accelerating Geometry-Aware Optimization for Deep Neural Network Training

  • 根据梯度方差动态调整每层学习率,实现层间自适应。
  • 在LLaMA和GPT上收敛速度显著快于现有最优优化器。
  • 适合追求高效训练的Transformer模型研究者使用。

几何感知优化算法(如Muon)在训练深度神经网络(DNN)中表现优异,通过选择不同层的合适范数,并基于范数约束的线性最小化算子(LMO)更新参数。然而,同一范数组内的各层局部曲率存在异质性且随训练过程动态变化。例如,近期研究表明,Transformer各层的尖锐度差异显著且随训练波动,但现有几何感知优化器对同组层采用固定学习率,效率不足。本文提出一种噪声自适应的分层学习率机制,在几何感知优化框架上实现显著加速。该方法实时估计由所选LMO诱导的对偶范数下的梯度方差,据此为每组内各层分配随时间变化的噪声自适应学习率。理论分析表明,本算法具备紧致收敛速率。在LLaMA与GPT等Transformer架构上的实验结果表明,该方法收敛速度优于当前最先进优化器。

原文摘要 · Abstract (English)

Geometry-aware optimization algorithms, such as Muon, have achieved remarkable success in training deep neural networks (DNNs). These methods leverage the underlying geometry of DNNs by selecting appropriate norms for different layers and updating parameters via norm-constrained linear minimization oracles (LMOs). However, even within a group of layers associated with the same norm, the local curvature can be heterogeneous across layers and vary dynamically over the course of training. For example, recent work shows that sharpness varies substantially across transformer layers and throughout training, yet standard geometry-aware optimizers impose fixed learning rates to layers within the same group, which may be inefficient for DNN training. In this paper, we introduce a noise-adaptive layerwise learning rate scheme on top of geometry-aware optimization algorithms and substantially accelerate DNN training compared to methods that use fixed learning rates within each group. Our method estimates gradient variance in the dual norm induced by the chosen LMO on the fly, and uses it to assign time-varying noise-adaptive layerwise learning rates within each group. We provide a theoretical analysis showing that our algorithm achieves a sharp convergence rate. Empirical results on transformer architectures such as LLaMA and GPT demonstrate that our approach achieves faster convergence than state-of-the-art optimizers.

优化器Transformer自适应学习率

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