mlsanityData to 14 Oct 2026

Adaptive Cumulative Mass Calibration with Conformal Prediction

arXiv 2505.15437 ↗First posted 21 May 2025stat.ML cs.LGcited by 1

Not on the map. Why?

Abstract (arXiv)

Reliable probability estimates by classifiers are essential in high-risk applications. In practice, however, predicted probabilities are often miscalibrated, and many existing post-hoc calibration methods typically lack guarantees that a specific notion of calibration is achieved after the correction procedure is applied. We introduce a *set-based* perspective on calibration through the notion of *cumulative mass calibration* and the corresponding error measures. We propose a new calibration procedure based on conformal prediction that forms cumulative probabilities with guaranteed marginal coverage. We introduce an _adaptive temperature scaling algorithm_, with the temperature tuned for each input to satisfy the conformal coverage constraint. As we show, this procedure can be efficiently implemented. Across image classification tasks, particularly in settings with many classes, our method improves newly introduced calibration error measures (_CMCE_ and α-CMCE) *and* standard metrics (such as ECE, cw-ECE, MCE) over the existing baselines.

Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Builds on 34 of its 47 references that are in the map. First posted on arXiv 21 May 2025.

Builds on

34 of 47 references.

References · 34

  1. 2015Deep Residual Learning for Image Recognitioncited by 52,096
  2. 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
  3. 2017On Calibration of Modern Neural Networkscited by 4,960
  4. 2016End to End Learning for Self-Driving Carscited by 1,009
  5. 2019When Does Label Smoothing Help?cited by 988
  6. 2019Aleatoric and Epistemic Uncertainty in Machine Learning: An Introduction to Concepts and Methodscited by 913
  7. 2017Selective Classification for Deep Neural Networkscited by 852
  8. 2019Measuring Calibration in Deep Learningcited by 446
  9. 2020Classification with Valid and Adaptive Coveragecited by 396
  10. 2020Simple and Principled Uncertainty Estimation with Deterministic Deep Learning via Distance Awarenesscited by 389
  11. 2020Uncertainty Sets for Image Classifiers using Conformal Predictioncited by 366
  12. 2019Beyond temperature scaling: Obtaining well-calibrated multiclass probabilities with Dirichlet calibrationcited by 364
22 more
  1. 2016Least Ambiguous Set-Valued Classifiers with Bounded Error Levelscited by 272
  2. 2019Evaluating model calibration in classificationcited by 168
  3. 2021Benchmarking Representation Learning for Natural World Image Collectionscited by 168
  4. 2019Calibration tests in multi-class classification: A unifying frameworkcited by 79
  5. 2020Distribution-free binary classification: prediction sets, confidence intervals and calibrationcited by 79
  6. 2021Calibrating Predictions to Decisions: A Novel Approach to Multi-Class Calibrationcited by 79
  7. 2020Outcome Indistinguishabilitycited by 75
  8. 2022Better Uncertainty Calibration via Proper Scores for Classification and Beyondcited by 65
  9. 2021Optimal strategies for reject option classifierscited by 61
  10. 2023Conformal Prediction for Deep Classifier via Label Rankingcited by 53
  11. 2021Parameterized Temperature Scaling for Boosting the Expressive Power in Post-Hoc Uncertainty Calibrationcited by 48
  12. 2022Sample-dependent Adaptive Temperature Scaling for Improved Calibrationcited by 48
  13. 2022On Uncertainty, Tempering, and Data Augmentation in Bayesian Classificationcited by 46
  14. 2025Decision Theoretic Foundations for Conformal Prediction: Optimal Uncertainty Quantification for Risk-Averse Agentscited by 40
  15. 2022Adaptive Temperature Scaling for Robust Calibration of Deep Neural Networkscited by 24
  16. 2024Probabilistic Conformal Prediction with Approximate Conditional Validitycited by 23
  17. 2015Large-scale probabilistic predictors with and without guarantees of validitycited by 18
  18. 2025Rectifying Conformity Scores for Better Conditional Coveragecited by 17
  19. 2024Does confidence calibration improve conformal prediction?cited by 12
  20. 2024Normalizing Flows for Conformal Regressioncited by 11
  21. 2024Confidence Calibration of Classifiers with Many Classescited by 11
  22. 2023Efficient Conformal Prediction under Data Heterogeneitycited by 9

Built on by

Cited by year

Citing papers per year: 2025: 1.

1 paper.