arXiv:2506.08419cs.LGmath.OC2025-06被引 2

根据梯度对齐动态调整学习率,减少调参成本。

Online Learning-guided Learning Rate Adaptation via Gradient Alignment

  • 通过连续梯度对齐度和局部曲率估计自适应调节学习率
  • 在多种初始学习率下表现稳定,性能媲美调优后的优化器
  • 适合追求少调参的工程应用与复杂模型训练

大规模深度学习模型的优化器性能高度依赖学习率的精细调整,通常需进行广泛的超参数网格搜索。本文提出一种基于梯度对齐的学习率自适应框架GALA(Gradient Alignment-based Learning rate Adaptation),通过追踪连续梯度间的对齐程度并结合局部曲率估计,动态调整学习率。基于收敛性分析,将学习率选择建模为一维在线学习问题。结合如Follow-the-Regularized-Leader等在线学习算法,GALA生成灵活自适应的学习率调度策略:当连续梯度对齐时提高学习率,否则降低。我们建立了配备GALA的归一化SGD在平滑非凸设定下的数据自适应收敛速率。实验表明,SGD和Adam等常见优化器在引入GALA后,能在广泛初始学习率范围内保持稳健性能,且无需额外调参即可达到竞争力表现。

原文摘要 · Abstract (English)

The performance of an optimizer on large-scale deep learning models depends critically on fine-tuning the learning rate, often requiring an extensive grid search over base learning rates, schedules, and other hyperparameters. In this paper, we propose a principled framework called GALA (Gradient Alignment-based Learning rate Adaptation), which dynamically adjusts the learning rate by tracking the alignment between consecutive gradients and using a local curvature estimate. Guided by the convergence analysis, we formulate the problem of selecting the learning rate as a one-dimensional online learning problem. When paired with an online learning algorithm such as Follow-the-Regularized-Leader, our method produces a flexible, adaptive learning rate schedule that tends to increase when consecutive gradients are aligned and decrease otherwise. We establish a data-adaptive convergence rate for normalized SGD equipped with GALA in the smooth, nonconvex setting. Empirically, common optimizers such as SGD and Adam, when augmented with GALA, demonstrate robust performance across a wide range of initial learning rates and perform competitively without the need for tuning.

学习率自适应优化器改进在线学习

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