mlsanityData to 7 Oct 2026

On Temperature Scaling and Conformal Prediction of Deep Classifiers

arXiv 2402.05806 ↗First posted 8 Feb 2024cs.LG stat.MLcited by 13

Not on the map. Why?

Abstract (arXiv)

In many classification applications, the prediction of a deep neural network (DNN) based classifier needs to be accompanied by some confidence indication. Two popular approaches for that aim are: 1) Calibration: modifies the classifier's softmax values such that the maximal value better estimates the correctness probability; and 2) Conformal Prediction (CP): produces a prediction set of candidate labels that contains the true label with a user-specified probability, guaranteeing marginal coverage but not, e.g., per class coverage. In practice, both types of indications are desirable, yet, so far the interplay between them has not been investigated. Focusing on the ubiquitous Temperature Scaling (TS) calibration, we start this paper with an extensive empirical study of its effect on prominent CP methods. We show that while TS calibration improves the class-conditional coverage of adaptive CP methods, surprisingly, it negatively affects their prediction set sizes. Motivated by this behavior, we explore the effect of TS on CP beyond its calibration application and reveal an intriguing trend under which it allows to trade prediction set size and conditional coverage of adaptive CP methods. Then, we establish a mathematical theory that explains the entire non-monotonic trend. Finally, based on our experiments and theory, we offer simple guidelines for practitioners to effectively combine adaptive CP with calibration, aligned with user-defined goals.

Key facts. Cited by 13 papers in the map (data through 7 Oct 2026). Builds on 23 of its 43 references that are in the map. First posted on arXiv 8 Feb 2024.

Builds on

23 of 43 references.

References · 23

  1. 2015Deep Residual Learning for Image Recognitioncited by 51,809
  2. 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
  3. 2016Densely Connected Convolutional Networkscited by 6,691
  4. 2017On Calibration of Modern Neural Networkscited by 4,860
  5. 2017Enhancing The Reliability of Out-of-distribution Image Detection in Neural Networkscited by 1,325
  6. 2021A Gentle Introduction to Conformal Prediction and Distribution-Free Uncertainty Quantificationcited by 777
  7. 2019Conformal Prediction Under Covariate Shiftcited by 511
  8. 2019Measuring Calibration in Deep Learningcited by 443
  9. 2019A Survey of Deep Learning Techniques for Autonomous Drivingcited by 401
  10. 2020Classification with Valid and Adaptive Coveragecited by 390
  11. 2020Uncertainty Sets for Image Classifiers using Conformal Predictioncited by 365
  12. 2022Conformal prediction beyond exchangeabilitycited by 348
11 more
  1. 2016Least Ambiguous Set-Valued Classifiers with Bounded Error Levelscited by 270
  2. 2022Mitigating Neural Network Overconfidence with Logit Normalizationcited by 251
  3. 2023Conformal Prediction With Conditional Guaranteescited by 146
  4. 2023Class-Conditional Conformal Prediction with Many Classescited by 109
  5. 2021Learning Optimal Conformal Classifierscited by 108
  6. 2020Local Temperature Scaling for Probability Calibrationcited by 95
  7. 2020Distribution-free binary classification: prediction sets, confidence intervals and calibrationcited by 79
  8. 2023Federated Conformal Predictors for Distributed Uncertainty Quantificationcited by 46
  9. 2019Bin-wise Temperature Scaling (BTS): Improvement in Confidence Calibration Performance through Simple Scaling Techniquescited by 23
  10. 2024Does confidence calibration improve conformal prediction?cited by 12
  11. 2022Estimating Test Performance for AI Medical Devices under Distribution Shift with Conformal Predictioncited by 2

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2024: 1; 2025: 8; 2026: 4.

13 papers.

Sort: