English

When Does Dynamic Preconditioning Preserve the Polyak-Ruppert CLT? A Stabilization Threshold

Statistics Theory 2026-04-28 v1 Optimization and Control Machine Learning Statistics Theory

Abstract

Polyak-Ruppert averaging yields an asymptotically normal estimator with sandwich covariance H1SH1H^{-1}SH^{-1}, the foundation of online inference. When the gradient step is preconditioned by a data-driven matrix PtP_t, we ask how fast PtP_t must stabilize for the central limit theorem (CLT) to remain valid. We resolve this via an exact preconditioner-isolating decomposition of the averaged error that confines PtP_t to a dynamic remainder RnR_n, leaving the martingale and Taylor terms preconditioner-free. Let Mt=(PtH)1M_t = (P_t H)^{-1} denote the effective inverse drift matrix, with MtMt1optβ\|M_t - M_{t-1}\|_{\mathrm{op}} \lesssim t^{-\beta} and step-size exponent α(1/2,1)\alpha \in (1/2, 1). We identify a stabilization-rate threshold β>(α+1)/2\beta > (\alpha+1)/2 and prove that, within the class of polynomial rate hypotheses used in our upper bound, it cannot be weakened: the dynamic remainder nRn\sqrt{n}\,R_n vanishes in L2L^2 whenever β>(α+1)/2\beta > (\alpha+1)/2, and we exhibit sequences satisfying those hypotheses for which it does not vanish when β(α+1)/2\beta \le (\alpha+1)/2. A single stabilization argument certifies three SA variants - SA-AdaGrad, SA-RMSProp, and SA-ONS - with gain ρt=c/t\rho_t = c/t, each delivering one-step L2(op)L^2(\mathrm{op}) stabilization of order t1t^{-1}, yielding the CLT n(xˉnx)N(0,H1SH1)\sqrt{n}(\bar{x}_n - x^*) \to N(0, H^{-1}SH^{-1}); under bounded inputs the pathwise rate β=1\beta = 1 further preserves the n1/6n^{-1/6} Wasserstein rate at α=2/3\alpha^* = 2/3. Under standard regularity conditions, Wald-type online inference remains valid for dynamically preconditioned averaged SGD whose stabilization rate exceeds the threshold.

Keywords

Cite

@article{arxiv.2604.23498,
  title  = {When Does Dynamic Preconditioning Preserve the Polyak-Ruppert CLT? A Stabilization Threshold},
  author = {Sunyoung An and Xiaoming Huo},
  journal= {arXiv preprint arXiv:2604.23498},
  year   = {2026}
}

Comments

46 pages, 5 figures; includes supplementary material with deferred proofs and additional experiments

R2 v1 2026-07-01T12:35:27.258Z