On Local Posterior Structure in Deep Ensembles
Not on the map. Why?
Abstract (arXiv)
Bayesian Neural Networks (BNNs) often improve model calibration and predictive uncertainty quantification compared to point estimators such as maximum-a-posteriori (MAP). Similarly, deep ensembles (DEs) are also known to improve calibration, and therefore, it is natural to hypothesize that deep ensembles of BNNs (DE-BNNs) should provide even further improvements. In this work, we systematically investigate this across a number of datasets, neural network architectures, and BNN approximation methods and surprisingly find that when the ensembles grow large enough, DEs consistently outperform DE-BNNs on in-distribution data. To shine light on this observation, we conduct several sensitivity and ablation studies. Moreover, we show that even though DE-BNNs outperform DEs on out-of-distribution metrics, this comes at the cost of decreased in-distribution performance. As a final contribution, we open-source the large pool of trained models to facilitate further research on this topic.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 33 of its 57 references that are in the map. First posted on arXiv 17 Mar 2025.
Builds on
33 of 57 references.
References · 33
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,399
- 2015Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shiftcited by 7,005
- 2017On Calibration of Modern Neural Networkscited by 4,960
- 2015Dropout as a Bayesian Approximation: Representing Model Uncertainty in Deep Learningcited by 4,448
- 2016Simple and Scalable Predictive Uncertainty Estimation using Deep Ensemblescited by 3,926
- 2016Wide Residual Networkscited by 3,070
- 2015Variational Inference with Normalizing Flowscited by 2,170
- 2019Normalizing Flows for Probabilistic Modeling and Inferencecited by 1,323
- 2018Averaging Weights Leads to Wider Optima and Better Generalizationcited by 1,209
- 2017MoleculeNet: A Benchmark for Molecular Machine Learningcited by 1,051
- 2014Deep Neural Networks are Easily Fooled: High Confidence Predictions for Unrecognizable Imagescited by 776
- 2018Loss Surfaces, Mode Connectivity, and Fast Ensembling of DNNscited by 588
21 more
- 2019A Simple Baseline for Bayesian Uncertainty in Deep Learningcited by 560
- 2015Weight Uncertainty in Neural Networkscited by 554
- 2020Bayesian Deep Learning and a Probabilistic Perspective of Generalizationcited by 480
- 2021Equivariant message passing for the prediction of tensorial properties and molecular spectracited by 441
- 2021Laplace Redux -- Effortless Bayesian Deep Learningcited by 365
- 2021What Are Bayesian Neural Network Posteriors Really Like?cited by 308
- 2017Stochastic Gradient Descent as Approximate Bayesian Inferencecited by 300
- 2020Being Bayesian, Even Just a Bit, Fixes Overconfidence in ReLU Networkscited by 231
- 2020Efficient and Scalable Bayesian Neural Nets with Rank-1 Factorscited by 171
- 2021Repulsive Deep Ensembles are Bayesiancited by 104
- 2024Variational Bayesian Last Layerscited by 69
- 2024Variational Learning is Effective for Large Deep Networkscited by 65
- 2022Do Bayesian Neural Networks Need To Be Fully Stochastic?cited by 63
- 2013Understanding predictive information criteria for Bayesian modelscited by 52
- 2019Filter Response Normalization Layer: Eliminating Batch Dependence in the Training of Deep Neural Networkscited by 45
- 2017Using stacking to average Bayesian predictive distributionscited by 39
- 2019Alchemy: A Quantum Chemistry Dataset for Benchmarking AI Modelscited by 36
- 2021Mixtures of Laplace Approximations for Improved Post-Hoc Uncertainty in Deep Learningcited by 25
- 2024Reparameterization invariance in approximate Bayesian inferencecited by 17
- 2022Posterior Refinement Improves Sample Efficiency in Bayesian Neural Networkscited by 14
- 2021Should Ensemble Members Be Calibrated?cited by 11
Built on by
Not cited yet.