A Confidence Interval for the $\ell_2$ Expected Calibration Error

Yan Sun, Pratik Chaudhari, Ian J. Barnett, Edgar Dobriban
8/16/2024
stat.MLcs.LG

No official code found. Best implementation: sylydya/CI-ECE (1 star, last commit 2024-08-27).

Abstract

Recent advances in machine learning have significantly improved prediction accuracy in various applications. However, ensuring the calibration of probabilistic predictions remains a significant challenge. Despite efforts to enhance model calibration, the rigorous statistical evaluation of model calibration remains less explored. In this work, we develop confidence intervals the $\ell_2$ Expected Calibration Error (ECE). We consider top-1-to-$k$ calibration, which includes both the popular notion of confidence calibration as well as full calibration. For a debiased estimator of the ECE, we show asymptotic normality, but with different convergence rates and asymptotic variances for calibrated and miscalibrated models. We develop methods to construct asymptotically valid confidence intervals for the ECE, accounting for this behavior as well as non-negativity. Our theoretical findings are supported through extensive experiments, showing that our methods produce valid confidence intervals with shorter lengths compared to those obtained by resampling-based methods.

View on arXivView PDF

Code Implementations(1)

sylydya/CI-ECECommunity100%
11Shell, PythonAug 12, 20242 years ago

Similar Papers

Cite this paper

@article{sun2024confidence,
  title  = {A Confidence Interval for the $\ell_2$ Expected Calibration Error},
  author = {Yan Sun and Pratik Chaudhari and Ian J. Barnett and Edgar Dobriban},
  year   = {2024},
  eprint = {2408.08998},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2408.08998v3}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2408.08998.svg)](https://codeofpaper.com/paper/2408.08998)

Discussion