Hamiltonian Monte Carlo on ReLU Neural Networks is Inefficient
Vu C. Dinh, Lam Si Tung Ho, Cuong V. Nguyen
Abstract
We analyze the error rates of the Hamiltonian Monte Carlo algorithm with leapfrog integrator for Bayesian neural network inference. We show that due to the non-differentiability of activation functions in the ReLU family, leapfrog HMC for networks with these activation functions has a large local error rate of rather than the classical error rate of . This leads to a higher rejection rate of the proposals, making the method inefficient. We then verify our theoretical findings through empirical simulations as well as experiments on a real-world dataset that highlight the inefficiency of HMC inference on ReLU-based neural networks compared to analytical networks.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Builds on5
- A mathematical model for automatic differentiation in machine learningJérôme Bolte, Edouard PauwelsNeurIPS 2020 · 84 citations
- Numerical influence of ReLU'(0) on backpropagationDavid Bertoin, Jérôme Bolte, Sébastien Gerchinovitz, Edouard PauwelsNeurIPS 2021 · 30 citations
- Mixed Hamiltonian Monte Carlo for Mixed Discrete and Continuous VariablesGuangyao ZhouNeurIPS 2020 · 25 citations
- MCMC Should Mix: Learning Energy-Based Model with Neural Transport Latent Space MCMCErik Nijkamp, Ruiqi Gao, Pavel Sountsov, Srinivas Vasudevan et al.ICLR 2022 · 25 citations
- A Gradient Based Strategy for Hamiltonian Monte Carlo Hyperparameter OptimizationAndrew Campbell, Wenlong Chen, Vincent Stimper, José Miguel Hernández-Lobato et al.ICML 2021 · 20 citations
Related papers
- On the Expressiveness of Approximate Inference in Bayesian Neural NetworksAndrew Y. K. Foong, David R. Burt, Yingzhen Li, Richard E. TurnerNeurIPS 2020 · 142 citations
- Structured Stochastic Gradient MCMCAntonios Alexos, Alex J. Boyd, Stephan MandtICML 2022 · 14 citations
- What Are Bayesian Neural Network Posteriors Really Like?Pavel Izmailov, Sharad Vikram, Matthew D. Hoffman, Andrew Gordon WilsonICML 2021 · 458 citations
- Liberty or Depth: Deep Bayesian Neural Nets Do Not Need Complex Weight Posterior ApproximationsSebastian Farquhar, Lewis Smith, Yarin GalNeurIPS 2020 · 47 citations
- Posterior Refinement Improves Sample Efficiency in Bayesian Neural NetworksAgustinus Kristiadi, Runa Eschenhagen, Philipp HennigNeurIPS 2022 · 17 citations
