Universality of AdaGrad Stepsizes for Stochastic Optimization: Inexact Oracle, Acceleration and Variance Reduction
Anton Rodomanov, Xiaowen Jiang, Sebastian U. Stich
Abstract
We present adaptive gradient methods (both basic and accelerated) for solving convex composite optimization problems in which the main part is approximately smooth (a.k.a. -smooth) and can be accessed only via a (potentially biased) stochastic gradient oracle. This setting covers many interesting examples including Hölder smooth problems and various inexact computations of the stochastic gradient. Our methods use AdaGrad stepsizes and are adaptive in the sense that they do not require knowing any problem-dependent constants except an estimate of the diameter of the feasible set but nevertheless achieve the best possible convergence rates as if they knew the corresponding constants. We demonstrate that AdaGrad stepsizes work in a variety of situations by proving, in a unified manner, three types of new results. First, we establish efficiency guarantees for our methods in the classical setting where the oracle's variance is uniformly bounded. We then show that, under more refined assumptions on the variance, the same methods without any modifications enjoy implicit variance reduction properties allowing us to express their complexity estimates in terms of the variance only at the minimizer. Finally, we show how to incorporate explicit SVRG-type variance reduction into our methods and obtain even faster algorithms. In all three cases, we present both basic and accelerated algorithms achieving state-of-the-art complexity bounds. As a direct corollary of our results, we obtain universal stochastic gradient methods for Hölder smooth problems which can be used in all situations.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext cb4b013e-b2dc-40e8-9be3-cca241674e6dCited by top-tier papers6
- Stabilized Proximal-Point Methods for Federated OptimizationXiaowen Jiang, Anton Rodomanov, Sebastian U. StichNeurIPS 2024 · 13 citations
- SGD with Adaptive Preconditioning: Unified Analysis and Momentum AccelerationDmitry KovalevICLR 2026 · 13 citations
- Nesterov Finds GRAAL: Optimal and Adaptive Gradient Method for Convex OptimizationEkaterina Borodich, Dmitry KovalevICLR 2026 · 10 citations
- FedMuon: Federated Learning with Bias-corrected LMO-based OptimizationYuki Takezawa, Anastasia Koloskova, Xiaowen Jiang, Sebastian U. StichICLR 2026 · 9 citations
- Gradient-Variation Online Adaptivity for Accelerated Optimization with Hölder SmoothnessYuheng Zhao, Yu-Hu Yan, Kfir Y. Levy, Peng ZhaoNeurIPS 2025 · 7 citations
Builds on12
- Prodigy: An Expeditiously Adaptive Parameter-Free LearnerKonstantin Mishchenko, Aaron DefazioICML 2024 · 131 citations
- Learning-Rate-Free Learning by D-AdaptationAaron Defazio, Konstantin MishchenkoICML 2023 · 117 citations
- DoG is SGD's Best Friend: A Parameter-Free Dynamic Step Size ScheduleMaor Ivgi, Oliver Hinder, Yair CarmonICML 2023 · 98 citations
- Accelerating SGD with momentum for over-parameterized learningChaoyue Liu, Mikhail BelkinICLR 2020 · 93 citations
- DoWG Unleashed: An Efficient Universal Parameter-Free Gradient Descent MethodAhmed Khaled, Konstantin Mishchenko, Chi JinNeurIPS 2023 · 49 citations
Related papers
- Universal Gradient Methods for Stochastic Convex OptimizationAnton Rodomanov, Ali Kavis, Yongtao Wu, Kimon Antonakopoulos et al.ICML 2024 · 8 citations
- On the Convergence of AdaGrad(Norm) on ℝd: Beyond Convexity, Non-Asymptotic Rate and AccelerationZijian Liu, Ta Duy Nguyen, Alina Ene, Huy L. NguyenICLR 2023
- Extra-Newton: A First Approach to Noise-Adaptive Accelerated Second-Order MethodsKimon Antonakopoulos, Ali Kavis, Volkan CevherNeurIPS 2022 · 17 citations
- Adaptive Gradient Methods for Constrained Convex Optimization and Variational InequalitiesAlina Ene, Huy L. Nguyen, Adrian VladuAAAI 2021 · 35 citations
- A simpler approach to accelerated optimization: iterative averaging meets optimismPooria Joulani, Anant Raj, András György, Csaba SzepesváriICML 2020 · 30 citations
