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.

Ссылки и действия