English

Calibrated Top-1 Uncertainty estimates for classification by score based models

Machine Learning 2020-06-17 v4 Machine Learning

Abstract

While the accuracy of modern deep learning models has significantly improved in recent years, the ability of these models to generate uncertainty estimates has not progressed to the same degree. Uncertainty methods are designed to provide an estimate of class probabilities when predicting class assignment. While there are a number of proposed methods for estimating uncertainty, they all suffer from a lack of calibration: predicted probabilities can be off from empirical ones by a few percent or more. By restricting the scope of our predictions to only the probability of Top-1 error, we can decrease the calibration error of existing methods to less than one percent. As a result, the scores of the methods also improve significantly over benchmarks.

Keywords

Cite

@article{arxiv.1903.09215,
  title  = {Calibrated Top-1 Uncertainty estimates for classification by score based models},
  author = {Adam M. Oberman and Chris Finlay and Alexander Iannantuono and Tiago Salvador},
  journal= {arXiv preprint arXiv:1903.09215},
  year   = {2020}
}

Comments

12 pages, 5 figures, 6 tables (major revision, new benchmark allows us to show model calibration is better)

R2 v1 2026-06-23T08:15:34.462Z