English

Instance-Wise Monotonic Calibration by Constrained Transformation

Machine Learning 2025-07-10 v1 Machine Learning

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

Keywords

Cite

@article{arxiv.2507.06516,
  title  = {Instance-Wise Monotonic Calibration by Constrained Transformation},
  author = {Yunrui Zhang and Gustavo Batista and Salil S. Kanhere},
  journal= {arXiv preprint arXiv:2507.06516},
  year   = {2025}
}

Comments

Accepted to Conference on Uncertainty in Artificial Intelligence (UAI)

R2 v1 2026-07-01T03:52:37.158Z