English

Structured Partial Stochasticity in Bayesian Neural Networks

Machine Learning 2024-07-03 v2 Machine Learning

Abstract

Bayesian neural network posterior distributions have a great number of modes that correspond to the same network function. The abundance of such modes can make it difficult for approximate inference methods to do their job. Recent work has demonstrated the benefits of partial stochasticity for approximate inference in Bayesian neural networks; inference can be less costly and performance can sometimes be improved. I propose a structured way to select the deterministic subset of weights that removes neuron permutation symmetries, and therefore the corresponding redundant posterior modes. With a drastically simplified posterior distribution, the performance of existing approximate inference schemes is found to be greatly improved.

Keywords

Cite

@article{arxiv.2405.17666,
  title  = {Structured Partial Stochasticity in Bayesian Neural Networks},
  author = {Tommy Rochussen},
  journal= {arXiv preprint arXiv:2405.17666},
  year   = {2024}
}

Comments

Accepted at 6th Symposium on Advances in Approximate Bayesian Inference (non-archival track)

R2 v1 2026-06-28T16:42:57.761Z