Calibrated and uncertain? Evaluating uncertainty estimates in binary classification models
Abstract Rigorous statistical methods, including parameter estimation with accompanying uncertainties, underpin the validity of scientific discovery, especially in the natural sciences. With increasingly complex data models such as deep learning techniques, uncertainty quantification has become exceedingly difficult and a plethora of techniques have been proposed. In this case study, we use the unifying framework of approximate Bayesian inference combined with empirical tests on carefully created synthetic classification datasets to investigate qualitative properties of six different probabilistic machine learning algorithms for class probability and uncertainty estimation: (i) a neural network ensemble (NNE), (ii) NNE with conflictual loss, (iii) evidential deep learning, (iv) a single neural network with Monte Carlo dropout, (v) Gaussian process classification and (vi) a Dirichlet process mixture model. We check if the algorithms produce uncertainty estimates which reflect commonly desired properties, such as being well calibrated and exhibiting an increase in uncertainty for out-of-distribution (OOD) data points. Our results indicate that all algorithms show reasonably good calibration performance on our synthetic test sets, but none of the deep learning based algorithms provide uncertainties that consistently reflect lack of experimental evidence for OOD data points. We hope our study may serve as a clarifying example for researchers that are using or developing methods of uncertainty estimation for scientific data-driven modeling and analysis.
Authors
- Aurora Singstad Grefsrud
Institutions
- Western Norway University of Applied Sciences (NO)
- University of Bergen (NO)
Publication Details
- Journal
- Machine Learning Science and Technology
- Published
- 2026-02-13
- DOI
- https://doi.org/10.1088/2632-2153/ae45ed
- Primary Topic
- Gaussian Processes and Bayesian Inference
- Type
- article
- Field-Weighted Citation Impact
- 0.00