Uncertainty-Aware Post-Hoc Calibration: Mitigating Confidently Incorrect Predictions Beyond Calibration Metrics
2510.17915v1
cs.LG, cs.AI, cs.NE, 68T07
2025-10-23
Авторы:
Hassan Gharoun, Mohammad Sadegh Khorshidi, Kasra Ranjbarigderi, Fang Chen, Amir H. Gandomi
Abstract
Despite extensive research on neural network calibration, existing methods
typically apply global transformations that treat all predictions uniformly,
overlooking the heterogeneous reliability of individual predictions.
Furthermore, the relationship between improved calibration and effective
uncertainty-aware decision-making remains largely unexplored. This paper
presents a post-hoc calibration framework that leverages prediction reliability
assessment to jointly enhance calibration quality and uncertainty-aware
decision-making. The framework employs proximity-based conformal prediction to
stratify calibration samples into putatively correct and putatively incorrect
groups based on semantic similarity in feature space. A dual calibration
strategy is then applied: standard isotonic regression calibrated confidence in
putatively correct predictions, while underconfidence-regularized isotonic
regression reduces confidence toward uniform distributions for putatively
incorrect predictions, facilitating their identification for further
investigations. A comprehensive evaluation is conducted using calibration
metrics, uncertainty-aware performance measures, and empirical conformal
coverage. Experiments on CIFAR-10 and CIFAR-100 with BiT and CoAtNet backbones
show that the proposed method achieves lower confidently incorrect predictions,
and competitive Expected Calibration Error compared with isotonic and
focal-loss baselines. This work bridges calibration and uncertainty
quantification through instance-level adaptivity, offering a practical post-hoc
solution that requires no model retraining while improving both probability
alignment and uncertainty-aware decision-making.