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