Adapt or Forget: Provable Tradeoffs Between Adam and SGD in Nonstationary Optimization
Not on the map. Why?
Abstract (arXiv)
We provide a theoretical analysis of Adam under non-stationary stochastic objectives, separating two regimes: Euclidean tracking under adaptive strong monotonicity of the Adam-preconditioned mean-gradient operator, and high-probability projected stationarity guarantees under general L-smooth objectives. In the tracking regime, we derive finite-time expected and high-probability bounds that decompose sharply into four components: initialization, objective drift, a first-moment tracking error governed by β1, and a preconditioner perturbation governed by β2. We characterize the burn-in time required for the transient terms to decay to the asymptotic tracking bound under constant and step-decay schedules. We also prove a high-probability bound on the average projected stationarity gap for Adam under distribution shift. Across both analyses, our bounds reveal a noise--drift tradeoff: in noise-dominated regimes, first-moment averaging and adaptive preconditioning can yield favorable upper guarantees, whereas in drift-dominated regimes, stale first-moment information and preconditioner perturbations can enlarge Adam's tracking guarantee, potentially allowing vanilla SGD to attain a smaller tracking error. Our explicit (β1,β2,ε)-dependent bounds identify mechanisms through which adaptive step-sizing can help or hurt under nonstationarity and provide theoretical explanations consistent with Adam's empirical instability and stabilization under distribution shift.
Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Builds on 31 of its 48 references that are in the map. First posted on arXiv 5 May 2026.
Builds on
31 of 48 references.
References · 31
- 2020Language Models are Few-Shot Learnerscited by 32,419
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
- 2019Advances and Open Problems in Federated Learningcited by 3,046
- 2015Trust Region Policy Optimizationcited by 2,580
- 2018Continual Lifelong Learning with Neural Networks: A Reviewcited by 1,330
- 2019On the Convergence of Adam and Beyondcited by 841
- 2019Introduction to Online Convex Optimizationcited by 803
- 2017The Marginal Value of Adaptive Gradient Methods in Machine Learningcited by 250
- 2023Understanding plasticity in neural networkscited by 174
- 2020A Simple Convergence Proof of Adam and Adagradcited by 171
- 2024Why Transformers Need Adam: A Hessian Perspectivecited by 124
- 2023Noise Is Not the Main Factor Behind the Gap Between SGD and Adam on Transformers, but Sign Descent Might Becited by 114
19 more
- 2023Convergence of Adam Under Relaxed Assumptionscited by 99
- 2018Tight Analyses for Non-Smooth Stochastic Gradient Descentcited by 97
- 2024Heavy-Tailed Class Imbalance and Why Adam Outperforms Gradient Descent on Language Modelscited by 80
- 2021Adaptivity and Non-stationarity: Problem-dependent Dynamic Regret for Online Convex Optimizationcited by 66
- 2023Maintaining Plasticity in Deep Continual Learningcited by 43
- 2024Slow and Steady Wins the Race: Maintaining Plasticity with Hare and Tortoise Networkscited by 43
- 2019Continuous Time Analysis of Momentum Methodscited by 32
- 2016On the Influence of Momentum Acceleration on Online Learningcited by 24
- 2024On Convergence of Adam for Stochastic Optimization under Relaxed Assumptionscited by 21
- 2024ADOPT: Modified Adam Can Converge with Any β2 with the Optimal Ratecited by 18
- 2021Stochastic Optimization under Distributional Driftcited by 15
- 2025Adaptive Preconditioners Trigger Loss Spikes in Adamcited by 13
- 2024Adam on Local Time: Addressing Nonstationarity in RL with Relative Adam Timestepscited by 11
- 2024A Comprehensive Framework for Analyzing the Convergence of Adam: Bridging the Gap with SGDcited by 9
- 2024Statistical Efficiency of Distributional Temporal Difference Learning and Freedman's Inequality in Hilbert Spacescited by 8
- 2026SGD with Dependent Data: Optimal Estimation, Regret, and Inferencecited by 6
- 2026Why Adam Can Beat SGD: Second-Moment Normalization Yields Sharper Tailscited by 4
- 2026On the Provable Suboptimality of Momentum SGD in Nonstationary Stochastic Optimizationcited by 2
- 2026Rethinking Adam for Time Series Forecasting: A Simple Heuristic to Improve Optimization under Distribution Shiftscited by 1
Built on by
Citing papers per year: 2026: 1.
1 paper.