Gaussian Approximation and Concentration of Constant Learning-Rate Stochastic Gradient Descent
Ziyang Wei, Jiaqi Li, Zhipeng Lou, Wei Biao Wu
摘要
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.
问问这篇 Paper
智能体会读完全文。
Lune 把这篇 Paper 索引到了每一个公式,引用它的顶会 Paper 也一样。你提问,回答直接引用原文。
引用它的顶会 Paper1
问问它们各自怎么用它它引用的顶会 Paper3
- Adaptive SGD with Polyak stepsize and Line-search: Robust Convergence and Variance ReductionXiaowen Jiang, Sebastian U. StichNeurIPS 2023 · 被引用 40 次
- Tight High Probability Bounds for Linear Stochastic Approximation with Fixed StepsizeAlain Durmus, Eric Moulines, Alexey Naumov, Sergey Samsonov 等NeurIPS 2021 · 被引用 36 次
- 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 等NeurIPS 2024 · 被引用 23 次
相关 Paper
- Statistical inference for Linear Stochastic Approximation with Markovian NoiseSergey Samsonov, Marina Sheshukova, Eric Moulines, Alexey NaumovNeurIPS 2025 · 被引用 12 次
- Sharp Gaussian approximations for Decentralized Federated LearningSoham Bonnerjee, Sayar Karmakar, Wei Biao WuNeurIPS 2025 · 被引用 2 次
- 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 次
- Statistical Guarantees for High-Dimensional Stochastic Gradient DescentJiaqi Li, Zhipeng Lou, Johannes Schmidt-Hieber, Wei Biao WuNeurIPS 2025 · 被引用 3 次
- Robustness Analysis of Non-Convex Stochastic Gradient Descent using Biased ExpectationsKevin Scaman, Cédric MalherbeNeurIPS 2020 · 被引用 37 次
