Adapt or Forget: Provable Tradeoffs Between Adam and SGD in Nonstationary Optimization
Abstract
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 -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 , and a preconditioner perturbation governed by . We characterize the burn-in time to reach Adam's irreducible tracking floor 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 improve the high-probability error, whereas in drift-dominated regimes, stale first-moment information and preconditioner perturbations can compound the cost of nonstationarity, allowing vanilla SGD to achieve a smaller tracking floor. Our explicit -dependent bounds delineate when adaptive step-sizing is beneficial versus harmful, and provide a theoretical mechanism for Adam's empirical instability and stabilization under distribution shift.
Keywords
Cite
@article{arxiv.2605.04269,
title = {Adapt or Forget: Provable Tradeoffs Between Adam and SGD in Nonstationary Optimization},
author = {Sharan Sahu and Abir Sarkar and Cameron J. Hogan and Martin T. Wells},
journal= {arXiv preprint arXiv:2605.04269},
year = {2026}
}
Comments
39 pages, 11 figures, 1 table