mlsanityData to 13 Oct 2026

OptEMA: Adaptive Exponential Moving Average for Stochastic Optimization with Zero-Noise Optimality

arXiv 2603.09923 ↗First posted 10 Mar 2026cs.LG cs.NA math.NAcited by 0

Not on the map. Why?

Abstract (arXiv)

Exponential moving averages (EMAs) are a central component of widely used adaptive optimizers such as Adam. However, existing analyses of Adam-style methods often yield suboptimal guarantees in the zero-noise regime, rely on open-loop parameter schedules, or require prior knowledge of smoothness constants. Motivated by these limitations, we introduce OptEMA and analyze two complementary variants: OptEMA-M, which applies an adaptive, decreasing EMA coefficient to the first moment with a fixed second-moment decay, and OptEMA-V, which swaps these roles. At the heart of these variants is a Corrected AdaGrad-Norm coefficient schedule. This formulation renders OptEMA algorithmically closed-loop and Lipschitz-free, meaning its effective stepsizes are trajectory-dependent and require no parameterization via the Lipschitz constant. Under lower-boundedness, unbiasedness, bounded variance, average smoothness, and a bounded stochastic-gradient condition used to control the adaptive normalizers, we prove that both variants achieve the unified noise-adaptive rate tildeO left(T-1/2+σ1/2T-1/4right) for the averaged gradient norm. In the zero-noise regime, these bounds automatically reduce to the nearly optimal deterministic rate widetildeO(T-1/2) without manual hyperparameter retuning.

Key facts. No citing papers in the map yet (data through 13 Oct 2026). Builds on 22 of its 29 references that are in the map. First posted on arXiv 10 Mar 2026.

Builds on

22 of 29 references.

References · 22

  1. 2018Adafactor: Adaptive Learning Rates with Sublinear Memory Costcited by 900
  2. 2019On the Convergence of Adam and Beyondcited by 837
  3. 2013Stochastic First- and Zeroth-order Methods for Nonconvex Stochastic Programmingcited by 718
  4. 2018signSGD: Compressed Optimisation for Non-Convex Problemscited by 567
  5. 2023Symbolic Discovery of Optimization Algorithmscited by 425
  6. 2019Lower Bounds for Non-Convex Stochastic Optimizationcited by 306
  7. 2019Momentum-Based Variance Reduction in Non-Convex SGDcited by 303
  8. 2019Adaptive Gradient Methods with Dynamic Bound of Learning Ratecited by 183
  9. 2018AdaGrad stepsizes: Sharp convergence over nonconvex landscapescited by 182
  10. 2018On the Convergence of A Class of Adam-Type Algorithms for Non-Convex Optimizationcited by 174
  11. 2013Mini-batch Stochastic Approximation Methods for Nonconvex Stochastic Composite Optimizationcited by 142
  12. 2022Adam Can Converge Without Any Modification On Update Rulescited by 114
10 more
  1. 2023Learning-Rate-Free Learning by D-Adaptationcited by 113
  2. 2023DoG is SGD's Best Friend: A Parameter-Free Dynamic Step Size Schedulecited by 85
  3. 2024MARS: Unleashing the Power of Variance Reduction for Training Large Modelscited by 61
  4. 2021SUPER-ADAM: Faster and Universal Framework of Adaptive Gradientscited by 40
  5. 2026Convergence of Muon with Newton-Schulzcited by 33
  6. 2025On the Convergence of Muon and Beyondcited by 29
  7. 2021Towards Practical Adam: Non-Convexity, Convergence Theory, and Mini-Batch Accelerationcited by 25
  8. 2023Convergence of Adam for Non-convex Objectives: Relaxed Hyperparameters and Non-ergodic Casecited by 14
  9. 2022META-STORM: Generalized Fully-Adaptive Variance Reduced SGD for Unbounded Functionscited by 5
  10. 2024Adaptive Variance Reduction for Stochastic Optimization under Weaker Assumptionscited by 5

Built on by

Not cited yet.