Minimum-Risk Recalibration of Classifiers
Zeyu Sun, Dogyoon Song, Alfred O. Hero III
Abstract
Recalibrating probabilistic classifiers is vital for enhancing the reliability and accuracy of predictive models. Despite the development of numerous recalibration algorithms, there is still a lack of a comprehensive theory that integrates calibration and sharpness (which is essential for maintaining predictive power). In this paper, we introduce the concept of minimum-risk recalibration within the framework of mean-squared-error (MSE) decomposition, offering a principled approach for evaluating and recalibrating probabilistic classifiers. Using this framework, we analyze the uniform-mass binning (UMB) recalibration method and establish a finite-sample risk upper bound of order where is the number of bins and is the sample size. By balancing calibration and sharpness, we further determine that the optimal number of bins for UMB scales with , resulting in a risk bound of approximately . Additionally, we tackle the challenge of label shift by proposing a two-stage approach that adjusts the recalibration function using limited labeled data from the target domain. Our results show that transferring a calibrated classifier requires significantly fewer target samples compared to recalibrating from scratch. We validate our theoretical findings through numerical simulations, which confirm the tightness of the proposed bounds, the optimal number of bins, and the effectiveness of label shift adaptation.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Cited by top-tier papers7
- Information-theoretic Generalization Analysis for Expected Calibration ErrorFutoshi Futami, Masahiro FujisawaNeurIPS 2024 · 22 citations
- Efficient Calibration for Decision MakingParikshit Gopalan, Konstantinos Stavropoulos, Kunal Talwar, Pranay TankalaSTOC 2026 · 3 citations
- Optimal and Provable Calibration in High-Dimensional Binary Classification: Angular Calibration and Platt ScalingYufan Li, Pragya SurNeurIPS 2025 · 2 citations
- Addressing Label Shift in Distributed Learning via Entropy RegularizationZhiyuan Wu, Changkyu Choi, Xiangcheng Cao, Volkan Cevher et al.ICLR 2025
- PAC-Bayes Analysis for Recalibration in ClassificationMasahiro Fujisawa, Futoshi FutamiICML 2025
Builds on12
- Calibrate Before Use: Improving Few-shot Performance of Language ModelsZihao Zhao, Eric Wallace, Shi Feng, Dan Klein et al.ICML 2021 · 1,843 citations
- Mix-n-Match : Ensemble and Compositional Methods for Uncertainty Calibration in Deep LearningJize Zhang, Bhavya Kailkhura, Thomas Yong-Jin HanICML 2020 · 276 citations
- A Unified View of Label Shift EstimationSaurabh Garg, Yifan Wu, Sivaraman Balakrishnan, Zachary C. LiptonNeurIPS 2020 · 186 citations
- Calibration of Neural Networks using SplinesKartik Gupta, Amir Rahimi, Thalaiyasingam Ajanthan, Thomas Mensink et al.ICLR 2021 · 128 citations
- Maximum Likelihood with Bias-Corrected Calibration is Hard-To-Beat at Label Shift AdaptationAmr Alexandari, Anshul Kundaje, Avanti ShrikumarICML 2020 · 123 citations
Related papers
- Modular Conformal CalibrationCharles Marx, Shengjia Zhao, Willie Neiswanger, Stefano ErmonICML 2022 · 19 citations
- Transferable Calibration with Lower Bias and Variance in Domain AdaptationXimei Wang, Mingsheng Long, Jianmin Wang, Michael I. JordanNeurIPS 2020 · 70 citations
- Calibration by Distribution Matching: Trainable Kernel Calibration MetricsCharlie Marx, Sofian Zalouk, Stefano ErmonNeurIPS 2023 · 21 citations
- Confidence Calibration for Domain Generalization under Covariate ShiftYunye Gong, Xiao Lin, Yi Yao, Thomas G. Dietterich et al.ICCV 2021 · 35 citations
- Top-label calibration and multiclass-to-binary reductionsChirag Gupta, Aaditya RamdasICLR 2022 · 51 citations
