Steady-State Behavior of Constant-Stepsize Stochastic Approximation: Gaussian Approximation and Tail Bounds
Yuyang Wang, Felix Wang, Zedong Wang, Ijay Narang, Yuzhou Wang, Siva Maguluri
Abstract
Constant-stepsize stochastic approximation (SA) is widely used in learning for computational efficiency. For a fixed stepsize, the iterates typically admit a stationary distribution that is rarely tractable. Prior work shows that as the stepsize α ↓ 0, the centered-and-scaled steady state converges weakly to a Gaussian random vector. However, for fixed α, this weak convergence offers no usable error bound for approximating the steady-state by its Gaussian limit. This paper provides explicit, non-asymptotic error bounds for fixed α. We first prove general-purpose theorems that bound the Wasserstein distance between the centered-scaled steady state and an appropriate Gaussian distribution, under regularity conditions for drift and moment conditions for noise. To ensure broad applicability, we cover both i.i.d. and Markovian noise models. We then instantiate these theorems for three representative SA settings: (i) stochastic gradient descent (SGD) for smooth strongly convex objectives, (ii) linear SA, and (iii) contractive nonlinear SA. We obtain dimension-and stepsize-dependent, explicit bounds in Wasserstein distance of order α 1/2 log(1/α) for small α. Building on the Wasserstein approximation error, we further derive non-uniform Berry-Esseen-type tail bounds that compare the steady-state tail probability to Gaussian tails. We achieve an explicit error term that decays in both the deviation level and stepsize α. Our analysis combines steady-state Stein's method with moment bounds on the SA iterations, and uses Poisson equation techniques to manage temporal dependence in the Markovian noise setting. We adapt the same toolkit for SGD beyond strongly convexity and study general convex objectives. We identify a non-Gaussian (Gibbs) limiting law under the correct scaling, which is validated numerically, and provide a corresponding pre-limit Wasserstein error bound.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Builds on4
- Training language models to follow instructions with human feedbackLong Ouyang, Jeffrey Wu, Xu Jiang, Diogo Almeida et al.NeurIPS 2022 · 24,707 citations
- An Analysis of Constant Step Size SGD in the Non-convex Regime: Asymptotic Normality and BiasLu Yu, Krishnakumar Balasubramanian, Stanislav Volgushev, Murat A. ErdogduNeurIPS 2021 · 66 citations
- Finite-Sample Analysis of Contractive Stochastic Approximation Using Smooth Convex EnvelopesZaiwei Chen, Siva Theja Maguluri, Sanjay Shakkottai, Karthikeyan ShanmugamNeurIPS 2020 · 66 citations
- Gaussian Approximation and Concentration of Constant Learning-Rate Stochastic Gradient DescentZiyang Wei, Jiaqi Li, Zhipeng Lou, Wei Biao WuNeurIPS 2025 · 2 citations
Related papers
- The Collusion of Memory and Nonlinearity in Stochastic Approximation With Constant StepsizeDongyan Lucy Huo, Yixuan Zhang, Yudong Chen, Qiaomin XieNeurIPS 2024 · 9 citations
- Robustness Analysis of Non-Convex Stochastic Gradient Descent using Biased ExpectationsKevin Scaman, Cédric MalherbeNeurIPS 2020 · 37 citations
- Time-independent Generalization Bounds for SGLD in Non-convex SettingsTyler Farghly, Patrick RebeschiniNeurIPS 2021 · 30 citations
- Tight High Probability Bounds for Linear Stochastic Approximation with Fixed StepsizeAlain Durmus, Eric Moulines, Alexey Naumov, Sergey Samsonov et al.NeurIPS 2021 · 36 citations
- Algorithmic Stability of Heavy-Tailed SGD with General Loss FunctionsAnant Raj, Lingjiong Zhu, Mert Gürbüzbalaban, Umut SimsekliICML 2023 · 21 citations
