Gaussian Mean Field Variational Inference can Overestimate Predictive Variance
Not on the map. Why?
Abstract (arXiv)
Mean Field Variational Inference (MFVI) is widely understood to underestimate posterior variance. By analysing conjugate Bayesian Linear Regression (BLR), we show that this characterization is incomplete: while MFVI underestimates the variance in parameter space, it can overestimate the predictive variance compared to the exact posterior. We show that if the MFVI posterior underestimates predictive variances in some directions, it necessarily overestimates them in others. Crucially, this overestimation occurs in directions where the training data concentrates. This leads to the surprising result that, for a test point drawn from the training distribution, MFVI's expected predictive variance exceeds that of the exact posterior. We demonstrate a pathological case of this effect, where the MFVI posterior fails to reduce predictive variance compared to the prior on in distribution data. We connect these results to the Cold Posterior Effect, arguing that varying the temperature can correct this overestimation, yielding predictions closer to those of the exact posterior. We validate our theory on synthetic and real-world regression tasks.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 25 of its 39 references that are in the map. First posted on arXiv 24 Jun 2026.
Builds on
25 of 39 references.
References · 25
- 2016Variational Inference: A Review for Statisticianscited by 1,472
- 2021What Are Bayesian Neural Network Posteriors Really Like?cited by 308
- 2016Eigenvalues of the Hessian in Deep Learning: Singularity and Beyondcited by 159
- 2019Practical Deep Learning with Bayesian Principlescited by 158
- 2019Functional Variational Bayesian Neural Networkscited by 127
- 2021Bayesian Neural Network Priors Revisitedcited by 121
- 2019'In-Between' Uncertainty in Bayesian Neural Networkscited by 80
- 2024Variational Learning is Effective for Large Deep Networkscited by 65
- 2020A statistical theory of cold posteriors in deep neural networkscited by 54
- 2022On Uncertainty, Tempering, and Data Augmentation in Bayesian Classificationcited by 46
- 2020Understanding Variational Inference in Function-Spacecited by 40
- 2021Data augmentation in Bayesian neural networks and the cold posterior effectcited by 36
13 more
- 2020Cold Posteriors and Aleatoric Uncertaintycited by 22
- 2021Disentangling the Roles of Curation, Data-Augmentation and the Prior in the Cold Posterior Effectcited by 22
- 2022Wide Mean-Field Bayesian Neural Networks Ignore the Datacited by 22
- 2022Partitioned Variational Inference: A Framework for Probabilistic Federated Learningcited by 13
- 2022How Tempering Fixes Data Augmentation in Bayesian Neural Networkscited by 13
- 2024Variational Inference for Uncertainty Quantification: an Analysis of Trade-offscited by 11
- 2023The Shrinkage-Delinkage Trade-off: An Analysis of Factorized Gaussian Approximations for Variational Inferencecited by 10
- 2024Variational Inference in Location-Scale Families: Exact Recovery of the Mean and Correlation Matrixcited by 9
- 2024Can a Confident Prior Replace a Cold Posterior?cited by 5
- 2025VIKING: Deep variational inference with stochastic projectionscited by 3
- 2025Learning Hyperparameters via a Data-Emphasized Variational Objectivecited by 2
- 2026Symmetry Guarantees Statistic Recovery in Variational Inferencecited by 1
- 2026Even More Guarantees for Variational Inference in the Presence of Symmetriescited by 1
Built on by
Not cited yet.