arXiv:2508.15071cs.LGmath.OC2025-08NeurIPS被引 1

提出NGN-M优化器,提升步长稳定性并保持高性能。

Enhancing Optimizer Stability: Momentum Adaptation of The NGN Step-size

  • 基于动量改进NGN步长,降低对超参数依赖。
  • 在宽松假设下仍达1/√K收敛速率,无需有界梯度假设。
  • 实测对步长选择更鲁棒,性能媲美或超越主流优化器。

现代结合动量与自适应步长的优化算法在众多复杂深度学习任务中表现优异,但其效果常高度依赖超参数,尤其是步长,调参过程繁琐且耗时。为此,近期研究聚焦于提升优化器在广泛超参数下的稳定性 [Schaipp et al., 2024]。本文提出一种新算法,通过改进NGN步长方法 [Orvieto and Xiao, 2024],在保持当前最优性能的同时,显著提升对步长超参数的稳定性。具体而言,我们设计了动量版本(NGN-M),在比以往方法更宽松的假设下,仍能实现标准的 $/mathcal{O}(1/\\/sqrt{K})$ 收敛速率,且无需插值条件或有界随机梯度/迭代假设。此外,实验表明,将NGN步长与动量结合后,对步长选择更具鲁棒性,性能可媲美甚至超越其他先进优化器。

原文摘要 · Abstract (English)

Modern optimization algorithms that incorporate momentum and adaptive step-size offer improved performance in numerous challenging deep learning tasks. However, their effectiveness is often highly sensitive to the choice of hyperparameters, especially the step-size. Tuning these parameters is often difficult, resource-intensive, and time-consuming. Therefore, recent efforts have been directed toward enhancing the stability of optimizers across a wide range of hyperparameter choices [Schaipp et al., 2024]. In this paper, we introduce an algorithm that matches the performance of state-of-the-art optimizers while improving stability to the choice of the step-size hyperparameter through a novel adaptation of the NGN step-size method [Orvieto and Xiao, 2024]. Specifically, we propose a momentum-based version (NGN-M) that attains the standard convergence rate of $\mathcal{O}(1/\sqrt{K})$ under less restrictive assumptions, without the need for interpolation condition or assumptions of bounded stochastic gradients or iterates, in contrast to previous approaches. Additionally, we empirically demonstrate that the combination of the NGN step-size with momentum results in enhanced robustness to the choice of the step-size hyperparameter while delivering performance that is comparable to or surpasses other state-of-the-art optimizers.

优化器动量稳定性自适应

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