English

Wide Mean-Field Variational Bayesian Neural Networks Ignore the Data

Machine Learning 2021-06-15 v1 Machine Learning

Abstract

Variational inference enables approximate posterior inference of the highly over-parameterized neural networks that are popular in modern machine learning. Unfortunately, such posteriors are known to exhibit various pathological behaviors. We prove that as the number of hidden units in a single-layer Bayesian neural network tends to infinity, the function-space posterior mean under mean-field variational inference actually converges to zero, completely ignoring the data. This is in contrast to the true posterior, which converges to a Gaussian process. Our work provides insight into the over-regularization of the KL divergence in variational inference.

Keywords

Cite

@article{arxiv.2106.07052,
  title  = {Wide Mean-Field Variational Bayesian Neural Networks Ignore the Data},
  author = {Beau Coker and Weiwei Pan and Finale Doshi-Velez},
  journal= {arXiv preprint arXiv:2106.07052},
  year   = {2021}
}
R2 v1 2026-06-24T03:08:58.137Z