Gaussian Approximation and Concentration of Constant Learning-Rate Stochastic Gradient Descent
Ziyang Wei, Jiaqi Li, Zhipeng Lou, Wei Biao Wu
Abstract
We establish a comprehensive finite-sample and asymptotic theory for stochastic gradient descent (SGD) with constant learning rates. First, we propose a novel linear approximation technique to provide a quenched central limit theorem (CLT) for SGD iterates with refined tail properties, showing that regardless of the chosen initialization, the fluctuations of the algorithm around its target point converge to a multivariate normal distribution. Our conditions are substantially milder than those required in the classical CLTs for SGD, yet offering a stronger convergence result. Furthermore, we derive the first Berry-Esseen bound – the Gaussian approximation error – for the constant learning-rate SGD, which is sharp compared to the decaying learning-rate schemes in the literature. Beyond the moment convergence, we also provide the Nagaev-type inequality for the SGD tail probabilities by adopting the autoregressive approximation techniques, which entails non-asymptotic large-deviation guarantees. These results are verified via numerical simulations, paving the way for theoretically grounded uncertainty quantification, especially with non-asymptotic validity.
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.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext b5e9e2fd-4c43-41fc-b3cf-a0649f884c05Cited by top-tier papers1
Ask how each one uses itBuilds on3
- Adaptive SGD with Polyak stepsize and Line-search: Robust Convergence and Variance ReductionXiaowen Jiang, Sebastian U. StichNeurIPS 2023 · 40 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
- Gaussian Approximation and Multiplier Bootstrap for Polyak-Ruppert Averaged Linear Stochastic Approximation with Applications to TD LearningSergey Samsonov, Eric Moulines, Qi-Man Shao, Zhuo-Song Zhang et al.NeurIPS 2024 · 23 citations
Related papers
- Statistical inference for Linear Stochastic Approximation with Markovian NoiseSergey Samsonov, Marina Sheshukova, Eric Moulines, Alexey NaumovNeurIPS 2025 · 12 citations
- Sharp Gaussian approximations for Decentralized Federated LearningSoham Bonnerjee, Sayar Karmakar, Wei Biao WuNeurIPS 2025 · 2 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
- Statistical Guarantees for High-Dimensional Stochastic Gradient DescentJiaqi Li, Zhipeng Lou, Johannes Schmidt-Hieber, Wei Biao WuNeurIPS 2025 · 3 citations
- Robustness Analysis of Non-Convex Stochastic Gradient Descent using Biased ExpectationsKevin Scaman, Cédric MalherbeNeurIPS 2020 · 37 citations
