mlsanityData to 7 Oct 2026

A Polyak-Ruppert Central Limit Theorem for SA-Adam with Momentum and Non-Convergent Adaptive Preconditioning

arXiv 2606.17364 ↗First posted 15 Jun 2026math.ST math.OC stat.MLcited by 0

Not on the map. Why?

Abstract (arXiv)

Adaptive optimizers combining preconditioning, momentum, and weight decay (Adam and AdamW) are, under Polyak-Ruppert averaging, candidate engines for one-pass inference. Does the averaged iterate keep the classical Polyak-Ruppert central limit theorem (CLT), with sandwich covariance H-1SH-1 (Hessian H, gradient covariance S), under momentum and non-convergent preconditioning? The preconditioner-only analysis does not carry over: with momentum the canonical decomposition collapses to a tautology. Treating the augmented state (iterate, momentum buffer) as a time-varying linear stochastic approximation (SA), we prove (under local stabilization) positive drift stability, a non-autonomous Polyak-Ruppert CLT, and a projection identity. The upshot: the iterate-marginal covariance is exactly the plain stochastic gradient descent (SGD) sandwich H-1SH-1, so the adaptivity is asymptotically invisible. This holds for SA-Adam (sub-linearly vanishing momentum gain, γ∈(α,1); the sub-linear regime is essential), not constant-β deployed Adam. Coupled L2 weight decay yields the ridge-penalized sandwich, extending one-pass inference to regularized problems.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 17 of its 35 references that are in the map. First posted on arXiv 15 Jun 2026.

Builds on

17 of 35 references.

References · 17

  1. 2014Adam: A Method for Stochastic Optimizationcited by 36,960
  2. 2019On the Convergence of Adam and Beyondcited by 834
  3. 2020A Simple Convergence Proof of Adam and Adagradcited by 169
  4. 2014Analysis and Design of Optimization Algorithms via Integral Quadratic Constraintscited by 105
  5. 2017Bridging the Gap between Constant Step Size Stochastic Gradient Descent and Markov Chainscited by 84
  6. 2016Statistical Inference for Model Parameters in Stochastic Gradient Descentcited by 71
  7. 2020On Linear Stochastic Approximation: Fine-grained Polyak-Ruppert and Non-Asymptotic Concentrationcited by 65
  8. 2020Finite Time Analysis of Linear Two-timescale Stochastic Approximation with Markovian Noisecited by 52
  9. 2018Convergence and Dynamical Behavior of the ADAM Algorithm for Non-Convex Stochastic Optimizationcited by 47
  10. 2016Stochastic Heavy Ballcited by 40
  11. 2021Fast and Robust Online Inference with Stochastic Gradient Descent via Random Scalingcited by 28
  12. 2025SGD with Adaptive Preconditioning: Unified Analysis and Momentum Accelerationcited by 12
5 more
  1. 2023Weighted Averaged Stochastic Gradient Descent: Asymptotic Normality and Optimalitycited by 7
  2. 2023Acceleration of stochastic gradient descent with momentum by averaging: finite-sample rates and asymptotic normalitycited by 6
  3. 2024Non-asymptotic Analysis of Biased Adaptive Stochastic Approximationcited by 4
  4. 2026Inference of Online Newton Methods with Nesterov's Accelerated Sketchingcited by 1
  5. 2026When Does Dynamic Preconditioning Preserve the Polyak-Ruppert CLT? A Stabilization Thresholdcited by 1

Built on by

Not cited yet.