English

Do Bayesian Neural Networks Need To Be Fully Stochastic?

Machine Learning 2023-02-21 v2 Artificial Intelligence Machine Learning

Abstract

We investigate the benefit of treating all the parameters in a Bayesian neural network stochastically and find compelling theoretical and empirical evidence that this standard construction may be unnecessary. To this end, we prove that expressive predictive distributions require only small amounts of stochasticity. In particular, partially stochastic networks with only nn stochastic biases are universal probabilistic predictors for nn-dimensional predictive problems. In empirical investigations, we find no systematic benefit of full stochasticity across four different inference modalities and eight datasets; partially stochastic networks can match and sometimes even outperform fully stochastic networks, despite their reduced memory costs.

Keywords

Cite

@article{arxiv.2211.06291,
  title  = {Do Bayesian Neural Networks Need To Be Fully Stochastic?},
  author = {Mrinank Sharma and Sebastian Farquhar and Eric Nalisnick and Tom Rainforth},
  journal= {arXiv preprint arXiv:2211.06291},
  year   = {2023}
}

Comments

Published at AISTATS2023 (Oral)

R2 v1 2026-06-28T05:41:01.291Z