Bias-Reduced Uncertainty Estimation for Deep Neural Classifiers
Authors: Yonatan Geifman, Guy Uziel, Ran El-Yaniv
Keywords: Uncertainty estimation, Deep learning
TL;DR: We use snapshots from the training process to improve any uncertainty estimation method of a DNN classifier.
Abstract: We consider the problem of uncertainty estimation in the context of (non-Bayesian) deep neural classification. In this context, all known methods are based on extracting uncertainty signals from a trained network optimized to solve the classification problem at hand. We demonstrate that such techniques tend to introduce biased estimates for instances whose predictions are supposed to be highly confident. We argue that this deficiency is an artifact of the dynamics of training with SGD-like optimizers, and it has some properties similar to overfitting. Based on this observation, we develop an uncertainty estimation algorithm that selectively estimates the uncertainty of highly confident points, using earlier snapshots of the trained model, before their estimates are jittered (and way before they are ready for actual classification). We present extensive experiments indicating that the proposed algorithm provides uncertainty estimates that are consistently better than all known methods.