Simple and Scalable Predictive Uncertainty Estimation using Deep Ensembles
University College London · Google (United States)
Abstract
Deep neural networks (NNs) are powerful black box predictors that have recently achieved impressive performance on a wide spectrum of tasks. Quantifying predictive uncertainty in NNs is a challenging and yet unsolved problem. Bayesian NNs, which learn a distribution over weights, are currently the state-of-the-art for estimating predictive uncertainty; however these require significant modifications to the training procedure and are computationally expensive compared to standard (non-Bayesian) NNs. We propose an alternative to Bayesian NNs that is simple to implement, readily parallelizable, requires very little hyperparameter tuning, and yields high quality predictive uncertainty estimates. Through a series…
Citation impact
- FWCI
- —
- Percentile
- —
- References
- 53
Authors
3Topics & keywords
- Hyperparameter
- Computer science
- Robustness (evolution)
- Machine learning
- Bayesian probability
- Artificial intelligence
- Scalability
- Artificial neural network