Time-independent Generalization Bounds for SGLD in Non-convex Settings
Not on the map. Why?
Abstract (arXiv)
We establish generalization error bounds for stochastic gradient Langevin dynamics (SGLD) with constant learning rate under the assumptions of dissipativity and smoothness, a setting that has received increased attention in the sampling/optimization literature. Unlike existing bounds for SGLD in non-convex settings, ours are time-independent and decay to zero as the sample size increases. Using the framework of uniform stability, we establish time-independent bounds by exploiting the Wasserstein contraction property of the Langevin diffusion, which also allows us to circumvent the need to bound gradients using Lipschitz-like assumptions. Our analysis also supports variants of SGLD that use different discretization methods, incorporate Euclidean projections, or use non-isotropic noise.
Key facts. Cited by 27 papers in the map (data through 13 Oct 2026). Builds on 21 of its 33 references that are in the map. First posted on arXiv 25 Nov 2021.
Builds on
21 of 33 references.
References · 21
- 2015Train faster, generalize better: Stability of stochastic gradient descentcited by 549
- 2017Information-theoretic analysis of generalization capability of learning algorithmscited by 302
- 2017Non-convex learning via Stochastic Gradient Langevin Dynamics: a nonasymptotic analysiscited by 242
- 2020Stability of Stochastic Gradient Descent on Nonsmooth Convex Lossescited by 172
- 2019Information-Theoretic Generalization Bounds for SGLD via Data-Dependent Estimatescited by 119
- 2020Fine-Grained Analysis of Stability and Generalization for Stochastic Gradient Descentcited by 118
- 2017User-friendly guarantees for the Langevin Monte Carlo with inaccurate gradientcited by 116
- 2017Global Convergence of Langevin Dynamics Based Algorithms for Nonconvex Optimizationcited by 97
- 2019High probability generalization bounds for uniformly stable algorithms with nearly optimal ratecited by 97
- 2019Sharper bounds for uniformly stable algorithmscited by 92
- 2020Sharpened Generalization Bounds based on Conditional Mutual Information and an Application to Noisy, Iterative Algorithmscited by 89
- 2021Information-Theoretic Generalization Bounds for Stochastic Gradient Descentcited by 84
9 more
- 2017Generalization Bounds of SGLD for Non-convex Learning: Two Theoretical Viewpointscited by 82
- 2018Generalization Error Bounds for Noisy, Iterative Algorithmscited by 81
- 2014Consistency and fluctuations for stochastic gradient Langevin dynamicscited by 65
- 2019On Generalization Error Bounds of Noisy Gradient Methods for Non-Convex Learningcited by 57
- 2018The promises and pitfalls of Stochastic Gradient Langevin Dynamicscited by 56
- 2018Stability and Convergence Trade-off of Iterative Optimization Algorithmscited by 53
- 2019Nonasymptotic estimates for Stochastic Gradient Langevin Dynamics under local conditions in nonconvex optimizationcited by 38
- 2015Distributed Bayesian Learning with Stochastic Natural-gradient Expectation Propagation and the Posterior Servercited by 17
- 2015Learning with Differential Privacy: Stability, Learnability and the Sufficiency and Necessity of ERM Principlecited by 16
Built on by
Citing papers per year: 2022: 7; 2023: 7; 2024: 5; 2025: 6; 2026: 2.
27 papers.
Sort:
By area:
- 2022Neural Networks Efficiently Learn Low-Dimensional Representations with SGDcited by 63
- 2023Towards a Complete Analysis of Langevin Monte Carlo: Beyond Poincar'e Inequalitycited by 31
- 2022Beyond Lipschitz: Sharp Generalization and Excess Risk Bounds for Full-Batch GDcited by 19
- 2023Algorithmic Stability of Heavy-Tailed SGD with General Loss Functionscited by 15
- 2022Algorithmic Stability of Heavy-Tailed Stochastic Gradient Descent on Least Squarescited by 12
- 2023Time-Independent Information-Theoretic Generalization Bounds for SGLDcited by 12
- 2022Generalisation under gradient descent via deterministic PAC-Bayescited by 9
- 2024Uniform Generalization Bounds on Data-Dependent Hypothesis Sets via PAC-Bayesian Theory on Random Setscited by 9
- 2024Information-theoretic Generalization Analysis for Expected Calibration Errorcited by 9
- 2022Stability Based Generalization Bounds for Exponential Family Langevin Dynamicscited by 8
- 2023Uniform-in-Time Wasserstein Stability Bounds for (Noisy) Stochastic Gradient Descentcited by 6
- 2025Implicit Regularisation in Diffusion Models: An Algorithm-Dependent Generalisation Analysiscited by 6
15 more
- 2023Select without Fear: Almost All Mini-Batch Schedules Generalize Optimallycited by 5
- 2022A sharp uniform-in-time error estimate for Stochastic Gradient Langevin Dynamicscited by 4
- 2024Generalization Error Matters in Decentralized Learning Under Byzantine Attackscited by 4
- 2025Algorithm- and Data-Dependent Generalization Bounds for Diffusion Modelscited by 4
- 2023Generalization Bounds for Label Noise Stochastic Gradient Descentcited by 3
- 2025Generalization Bounds for Markov Algorithms through Entropy Flow Computationscited by 2
- 2025Temperature is All You Need for Generalization in Langevin Dynamics and other Markov Processescited by 2
- 2025Beyond Real Data: Synthetic Data through the Lens of Regularizationcited by 2
- 2022Black-Box Generalization: Stability of Zeroth-Order Learningcited by 1
- 2023Geometric ergodicity of SGLD via reflection couplingcited by 1
- 2024Convex SGD: Generalization Without Early Stoppingcited by 1
- 2025Generalization of Gibbs and Langevin Monte Carlo Algorithms in the Interpolation Regimecited by 1
- 2024Provable Accuracy Bounds for Hybrid Dynamical Optimization and Samplingcited by 0
- 2026Sufficient Conditions for Stability of Minimum-Norm Interpolating Deep ReLU Networkscited by 0
- 2026Generalization Bounds of Stochastic Gradient Descent in Homogeneous Neural Networkscited by 0
- 2026Generalization Bounds of Stochastic Gradient Descent in Homogeneous Neural Networks26 Feb 2026
- 2026Sufficient Conditions for Stability of Minimum-Norm Interpolating Deep ReLU Networks14 Feb 2026
- 2025Beyond Real Data: Synthetic Data through the Lens of Regularization9 Oct 2025
- 2025Generalization of Gibbs and Langevin Monte Carlo Algorithms in the Interpolation Regime7 Oct 2025
- 2025Implicit Regularisation in Diffusion Models: An Algorithm-Dependent Generalisation Analysis4 Jul 2025
- 2025Algorithm- and Data-Dependent Generalization Bounds for Diffusion Models4 Jun 2025
- 2025Temperature is All You Need for Generalization in Langevin Dynamics and other Markov Processes25 May 2025
- 2025Generalization Bounds for Markov Algorithms through Entropy Flow Computations11 Feb 2025
- 2024Provable Accuracy Bounds for Hybrid Dynamical Optimization and Sampling8 Oct 2024
- 2024Generalization Error Matters in Decentralized Learning Under Byzantine Attacks11 Jul 2024
- 2024Information-theoretic Generalization Analysis for Expected Calibration Error24 May 2024
- 2024Uniform Generalization Bounds on Data-Dependent Hypothesis Sets via PAC-Bayesian Theory on Random Sets26 Apr 2024
15 more
- 2024Convex SGD: Generalization Without Early Stopping8 Jan 2024
- 2023Time-Independent Information-Theoretic Generalization Bounds for SGLD2 Nov 2023
- 2023Generalization Bounds for Label Noise Stochastic Gradient Descent1 Nov 2023
- 2023Uniform-in-Time Wasserstein Stability Bounds for (Noisy) Stochastic Gradient Descent20 May 2023
- 2023Select without Fear: Almost All Mini-Batch Schedules Generalize Optimally3 May 2023
- 2023Towards a Complete Analysis of Langevin Monte Carlo: Beyond Poincar'e Inequality7 Mar 2023
- 2023Algorithmic Stability of Heavy-Tailed SGD with General Loss Functions27 Jan 2023
- 2023Geometric ergodicity of SGLD via reflection coupling17 Jan 2023
- 2022Neural Networks Efficiently Learn Low-Dimensional Representations with SGD29 Sep 2022
- 2022Generalisation under gradient descent via deterministic PAC-Bayes6 Sep 2022
- 2022A sharp uniform-in-time error estimate for Stochastic Gradient Langevin Dynamics19 Jul 2022
- 2022Algorithmic Stability of Heavy-Tailed Stochastic Gradient Descent on Least Squares2 Jun 2022
- 2022Beyond Lipschitz: Sharp Generalization and Excess Risk Bounds for Full-Batch GD26 Apr 2022
- 2022Black-Box Generalization: Stability of Zeroth-Order Learning14 Feb 2022
- 2022Stability Based Generalization Bounds for Exponential Family Langevin Dynamics9 Jan 2022
- 2022Neural Networks Efficiently Learn Low-Dimensional Representations with SGDcited by 63
- 2023Towards a Complete Analysis of Langevin Monte Carlo: Beyond Poincar'e Inequalitycited by 31
- 2022Beyond Lipschitz: Sharp Generalization and Excess Risk Bounds for Full-Batch GDcited by 19
- 2023Algorithmic Stability of Heavy-Tailed SGD with General Loss Functionscited by 15
- 2022Algorithmic Stability of Heavy-Tailed Stochastic Gradient Descent on Least Squarescited by 12
- 2023Time-Independent Information-Theoretic Generalization Bounds for SGLDcited by 12
- 2022Generalisation under gradient descent via deterministic PAC-Bayescited by 9
- 2024Uniform Generalization Bounds on Data-Dependent Hypothesis Sets via PAC-Bayesian Theory on Random Setscited by 9
- 2024Information-theoretic Generalization Analysis for Expected Calibration Errorcited by 9
- 2022Stability Based Generalization Bounds for Exponential Family Langevin Dynamicscited by 8
- 2023Uniform-in-Time Wasserstein Stability Bounds for (Noisy) Stochastic Gradient Descentcited by 6
- 2023Select without Fear: Almost All Mini-Batch Schedules Generalize Optimallycited by 5
12 most cited of 25 All 25 (JSON)
- 2026Generalization Bounds of Stochastic Gradient Descent in Homogeneous Neural Networks26 Feb 2026
- 2026Sufficient Conditions for Stability of Minimum-Norm Interpolating Deep ReLU Networks14 Feb 2026
- 2025Beyond Real Data: Synthetic Data through the Lens of Regularization9 Oct 2025
- 2025Generalization of Gibbs and Langevin Monte Carlo Algorithms in the Interpolation Regime7 Oct 2025
- 2025Temperature is All You Need for Generalization in Langevin Dynamics and other Markov Processes25 May 2025
- 2025Generalization Bounds for Markov Algorithms through Entropy Flow Computations11 Feb 2025
- 2024Provable Accuracy Bounds for Hybrid Dynamical Optimization and Sampling8 Oct 2024
- 2024Generalization Error Matters in Decentralized Learning Under Byzantine Attacks11 Jul 2024
- 2024Information-theoretic Generalization Analysis for Expected Calibration Error24 May 2024
- 2024Uniform Generalization Bounds on Data-Dependent Hypothesis Sets via PAC-Bayesian Theory on Random Sets26 Apr 2024
- 2024Convex SGD: Generalization Without Early Stopping8 Jan 2024
- 2023Time-Independent Information-Theoretic Generalization Bounds for SGLD2 Nov 2023
12 newest of 25 All 25 (JSON)