Instance-Wise Monotonic Calibration by Constrained Transformation

Yunrui Zhang, Gustavo Enrique Batista, Salil S. Kanhere

Keywords: Calibration, Classification, Uncertainty quantification, Safety in machine learning

TL;DR: We propose a novel monotonic post-hoc calibration method that leverages constrained transformations to achieve state-of-the-art performance.

Abstract: Deep neural networks often produce miscalibrated probability estimates, leading to overconfident predictions. A common approach for calibration is fitting a post-hoc calibration map on unseen validation data that transforms predicted probabilities. A key desirable property of the calibration map is instance-wise monotonicity (i.e., preserving the ranking of probability outputs). However, most existing post-hoc calibration methods do not guarantee monotonicity. Previous monotonic approaches either use an under-parameterized calibration map with limited expressive ability or rely on black-box neural networks, which lack interpretability and robustness. In this paper, we propose a family of novel monotonic post-hoc calibration methods, which employs a constrained calibration map parameterized linearly with respect to the number of classes. Our proposed approach ensures expressiveness, robustness, and interpretability while preserving the relative ordering of the probability output by formulating the proposed calibration map as a constrained optimization problem. Our proposed methods achieve state-of-the-art performance across datasets with different deep neural network models, outperforming existing calibration methods while being data and computation-efficient. Our code is available at https://github.com/YunruiZhang/Calibration-by-Constrained-Transformation

Code Link: https://github.com/YunruiZhang/Calibration-by-Constrained-Transformation

Readers: auai.org/UAI/2025/Conference, auai.org/UAI/2025/Conference/Area_Chairs, auai.org/UAI/2025/Conference/Reviewers, auai.org/UAI/2025/Conference/Submission540/Authors, auai.org/UAI/2025/Conference/Submission540/Reproducibility_Reviewers

Submission Number: 540