Preconditioned Stochastic Gradient Langevin Dynamics for Deep Neural Networks
Not on the map. Why?
Abstract (arXiv)
Effective training of deep neural networks suffers from two main issues. The first is that the parameter spaces of these models exhibit pathological curvature. Recent methods address this problem by using adaptive preconditioning for Stochastic Gradient Descent (SGD). These methods improve convergence by adapting to the local geometry of parameter space. A second issue is overfitting, which is typically addressed by early stopping. However, recent work has demonstrated that Bayesian model averaging mitigates this problem. The posterior can be sampled by using Stochastic Gradient Langevin Dynamics (SGLD). However, the rapidly changing curvature renders default SGLD methods inefficient. Here, we propose combining adaptive preconditioners with SGLD. In support of this idea, we give theoretical properties on asymptotic convergence and predictive risk. We also provide empirical results for Logistic Regression, Feedforward Neural Nets, and Convolutional Neural Nets, demonstrating that our preconditioned SGLD method gives state-of-the-art performance on these models.
Key facts. Cited by 121 papers in the map (data through 14 Oct 2026). First posted on arXiv 23 Dec 2015.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 15; 2022: 24; 2023: 26; 2024: 18; 2025: 20; 2026: 18.
121 papers.
Sort:
By area:
- 2021A Survey of Uncertainty in Deep Neural Networkscited by 680
- 2022Uncertainty Quantification in Scientific Machine Learning: Methods, Metrics, and Comparisonscited by 170
- 2022A Langevin-like Sampler for Discrete Distributionscited by 44
- 2023BayesDAG: Gradient-Based Posterior Inference for Causal Discoverycited by 33
- 2023Provable and Practical: Efficient Exploration in Reinforcement Learning via Langevin Monte Carlocited by 29
- 2023Quantification of Uncertainty with Adversarial Modelscited by 28
- 2023A Primer on Bayesian Neural Networks: Review and Debatescited by 27
- 2021Contextual Dropout: An Efficient Sample-Dependent Dropout Modulecited by 26
- 2022Reliable amortized variational inference with physics-based latent distribution correctioncited by 25
- 2021Disentangling the Roles of Curation, Data-Augmentation and the Prior in the Cold Posterior Effectcited by 22
- 2022Particle-based Variational Inference with Preconditioned Functional Gradient Flowcited by 20
- 2022Accelerating Score-based Generative Models with Preconditioned Diffusion Samplingcited by 19
38 more
- 2021Sampling with Mirrored Stein Operatorscited by 17
- 2024Reparameterization invariance in approximate Bayesian inferencecited by 17
- 2021When in Doubt: Neural Non-Parametric Uncertainty Quantification for Epidemic Forecastingcited by 16
- 2021Generalization Bounds for Noisy Iterative Algorithms Using Properties of Additive Noise Channelscited by 15
- 2025Stochastic Process Learning via Operator Flow Matchingcited by 14
- 2025Bayesian Influence Functions for Hessian-Free Data Attributioncited by 14
- 2022How Tempering Fixes Data Augmentation in Bayesian Neural Networkscited by 13
- 2024Scalable Bayesian Learning with posteriorscited by 13
- 2021Structured Stochastic Gradient MCMCcited by 12
- 2021Deep Bayesian inference for seismic imaging with taskscited by 12
- 2022The Sufficiency of Off-Policyness and Soft Clipping: PPO is still Insufficient according to an Off-Policy Measurecited by 12
- 2021Efficient and Generalizable Tuning Strategies for Stochastic Gradient MCMCcited by 11
- 2022Scalable Bayesian Uncertainty Quantification for Neural Network Potentials: Promise and Pitfallscited by 11
- 2023Tuning-Free Maximum Likelihood Training of Latent Variable Models via Coin Bettingcited by 11
- 2024Gradient Methods with Online Scalingcited by 11
- 2022Interacting Contour Stochastic Gradient Langevin Dynamicscited by 9
- 2023Scalable Stochastic Gradient Riemannian Langevin Dynamics in Non-Diagonal Metricscited by 8
- 2022Non-reversible Parallel Tempering for Deep Posterior Approximationcited by 7
- 2022PAC-Bayes Bounds for Bandit Problems: A Survey and Experimental Comparisoncited by 7
- 2023Inferring networks from time series: a neural approachcited by 7
- 2024Taming Score-Based Diffusion Priors for Infinite-Dimensional Nonlinear Inverse Problemscited by 7
- 2024Particle Semi-Implicit Variational Inferencecited by 7
- 2025Influence Dynamics and Stagewise Data Attributioncited by 7
- 2026Can Microcanonical Langevin Dynamics Leverage Mini-Batch Gradient Noise?cited by 7
- 2021Differentially Private Bayesian Neural Networks on Accuracy, Privacy and Reliabilitycited by 6
- 2022DiSECt: A Differentiable Simulator for Parameter Inference and Control in Robotic Cuttingcited by 6
- 2022Effects of Spectral Normalization in Multi-agent Reinforcement Learningcited by 6
- 2023Preconditioned Score-based Generative Modelscited by 6
- 2023Fit Like You Sample: Sample-Efficient Generalized Score Matching from Fast Mixing Diffusionscited by 6
- 2026D-Flow SGLD: Source-Space Posterior Sampling for Scientific Inverse Problems with Flow Matchingcited by 6
- 2021Batch Normalization Preconditioning for Neural Network Trainingcited by 5
- 2025From Global to Local: A Scalable Benchmark for Local Posterior Samplingcited by 5
- 2025Compressibility Measures Complexity: Minimum Description Length Meets Singular Learning Theorycited by 5
- 2022PAC-Bayesian Lifelong Learning For Multi-Armed Banditscited by 4
- 2022A Survey on Uncertainty Toolkits for Deep Learningcited by 4
- 2023Particle Mean Field Variational Bayescited by 4
- 2023Non-Log-Concave and Nonsmooth Sampling via Langevin Monte Carlo Algorithmscited by 4
- 2024NPB-REC: A Non-parametric Bayesian Deep-learning Approach for Undersampled MRI Reconstruction with Uncertainty Estimationcited by 4
50 most cited of 121 All 121 (JSON)
- 2026Patterning in Practice: Debiasing Reward Models with Susceptibilities1 Sep 2026
- 2026ReFP-AD: Rectified Flow Preconditioning for Energy-Based Anomaly Detection3 Aug 2026
- 2026Guiding Posterior Exploration with Optimizer-Derived Geometry28 Jul 2026
- 2026DP-IVON-Gradsq: Differentially Private Squared-Gradient Improved Variational Online Newton26 Jul 2026
- 2026Adaptive Runge-Kutta Step Control Buys Training Loss, Not Generalization: An Honest Compute-Matched Study of RK-Adam Optimizers16 Jul 2026
- 2026Score-Based Martingale Posteriors for Deep Neural Networks14 Jun 2026
- 2026Deterministic Denominator Design for Localized Tamed Stochastic Gradient Langevin Dynamics9 Jun 2026
- 2026Towards Automated Discovery: A Review of Generative Models, Multimodal Learning and Closed-Loop Workflows in Inverse Materials Design1 Jun 2026
- 2026Accurate Large-sample Uncertainty Quantification using Stochastic Gradient Markov Chain Monte Carlo29 May 2026
- 2026SparseOpt: Addressing Normalization-induced Gradient Skew in Sparse Training26 May 2026
- 2026Mechanistic Anomaly Detection via Functional Attribution21 Apr 2026
- 2026Bayesian Generative Adversarial Networks via Gaussian Approximation for Tabular Data Synthesis25 Feb 2026
38 more
- 2026D-Flow SGLD: Source-Space Posterior Sampling for Scientific Inverse Problems with Flow Matching25 Feb 2026
- 2026Robust Stochastic Gradient Posterior Sampling with Lattice Based Discretisation17 Feb 2026
- 2026Can Microcanonical Langevin Dynamics Leverage Mini-Batch Gradient Noise?6 Feb 2026
- 2026Safe Langevin Soft Actor Critic31 Jan 2026
- 2026Sampling via Stochastic Interpolants by Langevin-based Velocity and Initialization Estimation in Flow ODEs13 Jan 2026
- 2026Stagewise Reinforcement Learning and the Geometry of the Regret Landscape12 Jan 2026
- 2025Adaptive Stepsizing for Stochastic Gradient Langevin Dynamics in Bayesian Neural Networks11 Nov 2025
- 2025Compressibility Measures Complexity: Minimum Description Length Meets Singular Learning Theory14 Oct 2025
- 2025Influence Dynamics and Stagewise Data Attribution14 Oct 2025
- 2025Flatness-Aware Stochastic Gradient Langevin Dynamics2 Oct 2025
- 2025Bayesian Influence Functions for Hessian-Free Data Attribution30 Sep 2025
- 2025Two Birds with One Stone: Enhancing Uncertainty Quantification and Interpretability with Graph Functional Neural Process23 Aug 2025
- 2025Fast, Convex and Conditioned Network for Multi-Fidelity Vectors and Stiff Univariate Differential Equations8 Aug 2025
- 2025From Global to Local: A Scalable Benchmark for Local Posterior Sampling29 Jul 2025
- 2025Feel-Good Thompson Sampling for Contextual Bandits: a Markov Chain Monte Carlo Showdown21 Jul 2025
- 2025Single- to multi-fidelity history-dependent learning with uncertainty quantification and disentanglement: application to data-driven constitutive modeling17 Jul 2025
- 2025Quantifying Uncertainty in the Presence of Distribution Shifts23 Jun 2025
- 2025Conformal Object Detection by Sequential Risk Control29 May 2025
- 2025Enhancing Uncertainty Estimation and Interpretability via Bayesian Non-negative Decision Layer28 May 2025
- 2025Training Latent Diffusion Models with Interacting Particle Algorithms18 May 2025
- 2025JaxSGMC: Modular stochastic gradient MCMC in JAX16 May 2025
- 2025Cooperative Variance Estimation and Bayesian Neural Networks for Disentangling Aleatoric and Epistemic Uncertainties5 May 2025
- 2025Parameter Expanded Stochastic Gradient Markov Chain Monte Carlo2 Mar 2025
- 2025Bayesian Computation in Deep Learning25 Feb 2025
- 2025Active Learning with Bayesian Multi-Fidelity Laplace Neural Operators for Oscillatory Parametric PDEs1 Feb 2025
- 2025Stochastic Process Learning via Operator Flow Matching7 Jan 2025
- 2024BayesNAM: Leveraging Inconsistency for Reliable Explanations10 Nov 2024
- 2024Gradient Methods with Online Scaling4 Nov 2024
- 2024Dynamic neuron approach to deep neural networks: Decoupling neurons for renormalization group analysis1 Oct 2024
- 2024Functional Stochastic Gradient MCMC for Bayesian Neural Networks25 Sep 2024
- 2024Learning to Explore for Stochastic Gradient MCMC17 Aug 2024
- 2024Practical multi-fidelity machine learning: fusion of deterministic and Bayesian models21 Jul 2024
- 2024Particle Semi-Implicit Variational Inference30 Jun 2024
- 2024On Uniform, Bayesian, and PAC-Bayesian Deep Ensembles8 Jun 2024
- 2024Reparameterization invariance in approximate Bayesian inference5 Jun 2024
- 2024Scalable Bayesian Learning with posteriors31 May 2024
- 2024Taming Score-Based Diffusion Priors for Infinite-Dimensional Nonlinear Inverse Problems24 May 2024
- 2024NPB-REC: A Non-parametric Bayesian Deep-learning Approach for Undersampled MRI Reconstruction with Uncertainty Estimation6 Apr 2024
50 newest of 121 All 121 (JSON)
- 2022Accelerating Score-based Generative Models with Preconditioned Diffusion Samplingcited by 19
- 2025Stochastic Process Learning via Operator Flow Matchingcited by 14
- 2024Taming Score-Based Diffusion Priors for Infinite-Dimensional Nonlinear Inverse Problemscited by 7
- 2023Preconditioned Score-based Generative Modelscited by 6
- 2026D-Flow SGLD: Source-Space Posterior Sampling for Scientific Inverse Problems with Flow Matchingcited by 6
- 2025Training Latent Diffusion Models with Interacting Particle Algorithmscited by 4
- 2026ReFP-AD: Rectified Flow Preconditioning for Energy-Based Anomaly Detectioncited by 0
- 2026ReFP-AD: Rectified Flow Preconditioning for Energy-Based Anomaly Detection3 Aug 2026
- 2026D-Flow SGLD: Source-Space Posterior Sampling for Scientific Inverse Problems with Flow Matching25 Feb 2026
- 2025Training Latent Diffusion Models with Interacting Particle Algorithms18 May 2025
- 2025Stochastic Process Learning via Operator Flow Matching7 Jan 2025
- 2024Taming Score-Based Diffusion Priors for Infinite-Dimensional Nonlinear Inverse Problems24 May 2024
- 2023Preconditioned Score-based Generative Models13 Feb 2023
- 2022Accelerating Score-based Generative Models with Preconditioned Diffusion Sampling5 Jul 2022
- 2026Stagewise Reinforcement Learning and the Geometry of the Regret Landscapecited by 4
- 2024Fast Value Tracking for Deep Reinforcement Learningcited by 2
- 2026Safe Langevin Soft Actor Criticcited by 1
- 2026Safe Langevin Soft Actor Critic31 Jan 2026
- 2026Stagewise Reinforcement Learning and the Geometry of the Regret Landscape12 Jan 2026
- 2024Fast Value Tracking for Deep Reinforcement Learning19 Mar 2024
- 2023Gibbs Sampling the Posterior of Neural Networkscited by 0
- 2023Gibbs Sampling the Posterior of Neural Networks5 Jun 2023
- 2026Mechanistic Anomaly Detection via Functional Attributioncited by 1
- 2026Mechanistic Anomaly Detection via Functional Attribution21 Apr 2026
- 2021A Survey of Uncertainty in Deep Neural Networkscited by 680
- 2022Uncertainty Quantification in Scientific Machine Learning: Methods, Metrics, and Comparisonscited by 170
- 2022A Langevin-like Sampler for Discrete Distributionscited by 44
- 2023BayesDAG: Gradient-Based Posterior Inference for Causal Discoverycited by 33
- 2023Provable and Practical: Efficient Exploration in Reinforcement Learning via Langevin Monte Carlocited by 29
- 2023Quantification of Uncertainty with Adversarial Modelscited by 28
- 2023A Primer on Bayesian Neural Networks: Review and Debatescited by 27
- 2021Contextual Dropout: An Efficient Sample-Dependent Dropout Modulecited by 26
- 2022Reliable amortized variational inference with physics-based latent distribution correctioncited by 25
- 2021Disentangling the Roles of Curation, Data-Augmentation and the Prior in the Cold Posterior Effectcited by 22
- 2022Particle-based Variational Inference with Preconditioned Functional Gradient Flowcited by 20
- 2021Sampling with Mirrored Stein Operatorscited by 17
12 most cited of 108 All 108 (JSON)
- 2026Guiding Posterior Exploration with Optimizer-Derived Geometry28 Jul 2026
- 2026DP-IVON-Gradsq: Differentially Private Squared-Gradient Improved Variational Online Newton26 Jul 2026
- 2026Adaptive Runge-Kutta Step Control Buys Training Loss, Not Generalization: An Honest Compute-Matched Study of RK-Adam Optimizers16 Jul 2026
- 2026Score-Based Martingale Posteriors for Deep Neural Networks14 Jun 2026
- 2026Deterministic Denominator Design for Localized Tamed Stochastic Gradient Langevin Dynamics9 Jun 2026
- 2026Towards Automated Discovery: A Review of Generative Models, Multimodal Learning and Closed-Loop Workflows in Inverse Materials Design1 Jun 2026
- 2026Accurate Large-sample Uncertainty Quantification using Stochastic Gradient Markov Chain Monte Carlo29 May 2026
- 2026SparseOpt: Addressing Normalization-induced Gradient Skew in Sparse Training26 May 2026
- 2026Bayesian Generative Adversarial Networks via Gaussian Approximation for Tabular Data Synthesis25 Feb 2026
- 2026Robust Stochastic Gradient Posterior Sampling with Lattice Based Discretisation17 Feb 2026
- 2026Can Microcanonical Langevin Dynamics Leverage Mini-Batch Gradient Noise?6 Feb 2026
- 2026Sampling via Stochastic Interpolants by Langevin-based Velocity and Initialization Estimation in Flow ODEs13 Jan 2026
12 newest of 108 All 108 (JSON)