arXiv:2605.15530cs.LG2026-05

分层学习率可加速训练,源自博弈优化新视角

Rethinking Neural Network Learning Rates: A Stackelberg Perspective

论文配图:Rethinking Neural Network Learning Rates: A Stackelberg Perspective
图 1 · 摘自论文原文
  • 用斯塔克尔伯格博弈重构训练目标,分层设置学习率
  • 早期梯度更有效,收敛更快,理论保证在多种条件下成立
  • 适合追求训练效率的深度学习研究者与工程应用

神经网络通常使用统一的学习率训练所有层。尽管实证表明分层学习率能加速训练,但其原理和适用条件仍不明确。本文从斯塔克尔伯格优化角度重新审视非均匀学习率:将主体层学习率设小、最后一层设大,可视为对原始目标进行斯塔克尔伯格重构后的双时标交替梯度下降算法。在包含约束集和非光滑激活函数的广泛条件下,建立了有限时间收敛性保证。我们发现两种优势机制:(i) 某些问题实例使斯塔克尔伯格目标具有比原目标更强的优化结构,从而更快收敛至全局最优;(ii) 数值分析显示,斯塔克尔伯格目标在训练初期具有更陡峭的局部曲率,产生更丰富的梯度信息,推动学习加速。监督学习与强化学习实验验证了上述结论。

原文摘要 · Abstract (English)

Neural networks are typically trained with a single learning rate across all layers. While recent empirical evidence suggests that assigning layer-specific learning rates can accelerate training, a principled understanding of the conditions and mechanisms under which non-uniform learning rates are beneficial remains limited. In this work, we investigate non-uniform learning rates through the lens of Stackelberg optimization. Specifically, we demonstrate that training neural networks with a smaller learning rate for the body layers and a larger learning rate for the final layer can be interpreted as a two-time-scale alternating gradient descent algorithm applied to a Stackelberg reformulation of the original objective. We establish finite-time convergence guarantees for the algorithm under broad conditions that accommodate constraint sets and non-smooth activation functions. Beyond convergence, we identify two mechanisms by which non-uniform learning rates can outperform uniform learning rates: (i) we show that certain problem instances induce a Stackelberg objective with stronger optimization structure than the original objective, yielding faster convergence to globally optimal solutions, (ii) our numerical analysis reveals that the Stackelberg objective can exhibit substantially sharper local curvature, especially in early training, which leads to more informative gradients and learning acceleration. Experiments in supervised learning and reinforcement learning support our findings.

优化算法学习率博弈论收敛性

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