Post-hoc Uncertainty Calibration for Domain Drift Scenarios

Christian Tomani,Sebastian Gruber,Muhammed Ebrar Erdem,Daniel Cremers,Florian Buettner
DOI: https://doi.org/10.48550/arXiv.2012.10988
2021-06-23
Abstract:We address the problem of uncertainty calibration. While standard deep neural networks typically yield uncalibrated predictions, calibrated confidence scores that are representative of the true likelihood of a prediction can be achieved using post-hoc calibration methods. However, to date the focus of these approaches has been on in-domain calibration. Our contribution is two-fold. First, we show that existing post-hoc calibration methods yield highly over-confident predictions under domain shift. Second, we introduce a simple strategy where perturbations are applied to samples in the validation set before performing the post-hoc calibration step. In extensive experiments, we demonstrate that this perturbation step results in substantially better calibration under domain shift on a wide range of architectures and modelling tasks.
Machine Learning,Artificial Intelligence
What problem does this paper attempt to address?