English

Adapt or Forget: Provable Tradeoffs Between Adam and SGD in Nonstationary Optimization

Machine Learning 2026-05-07 v1 Machine Learning

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 LL-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\beta_1, and a preconditioner perturbation governed by β2\beta_2. 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 (β1,β2,ϵ)(\beta_1,\beta_2,\epsilon)-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