Enhancing Optimizer Stability: Momentum Adaptation of The NGN Step-size
2508.15071v1
cs.LG, math.OC, stat.ML
2025-08-23
Авторы:
Rustem Islamov, Niccolo Ajroldi, Antonio Orvieto, Aurelien Lucchi
Резюме на русском
## Контекст
Modern optimization algorithms, включая моментный и адаптивный шаг, показывают выдающиеся результаты в решении сложных задач диплорнга. Однако их эффективность сильно зависит от выбора гиперпараметров, особенно — шага. Настройка этих параметров является трудоемкой, ресурсозатратной и временно затягивающей процедурой. Из-за этого в последнее время наблюдается активность в разработке методов, улучшающих стабильность оптимизаторов в широком диапазоне гиперпараметрических значений.
## Метод
Мы предлагаем модификацию метода NGN step-size, интегрирующую моменту (NGN-M). Наша реализация сохраняет стандартный темп убывания ошибок $\mathcal{O}(1/\sqrt{K})$, но с более гибкими предположениями. Мы отказываемся от условий интерполяции и предположений о градиентах и итерациях, что позволяет использовать метод в более широких условиях. Этот подход обеспечивает более высокую стабильность оптимизатора в отношении шага, не требуя дорогостоящей настройки.
## Результаты
Мы проводили эксперименты на нескольких вычислительно тяжёлых задачах, включая задачи с динамическими градиентами и нерегулярным поведением. Наши результаты показывают, что NGN-M не только показывает лучшую стабильность в отношении шага, но и достигает или превышает результаты других современных оптимизаторов. Мы также провели анализ вариации гиперпараметров, подтвердив высокую надёжность наших результатов.
## Значимость
Наш метод может быть применён в широком кругу задач, где требуется эффективность и стабильность оптимизации. Он уменьшает трудоемкость настройки гиперпараметров, сделав оптимизацию более доступной для широкого круга пользователей. Благодаря улучшенной стабильности, он может привести к повышению производительности в задачах, где традиционные оптимизаторы не успевают.
## Выводы
Мы установили, что модификация NGN с моментом является эффективным решением для улучшения стабильности оптимизаторов. Будущие исследования будут направлены на дальнейшее оптимизацию алгоритма и его применение в новых типах задач, где стабильность и эффективность оптимизации являются ключевыми факторами.
Abstract
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.