Cyclical Stochastic Gradient MCMC for Bayesian Deep Learning
Not on the map. Why?
Abstract (arXiv)
The posteriors over neural network weights are high dimensional and multimodal. Each mode typically characterizes a meaningfully different representation of the data. We develop Cyclical Stochastic Gradient MCMC (SG-MCMC) to automatically explore such distributions. In particular, we propose a cyclical stepsize schedule, where larger steps discover new modes, and smaller steps characterize each mode. We also prove non-asymptotic convergence of our proposed algorithm. Moreover, we provide extensive experimental results, including ImageNet, to demonstrate the scalability and effectiveness of cyclical SG-MCMC in learning complex multimodal distributions, especially for fully Bayesian inference with modern deep neural networks.
Key facts. Cited by 203 papers in the map (data through 14 Oct 2026). First posted on arXiv 11 Feb 2019.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 44; 2022: 46; 2023: 41; 2024: 38; 2025: 28; 2026: 6.
203 papers.
Sort:
By area:
- 2021Laplace Redux -- Effortless Bayesian Deep Learningcited by 365
- 2021What Are Bayesian Neural Network Posteriors Really Like?cited by 308
- 2023Towards Building the Federated GPT: Federated Instruction Tuningcited by 185
- 2021Bayesian Neural Network Priors Revisitedcited by 121
- 2023Bayesian Low-rank Adaptation for Large Language Modelscited by 115
- 2021DEUP: Direct Epistemic Uncertainty Predictioncited by 104
- 2021Repulsive Deep Ensembles are Bayesiancited by 104
- 2021Priors in Bayesian Deep Learning: A Reviewcited by 101
- 2021Learning Neural Network Subspacescited by 82
- 2021The Promises and Pitfalls of Deep Kernel Learningcited by 81
- 2023LoRA ensembles for large language model fine-tuningcited by 67
- 2024Position: Bayesian Deep Learning is Needed in the Age of Large-Scale AIcited by 62
38 more
- 2022Assaying Out-Of-Distribution Generalization in Transfer Learningcited by 53
- 2024BLoB: Bayesian Low-Rank Adaptation by Backpropagation for Large Language Modelscited by 50
- 2021DICE: Diversity in Deep Ensembles via Conditional Redundancy Adversarial Estimationcited by 48
- 2022On Uncertainty, Tempering, and Data Augmentation in Bayesian Classificationcited by 46
- 2022A Langevin-like Sampler for Discrete Distributionscited by 44
- 2022Pre-Train Your Loss: Easy Bayesian Transfer Learning with Informative Priorscited by 43
- 2024Uncertainty quantification in fine-tuned LLMs using LoRA ensemblescited by 37
- 2021Data augmentation in Bayesian neural networks and the cold posterior effectcited by 36
- 2023BayesDiff: Estimating Pixel-wise Uncertainty in Diffusion via Bayesian Inferencecited by 35
- 2024Bayesian Reward Models for LLM Alignmentcited by 35
- 2021MixMo: Mixing Multiple Inputs for Multiple Outputs via Deep Subnetworkscited by 32
- 2021Diversity and Generalization in Neural Network Ensemblescited by 32
- 2022Fed-CBS: A Heterogeneity-Aware Client Sampling Mechanism for Federated Learning via Class-Imbalance Reductioncited by 32
- 2023Making Substitute Models More Bayesian Can Enhance Transferability of Adversarial Examplescited by 32
- 2021Exact Langevin Dynamics with Stochastic Gradientscited by 30
- 2022Accelerated Linearized Laplace Approximation for Bayesian Deep Learningcited by 28
- 2023Quantification of Uncertainty with Adversarial Modelscited by 28
- 2023Introducing an Improved Information-Theoretic Measure of Predictive Uncertaintycited by 28
- 2022RegMixup: Mixup as a Regularizer Can Surprisingly Improve Accuracy and Out Distribution Robustnesscited by 27
- 2023A Primer on Bayesian Neural Networks: Review and Debatescited by 27
- 2021Annealed Stein Variational Gradient Descentcited by 26
- 2021Mixtures of Laplace Approximations for Improved Post-Hoc Uncertainty in Deep Learningcited by 25
- 2022Markov Chain Monte Carlo-Based Machine Unlearning: Unlearning What Needs to be Forgottencited by 25
- 2022Approximate Bayesian Neural Operators: Uncertainty Quantification for Parametric PDEscited by 25
- 2021A Survey on Epistemic (Model) Uncertainty in Supervised Learning: Recent Advances and Applicationscited by 24
- 2022Bayesian autoencoders for data-driven discovery of coordinates, governing equations and fundamental constantscited by 24
- 2021Disentangling the Roles of Curation, Data-Augmentation and the Prior in the Cold Posterior Effectcited by 22
- 2021On Stein Variational Neural Network Ensemblescited by 21
- 2022Knowledge Removal in Sampling-based Bayesian Inferencecited by 21
- 2023Calibrating Transformers via Sparse Gaussian Processescited by 21
- 2021BNNpriors: A library for Bayesian neural network inference with different prior distributionscited by 20
- 2024Gaussian Stochastic Weight Averaging for Bayesian Low-Rank Adaptation of Large Language Modelscited by 20
- 2021Sparse MoEs meet Efficient Ensemblescited by 19
- 2021Bayesian OOD detection with aleatoric uncertainty and outlier exposurecited by 16
- 2021Sparse Uncertainty Representation in Deep Learning with Inducing Weightscited by 16
- 2021When in Doubt: Neural Non-Parametric Uncertainty Quantification for Epidemic Forecastingcited by 16
- 2024On Information-Theoretic Measures of Predictive Uncertaintycited by 16
- 2024SLED: Self Logits Evolution Decoding for Improving Factuality in Large Language Modelscited by 16
50 most cited of 203 All 203 (JSON)
- 2026Guiding Posterior Exploration with Optimizer-Derived Geometry28 Jul 2026
- 2026Uncertainty Decomposition via Cyclical SG-MCMC and Soft-label Learning for Subjective NLP23 May 2026
- 2026Position: The Time for Sampling Is Now! Charting a New Course for Bayesian Deep Learning20 May 2026
- 2026Slithering Through Gaps: Capturing Discrete Isolated Modes via Logistic Bridging12 Apr 2026
- 2026On the Interplay of Priors and Overparametrization in Bayesian Neural Network Posteriors23 Mar 2026
- 2026Can Microcanonical Langevin Dynamics Leverage Mini-Batch Gradient Noise?6 Feb 2026
- 2025One-Step Diffusion Samplers via Self-Distillation and Deterministic Flow4 Dec 2025
- 2025ReBaPL: Repulsive Bayesian Prompt Learning21 Nov 2025
- 2025Bayesian neural networks with interpretable priors from Mercer kernels27 Oct 2025
- 2025Single-Teacher View Augmentation: Boosting Knowledge Distillation via Angular Diversity26 Oct 2025
- 2025Two Birds with One Stone: Enhancing Uncertainty Quantification and Interpretability with Graph Functional Neural Process23 Aug 2025
- 2025Accelerating Hamiltonian Monte Carlo for Bayesian Inference in Neural Networks and Neural Operators19 Jul 2025
38 more
- 2025Last Layer Hamiltonian Monte Carlo11 Jul 2025
- 2025ConBatch-BAL: Batch Bayesian Active Learning under Budget Constraints7 Jul 2025
- 2025A Novel Active Learning Approach to Label One Million Unknown Malware Variants30 Jun 2025
- 2025Stochastic Weight Sharing for Bayesian Neural Networks23 May 2025
- 2025C-LoRA: Contextual Low-Rank Adaptation for Uncertainty Estimation in Large Language Models23 May 2025
- 2025Last Layer Empirical Bayes21 May 2025
- 2025JaxSGMC: Modular stochastic gradient MCMC in JAX16 May 2025
- 2025A Langevin sampling algorithm inspired by the Adam optimizer26 Apr 2025
- 2025Comparing Uncertainty Measurement and Mitigation Methods for Large Language Models: A Systematic Review25 Apr 2025
- 2025Conformalized Generative Bayesian Imaging: An Uncertainty Quantification Framework for Computational Imaging10 Apr 2025
- 2025Temporal Score Analysis for Understanding and Correcting Diffusion Artifacts20 Mar 2025
- 2025Keeping Yourself is Important in Downstream Tuning Multimodal Large Language Model6 Mar 2025
- 2025Reheated Gradient-based Discrete Sampling for Combinatorial Optimization6 Mar 2025
- 2025BLADE: Bayesian Langevin Active Discovery with Replica Exchange for Identification of Complex Systems4 Mar 2025
- 2025Parameter Expanded Stochastic Gradient Markov Chain Monte Carlo2 Mar 2025
- 2025Finer Disentanglement of Aleatoric Uncertainty Can Accelerate Chemical Histopathology Imaging27 Feb 2025
- 2025Enhancing Gradient-based Discrete Sampling via Parallel Tempering26 Feb 2025
- 2025Bayesian Computation in Deep Learning25 Feb 2025
- 2025Microcanonical Langevin Ensembles: Advancing the Sampling of Bayesian Neural Networks10 Feb 2025
- 2025Active Learning with Bayesian Multi-Fidelity Laplace Neural Operators for Oscillatory Parametric PDEs1 Feb 2025
- 2025Exploring Non-Convex Discrete Energy Landscapes: An Efficient Langevin-Like Sampler with Replica Exchange28 Jan 2025
- 2025CENSOR: Defense Against Gradient Inversion via Orthogonal Subspace Bayesian Sampling27 Jan 2025
- 2024Quantile deep learning models for multi-step ahead time series prediction24 Nov 2024
- 2024Ex Uno Pluria: Insights on Ensembling in Low Precision Number Systems22 Nov 2024
- 2024SLED: Self Logits Evolution Decoding for Improving Factuality in Large Language Models1 Nov 2024
- 2024Minimax optimality of deep neural networks on dependent data via PAC-Bayes bounds29 Oct 2024
- 2024Sampling from Bayesian Neural Network Posteriors with Symmetric Minibatch Splitting Langevin Dynamics14 Oct 2024
- 2024On Information-Theoretic Measures of Predictive Uncertainty14 Oct 2024
- 2024Provable Convergence and Limitations of Geometric Tempering for Langevin Dynamics13 Oct 2024
- 2024Temperature Optimization for Bayesian Deep Learning8 Oct 2024
- 2024Functional Stochastic Gradient MCMC for Bayesian Neural Networks25 Sep 2024
- 2024Taylor-Sensus Network: Embracing Noise to Enlighten Uncertainty for Scientific Data12 Sep 2024
- 2024MLLM-LLaVA-FL: Multimodal Large Language Model Assisted Federated Learning9 Sep 2024
- 2024Learning to Explore for Stochastic Gradient MCMC17 Aug 2024
- 2024A Practical Diffusion Path for Sampling20 Jun 2024
- 2024ARTIST: Improving the Generation of Text-rich Images with Disentangled Diffusion Models and Large Language Models17 Jun 2024
- 2024BLoB: Bayesian Low-Rank Adaptation by Backpropagation for Large Language Models17 Jun 2024
- 2024On Uniform, Bayesian, and PAC-Bayesian Deep Ensembles8 Jun 2024
50 newest of 203 All 203 (JSON)
- 2023BayesDiff: Estimating Pixel-wise Uncertainty in Diffusion via Bayesian Inferencecited by 35
- 2022Personalizing Text-to-Image Generation via Aesthetic Gradientscited by 13
- 2025Temporal Score Analysis for Understanding and Correcting Diffusion Artifactscited by 9
- 2024A Practical Diffusion Path for Samplingcited by 5
- 2024ARTIST: Improving the Generation of Text-rich Images with Disentangled Diffusion Models and Large Language Modelscited by 3
- 2025One-Step Diffusion Samplers via Self-Distillation and Deterministic Flowcited by 1
- 2025One-Step Diffusion Samplers via Self-Distillation and Deterministic Flow4 Dec 2025
- 2025Temporal Score Analysis for Understanding and Correcting Diffusion Artifacts20 Mar 2025
- 2024A Practical Diffusion Path for Sampling20 Jun 2024
- 2024ARTIST: Improving the Generation of Text-rich Images with Disentangled Diffusion Models and Large Language Models17 Jun 2024
- 2023BayesDiff: Estimating Pixel-wise Uncertainty in Diffusion via Bayesian Inference17 Oct 2023
- 2022Personalizing Text-to-Image Generation via Aesthetic Gradients25 Sep 2022
- 2024Bayesian Reward Models for LLM Alignmentcited by 35
- 2024Bayesian Reward Models for LLM Alignment20 Feb 2024
- 2024Fast Value Tracking for Deep Reinforcement Learningcited by 2
- 2024Fast Value Tracking for Deep Reinforcement Learning19 Mar 2024
- 2021Laplace Redux -- Effortless Bayesian Deep Learningcited by 365
- 2021What Are Bayesian Neural Network Posteriors Really Like?cited by 308
- 2023Towards Building the Federated GPT: Federated Instruction Tuningcited by 185
- 2021Bayesian Neural Network Priors Revisitedcited by 121
- 2023Bayesian Low-rank Adaptation for Large Language Modelscited by 115
- 2021DEUP: Direct Epistemic Uncertainty Predictioncited by 104
- 2021Repulsive Deep Ensembles are Bayesiancited by 104
- 2021Priors in Bayesian Deep Learning: A Reviewcited by 101
- 2021Learning Neural Network Subspacescited by 82
- 2021The Promises and Pitfalls of Deep Kernel Learningcited by 81
- 2023LoRA ensembles for large language model fine-tuningcited by 67
- 2024Position: Bayesian Deep Learning is Needed in the Age of Large-Scale AIcited by 62
12 most cited of 188 All 188 (JSON)
- 2026Guiding Posterior Exploration with Optimizer-Derived Geometry28 Jul 2026
- 2026Uncertainty Decomposition via Cyclical SG-MCMC and Soft-label Learning for Subjective NLP23 May 2026
- 2026Position: The Time for Sampling Is Now! Charting a New Course for Bayesian Deep Learning20 May 2026
- 2026Slithering Through Gaps: Capturing Discrete Isolated Modes via Logistic Bridging12 Apr 2026
- 2026On the Interplay of Priors and Overparametrization in Bayesian Neural Network Posteriors23 Mar 2026
- 2026Can Microcanonical Langevin Dynamics Leverage Mini-Batch Gradient Noise?6 Feb 2026
- 2025ReBaPL: Repulsive Bayesian Prompt Learning21 Nov 2025
- 2025Bayesian neural networks with interpretable priors from Mercer kernels27 Oct 2025
- 2025Two Birds with One Stone: Enhancing Uncertainty Quantification and Interpretability with Graph Functional Neural Process23 Aug 2025
- 2025Accelerating Hamiltonian Monte Carlo for Bayesian Inference in Neural Networks and Neural Operators19 Jul 2025
- 2025Last Layer Hamiltonian Monte Carlo11 Jul 2025
- 2025ConBatch-BAL: Batch Bayesian Active Learning under Budget Constraints7 Jul 2025
12 newest of 188 All 188 (JSON)