A Hitting Time Analysis of Stochastic Gradient Langevin Dynamics
Not on the map. Why?
Abstract (arXiv)
We study the Stochastic Gradient Langevin Dynamics (SGLD) algorithm for non-convex optimization. The algorithm performs stochastic gradient descent, where in each step it injects appropriately scaled Gaussian noise to the update. We analyze the algorithm's hitting time to an arbitrary subset of the parameter space. Two results follow from our general theory: First, we prove that for empirical risk minimization, if the empirical risk is point-wise close to the (smooth) population risk, then the algorithm achieves an approximate local minimum of the population risk in polynomial time, escaping suboptimal local minima that only exist in the empirical risk. Second, we show that SGLD improves on one of the best known learnability results for learning linear classifiers under the zero-one loss.
Key facts. Cited by 70 papers in the map (data through 7 Oct 2026). First posted on arXiv 18 Feb 2017.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 15; 2022: 17; 2023: 15; 2024: 9; 2025: 9; 2026: 5.
70 papers.
Sort:
By area:
- 2021Small random initialization is akin to spectral learning: Optimization and generalization guarantees for overparameterized low-rank matrix reconstructioncited by 77
- 2022High-dimensional limit theorems for SGD: Effective dynamics and critical scalingcited by 69
- 2023Revisiting the Noise Model of Stochastic Gradient Descentcited by 35
- 2024Stable Neural Stochastic Differential Equations in Analyzing Irregular Time Series Datacited by 35
- 2022Testing distributional assumptions of learning algorithmscited by 30
- 2021Power-law escape rate of SGDcited by 29
- 2021Learning General Halfspaces with General Massart Noise under the Gaussian Distributioncited by 27
- 2022Bayesian autoencoders for data-driven discovery of coordinates, governing equations and fundamental constantscited by 24
- 2021Accelerated replica exchange stochastic gradient Langevin diffusion enhanced Bayesian DeepONet for solving noisy parametric PDEscited by 22
- 2021Improved Algorithms for Efficient Active Learning Halfspaces with Massart and Tsybakov noisecited by 18
- 2021Forster Decomposition and Learning Halfspaces with Noisecited by 16
- 2021Generalization Bounds for Noisy Iterative Algorithms Using Properties of Additive Noise Channelscited by 15
38 more
- 2021Provable Compressed Sensing with Generative Priors via Langevin Dynamicscited by 14
- 2021On Convergence of Federated Averaging Langevin Dynamicscited by 14
- 2022Optimal SQ Lower Bounds for Learning Halfspaces with Massart Noisecited by 14
- 2022Generalization Error Bounds for Deep Neural Networks Trained by SGDcited by 14
- 2023Time-Independent Information-Theoretic Generalization Bounds for SGLDcited by 12
- 2023Deep Insights into Noisy Pseudo Labeling on Graph Datacited by 11
- 2021Boosting in the Presence of Massart Noisecited by 10
- 2021ReLU Regression with Massart Noisecited by 10
- 2024Accelerating Approximate Thompson Sampling with Underdamped Langevin Monte Carlocited by 10
- 2022Interacting Contour Stochastic Gradient Langevin Dynamicscited by 9
- 2021Stochastic Gradient Langevin Dynamics with Variance Reductioncited by 7
- 2023Quantum Langevin Dynamics for Optimizationcited by 7
- 2024Understanding the Generalization Benefits of Late Learning Rate Decaycited by 6
- 2025A Near-optimal Algorithm for Learning Margin Halfspaces with Massart Noisecited by 6
- 2025Reheated Gradient-based Discrete Sampling for Combinatorial Optimizationcited by 6
- 2022Generalization Bounds for Gradient Methods via Discrete and Continuous Priorcited by 5
- 2023Efficient displacement convex optimization with particle gradient descentcited by 5
- 2023Near-Optimal Bounds for Learning Gaussian Halfspaces with Random Classification Noisecited by 5
- 2025Learning Noisy Halfspaces with a Margin: Massart is No Harder than Randomcited by 5
- 2025From Global to Local: A Scalable Benchmark for Local Posterior Samplingcited by 5
- 2021Decentralized Bayesian Learning with Metropolis-Adjusted Hamiltonian Monte Carlocited by 4
- 2023Stochastic Quantum Sampling for Non-Logconcave Distributions and Estimating Partition Functionscited by 4
- 2024Sampling from multi-modal distributions with polynomial query complexity in fixed dimension via reverse diffusioncited by 4
- 2025Testing Noise Assumptions of Learning Algorithmscited by 4
- 2022Convergence Error Analysis of Reflected Gradient Langevin Dynamics for Globally Optimizing Non-Convex Constrained Problemscited by 3
- 2022A sharp uniform-in-time error estimate for Stochastic Gradient Langevin Dynamicscited by 3
- 2023Constructing Semantics-Aware Adversarial Examples with a Probabilistic Perspectivecited by 3
- 2026On Forgetting and Stability of Score-based Generative modelscited by 3
- 2022Tackling benign nonconvexity with smoothing and stochastic gradientscited by 2
- 2022Weak Convergence of Approximate reflection coupling and its Application to Non-convex Optimizationcited by 2
- 2023Using Stochastic Gradient Descent to Smooth Nonconvex Functions: Analysis of Implicit Graduated Optimizationcited by 2
- 2025Smoothed Agnostic Learning of Halfspaces over the Hypercubecited by 2
- 2026On the Superlinear Relationship between SGD Noise Covariance and Loss Landscape Curvaturecited by 2
- 2022Continuous-time Analysis for Variational Inequalities: An Overview and Desideratacited by 1
- 2022On PAC Learning Halfspaces in Non-interactive Local Privacy Model with Public Unlabeled Datacited by 1
- 2022Tradeoffs between convergence rate and noise amplification for momentum-based accelerated optimization algorithmscited by 1
- 2022Recycling Scraps: Improving Private Learning by Leveraging Intermediate Checkpointscited by 1
- 2023Improved Langevin Monte Carlo for stochastic optimization via landscape modificationcited by 1
50 most cited of 70 All 70 (JSON)
- 2026Avoiding unsafe sets when training with Langevin Dynamics8 Jul 2026
- 2026When Both Layers Learn: Training Dynamics of Representing Linear Models via ReLU Networks3 Jun 2026
- 2026Testable Learning of General Halfspaces under Massart Noise25 Feb 2026
- 2026On the Superlinear Relationship between SGD Noise Covariance and Loss Landscape Curvature5 Feb 2026
- 2026On Forgetting and Stability of Score-based Generative models29 Jan 2026
- 2025Smoothed Agnostic Learning of Halfspaces over the Hypercube21 Nov 2025
- 2025A Fully Polynomial-Time Algorithm for Robustly Learning Halfspaces over the Hypercube10 Nov 2025
- 2025From Global to Local: A Scalable Benchmark for Local Posterior Sampling29 Jul 2025
- 2025Continuous Policy and Value Iteration for Relaxed Stochastic Control Problems and Its Convergence9 Jun 2025
- 2025Quantum Speedups for Sampling and Non-convex Optimization with Stochastic Oracles4 Apr 2025
- 2025Reheated Gradient-based Discrete Sampling for Combinatorial Optimization6 Mar 2025
- 2025Learning Noisy Halfspaces with a Margin: Massart is No Harder than Random16 Jan 2025
38 more
- 2025A Near-optimal Algorithm for Learning Margin Halfspaces with Massart Noise16 Jan 2025
- 2025Testing Noise Assumptions of Learning Algorithms15 Jan 2025
- 2024Sampling from multi-modal distributions with polynomial query complexity in fixed dimension via reverse diffusion31 Dec 2024
- 2024Stability properties of gradient flow dynamics for the symmetric low-rank matrix factorization problem24 Nov 2024
- 2024Provable Accuracy Bounds for Hybrid Dynamical Optimization and Sampling8 Oct 2024
- 2024Randomness Helps Rigor: A Probabilistic Learning Rate Scheduler Bridging Theory and Deep Learning Practice10 Jul 2024
- 2024Langevin Dynamics: A Unified Perspective on Optimization via Lyapunov Potentials5 Jul 2024
- 2024A General Continuous-Time Formulation of Stochastic ADMM and Its Variants22 Apr 2024
- 2024Stable Neural Stochastic Differential Equations in Analyzing Irregular Time Series Data22 Feb 2024
- 2024Accelerating Approximate Thompson Sampling with Underdamped Langevin Monte Carlo22 Jan 2024
- 2024Understanding the Generalization Benefits of Late Learning Rate Decay21 Jan 2024
- 2023Adam-like Algorithm with Smooth Clipping Attains Global Minima: Analysis Based on Ergodicity of Functional SDEs29 Nov 2023
- 2023Quantum Langevin Dynamics for Optimization27 Nov 2023
- 2023Using Stochastic Gradient Descent to Smooth Nonconvex Functions: Analysis of Implicit Graduated Optimization15 Nov 2023
- 2023Communication-Efficient Federated Non-Linear Bandit Optimization3 Nov 2023
- 2023Time-Independent Information-Theoretic Generalization Bounds for SGLD2 Nov 2023
- 2023Stochastic Quantum Sampling for Non-Logconcave Distributions and Estimating Partition Functions17 Oct 2023
- 2023Deep Insights into Noisy Pseudo Labeling on Graph Data2 Oct 2023
- 2023Near-Optimal Bounds for Learning Gaussian Halfspaces with Random Classification Noise13 Jul 2023
- 2023Constructing Semantics-Aware Adversarial Examples with a Probabilistic Perspective1 Jun 2023
- 2023Subsampling Error in Stochastic Gradient Langevin Diffusions23 May 2023
- 2023Implicit Balancing and Regularization: Generalization and Convergence Guarantees for Overparameterized Asymmetric Matrix Sensing24 Mar 2023
- 2023Revisiting the Noise Model of Stochastic Gradient Descent5 Mar 2023
- 2023Efficient displacement convex optimization with particle gradient descent9 Feb 2023
- 2023Improved Langevin Monte Carlo for stochastic optimization via landscape modification8 Feb 2023
- 2023Reweighted Interacting Langevin Diffusions: an Accelerated Sampling Methodfor Optimization30 Jan 2023
- 2022Bayesian autoencoders for data-driven discovery of coordinates, governing equations and fundamental constants19 Nov 2022
- 2022Recycling Scraps: Improving Private Learning by Leveraging Intermediate Checkpoints4 Oct 2022
- 2022Tradeoffs between convergence rate and noise amplification for momentum-based accelerated optimization algorithms24 Sep 2022
- 2022On PAC Learning Halfspaces in Non-interactive Local Privacy Model with Public Unlabeled Data17 Sep 2022
- 2022A sharp uniform-in-time error estimate for Stochastic Gradient Langevin Dynamics19 Jul 2022
- 2022Continuous-time Analysis for Variational Inequalities: An Overview and Desiderata14 Jul 2022
- 2022High-dimensional limit theorems for SGD: Effective dynamics and critical scaling8 Jun 2022
- 2022Generalization Error Bounds for Deep Neural Networks Trained by SGD7 Jun 2022
- 2022Generalization Bounds for Gradient Methods via Discrete and Continuous Prior27 May 2022
- 2022Uniform Generalization Bound on Time and Inverse Temperature for Gradient Descent Algorithm and its Application to Analysis of Simulated Annealing25 May 2022
- 2022Weak Convergence of Approximate reflection coupling and its Application to Non-convex Optimization24 May 2022
- 2022An Empirical Study of the Occurrence of Heavy-Tails in Training a ReLU Gate26 Apr 2022
50 newest of 70 All 70 (JSON)
- 2026On Forgetting and Stability of Score-based Generative models29 Jan 2026
- 2021Small random initialization is akin to spectral learning: Optimization and generalization guarantees for overparameterized low-rank matrix reconstructioncited by 77
- 2022High-dimensional limit theorems for SGD: Effective dynamics and critical scalingcited by 69
- 2023Revisiting the Noise Model of Stochastic Gradient Descentcited by 35
- 2024Stable Neural Stochastic Differential Equations in Analyzing Irregular Time Series Datacited by 35
- 2022Testing distributional assumptions of learning algorithmscited by 30
- 2021Power-law escape rate of SGDcited by 29
- 2021Learning General Halfspaces with General Massart Noise under the Gaussian Distributioncited by 27
- 2022Bayesian autoencoders for data-driven discovery of coordinates, governing equations and fundamental constantscited by 24
- 2021Accelerated replica exchange stochastic gradient Langevin diffusion enhanced Bayesian DeepONet for solving noisy parametric PDEscited by 22
- 2021Improved Algorithms for Efficient Active Learning Halfspaces with Massart and Tsybakov noisecited by 18
- 2021Forster Decomposition and Learning Halfspaces with Noisecited by 16
- 2021Generalization Bounds for Noisy Iterative Algorithms Using Properties of Additive Noise Channelscited by 15
12 most cited of 69 All 69 (JSON)
- 2026Avoiding unsafe sets when training with Langevin Dynamics8 Jul 2026
- 2026When Both Layers Learn: Training Dynamics of Representing Linear Models via ReLU Networks3 Jun 2026
- 2026Testable Learning of General Halfspaces under Massart Noise25 Feb 2026
- 2026On the Superlinear Relationship between SGD Noise Covariance and Loss Landscape Curvature5 Feb 2026
- 2025Smoothed Agnostic Learning of Halfspaces over the Hypercube21 Nov 2025
- 2025A Fully Polynomial-Time Algorithm for Robustly Learning Halfspaces over the Hypercube10 Nov 2025
- 2025From Global to Local: A Scalable Benchmark for Local Posterior Sampling29 Jul 2025
- 2025Continuous Policy and Value Iteration for Relaxed Stochastic Control Problems and Its Convergence9 Jun 2025
- 2025Quantum Speedups for Sampling and Non-convex Optimization with Stochastic Oracles4 Apr 2025
- 2025Reheated Gradient-based Discrete Sampling for Combinatorial Optimization6 Mar 2025
- 2025Learning Noisy Halfspaces with a Margin: Massart is No Harder than Random16 Jan 2025
- 2025A Near-optimal Algorithm for Learning Margin Halfspaces with Massart Noise16 Jan 2025
12 newest of 69 All 69 (JSON)