Scaling Hamiltonian Monte Carlo Inference for Bayesian Neural Networks with Symmetric Splitting

Abstract

Hamiltonian Monte Carlo (HMC) is a Markov chain Monte Carlo (MCMC) approach that exhibits favourable exploration properties in high-dimensional models such as neural networks. Unfortunately, HMC has limited use in large-data regimes and little work has explored suitable approaches that aim to preserve the entire Hamiltonian. In our work, we introduce a new symmetric integration scheme for split HMC that does not rely on stochastic gradients. We show that our new formulation is more efficient than previous approaches and is easy to implement with a single GPU. As a result, we are able to perform full HMC over common deep learning architectures using entire data sets. In addition, when we compare with stochastic gradient MCMC, we show that our method achieves better performance in both accuracy and uncertainty quantification. Our approach demonstrates HMC as a feasible option when considering inference schemes for large-scale machine learning problems.

arXiv comment11 pages, 13 figures

References in corpus (13)

• Scikit-learn: Machine Learning in Python
• Fashion-MNIST: a Novel Image Dataset for Benchmarking Machine Learning Algorithms
• Weight Uncertainty in Neural Networks
• GPyTorch: Blackbox Matrix-Matrix Gaussian Process Inference with GPU Acceleration
• Bayesian Active Learning for Classification and Preference Learning
• Stochastic Gradient Hamiltonian Monte Carlo
• Cyclical Stochastic Gradient MCMC for Bayesian Deep Learning
• NeuTra-lizing Bad Geometry in Hamiltonian Monte Carlo Using Neural Transport
• Can Autonomous Vehicles Identify, Recover From, and Adapt to Distribution Shifts?
• Generalizing the No-U-Turn Sampler to Riemannian Manifolds
• Bayesian Inference for Large Scale Image Classification
• Semi-Separable Hamiltonian Monte Carlo for Inference in Bayesian Hierarchical Models
• URSABench: Comprehensive Benchmarking of Approximate Bayesian Inference Methods for Deep Neural Networks

Cited by in corpus (7)

• A Review of Uncertainty Quantification in Deep Learning: Techniques, Applications and Challenges
• What Are Bayesian Neural Network Posteriors Really Like?
• Bayesian graph convolutional neural networks via tempered MCMC
• Revisiting Bayesian Autoencoders with MCMC
• On Sequential Bayesian Inference for Continual Learning
• Flow field tomography with uncertainty quantification using a Bayesian physics-informed neural network
• L2M: Practical posterior Laplace approximation with optimization-driven second moment estimation