English

Information-theoretic Generalization Analysis for Expected Calibration Error

Machine Learning 2025-05-27 v2 Statistics Theory Machine Learning Statistics Theory

Abstract

While the expected calibration error (ECE), which employs binning, is widely adopted to evaluate the calibration performance of machine learning models, theoretical understanding of its estimation bias is limited. In this paper, we present the first comprehensive analysis of the estimation bias in the two common binning strategies, uniform mass and uniform width binning. Our analysis establishes upper bounds on the bias, achieving an improved convergence rate. Moreover, our bounds reveal, for the first time, the optimal number of bins to minimize the estimation bias. We further extend our bias analysis to generalization error analysis based on the information-theoretic approach, deriving upper bounds that enable the numerical evaluation of how small the ECE is for unknown data. Experiments using deep learning models show that our bounds are nonvacuous thanks to this information-theoretic generalization analysis approach.

Keywords

Cite

@article{arxiv.2405.15709,
  title  = {Information-theoretic Generalization Analysis for Expected Calibration Error},
  author = {Futoshi Futami and Masahiro Fujisawa},
  journal= {arXiv preprint arXiv:2405.15709},
  year   = {2025}
}

Comments

Accepted by the 38th Conference on Neural Information Processing Systems (NeurIPS2024), 52 pages, 6 figures

R2 v1 2026-06-28T16:39:16.547Z