Non-Euclidean SGD for Structured Optimization: Unified Analysis and Improved Rates

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Kovalev, Dmitry, Borodich, Ekaterina
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866918202139738112
author Kovalev, Dmitry
Borodich, Ekaterina
author_facet Kovalev, Dmitry
Borodich, Ekaterina
contents Recently, several instances of non-Euclidean SGD, including SignSGD, Lion, and Muon, have attracted significant interest from the optimization community due to their practical success in training deep neural networks. Consequently, a number of works have attempted to explain this success by developing theoretical convergence analyses. Unfortunately, these results cannot properly justify the superior performance of these methods, as they could not beat the convergence rate of vanilla Euclidean SGD. We resolve this important open problem by developing a new unified convergence analysis under the structured smoothness and gradient noise assumption. In particular, our results indicate that non-Euclidean SGD (i) can exploit the sparsity or low-rank structure of the upper bounds on the Hessian and gradient noise, (ii) can provably benefit from popular algorithmic tools such as extrapolation or momentum variance reduction, and (iii) can match the state-of-the-art convergence rates of adaptive and more complex optimization algorithms such as AdaGrad and Shampoo.
format Preprint
id arxiv_https___arxiv_org_abs_2511_11466
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Non-Euclidean SGD for Structured Optimization: Unified Analysis and Improved Rates
Kovalev, Dmitry
Borodich, Ekaterina
Optimization and Control
Machine Learning
Recently, several instances of non-Euclidean SGD, including SignSGD, Lion, and Muon, have attracted significant interest from the optimization community due to their practical success in training deep neural networks. Consequently, a number of works have attempted to explain this success by developing theoretical convergence analyses. Unfortunately, these results cannot properly justify the superior performance of these methods, as they could not beat the convergence rate of vanilla Euclidean SGD. We resolve this important open problem by developing a new unified convergence analysis under the structured smoothness and gradient noise assumption. In particular, our results indicate that non-Euclidean SGD (i) can exploit the sparsity or low-rank structure of the upper bounds on the Hessian and gradient noise, (ii) can provably benefit from popular algorithmic tools such as extrapolation or momentum variance reduction, and (iii) can match the state-of-the-art convergence rates of adaptive and more complex optimization algorithms such as AdaGrad and Shampoo.
title Non-Euclidean SGD for Structured Optimization: Unified Analysis and Improved Rates
topic Optimization and Control
Machine Learning
url https://arxiv.org/abs/2511.11466