Revisiting the Calibration of Modern Neural Networks

By Matthias Minderer, Josip Djolonga, Rob Romijnders, Frances Ann Hubis, Xiaohua Zhai, Neil Houlsby, Dustin Tran, Mario Lucic. 2021

562 citations recorded by the source corpus.

Accurate estimation of predictive uncertainty (model calibration) is essential for the safe application of neural networks. Many instances of miscalibration in modern neural networks have been reported, suggesting a trend that newer, more accurate models produce poorly calibrated predictions. Here, we revisit this question for recent state-of-the-art image classification models. We systematically relate model calibration and accuracy, and find that the most recent models, notably those not using convolutions, are among the best calibrated. Trends observed in prior model generations, such as decay of calibration with distribution shift or model size, are less pronounced in recent architectures. We also show that model size and amount of pretraining do not fully explain these differences, suggesting that architecture is a major determinant of calibration properties.

Subfields: Artificial Intelligence, Evaluation & Benchmarks, Safety, Robustness & Interpretability

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Artificial Intelligence

  1. Neil Houlsby
  2. Quoc V. Le
  3. Jianfeng Gao
  4. Mohit Bansal
  5. Yoshua Bengio
  6. Percy Liang
  7. Xiaohua Zhai
  8. Mostafa Dehghani

In Evaluation & Benchmarks

  1. Neil Houlsby
  2. Xiaohua Zhai
  3. Lucas Beyer
  4. Alexander Kolesnikov
  5. Sjoerd van Steenkiste
  6. Mitchell Wortsman
  7. Gabriel Ilharco
  8. Fisher Yu

In Safety, Robustness & Interpretability

  1. Justin Gilmer
  2. Ludwig Schmidt
  3. Michael Tschannen
  4. Xiao Wang
  5. Yi Tay
  6. Alexander Kolesnikov
  7. Carlos Riquelme
  8. Yelong Shen

Citations in the corpus