English

Hamiltonian Monte Carlo on ReLU Neural Networks is Inefficient

Machine Learning 2024-10-30 v1 Machine Learning

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 Ω(ϵ)\Omega(\epsilon) rather than the classical error rate of O(ϵ3)O(\epsilon^3). 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.

Cite

@article{arxiv.2410.22065,
  title  = {Hamiltonian Monte Carlo on ReLU Neural Networks is Inefficient},
  author = {Vu C. Dinh and Lam Si Tung Ho and Cuong V. Nguyen},
  journal= {arXiv preprint arXiv:2410.22065},
  year   = {2024}
}

Comments

Paper published at NeurIPS 2024