How Flawed Is ECE? An Analysis via Logit Smoothing

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Chidambaram, Muthu, Lee, Holden, McSwiggen, Colin, Rezchikov, Semon
Format: Preprint
Veröffentlicht: 2024
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866911899706195968
author Chidambaram, Muthu
Lee, Holden
McSwiggen, Colin
Rezchikov, Semon
author_facet Chidambaram, Muthu
Lee, Holden
McSwiggen, Colin
Rezchikov, Semon
contents Informally, a model is calibrated if its predictions are correct with a probability that matches the confidence of the prediction. By far the most common method in the literature for measuring calibration is the expected calibration error (ECE). Recent work, however, has pointed out drawbacks of ECE, such as the fact that it is discontinuous in the space of predictors. In this work, we ask: how fundamental are these issues, and what are their impacts on existing results? Towards this end, we completely characterize the discontinuities of ECE with respect to general probability measures on Polish spaces. We then use the nature of these discontinuities to motivate a novel continuous, easily estimated miscalibration metric, which we term Logit-Smoothed ECE (LS-ECE). By comparing the ECE and LS-ECE of pre-trained image classification models, we show in initial experiments that binned ECE closely tracks LS-ECE, indicating that the theoretical pathologies of ECE may be avoidable in practice.
format Preprint
id arxiv_https___arxiv_org_abs_2402_10046
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle How Flawed Is ECE? An Analysis via Logit Smoothing
Chidambaram, Muthu
Lee, Holden
McSwiggen, Colin
Rezchikov, Semon
Machine Learning
Probability
68T37 (Primary) 62-08, 60E05 (Secondary)
Informally, a model is calibrated if its predictions are correct with a probability that matches the confidence of the prediction. By far the most common method in the literature for measuring calibration is the expected calibration error (ECE). Recent work, however, has pointed out drawbacks of ECE, such as the fact that it is discontinuous in the space of predictors. In this work, we ask: how fundamental are these issues, and what are their impacts on existing results? Towards this end, we completely characterize the discontinuities of ECE with respect to general probability measures on Polish spaces. We then use the nature of these discontinuities to motivate a novel continuous, easily estimated miscalibration metric, which we term Logit-Smoothed ECE (LS-ECE). By comparing the ECE and LS-ECE of pre-trained image classification models, we show in initial experiments that binned ECE closely tracks LS-ECE, indicating that the theoretical pathologies of ECE may be avoidable in practice.
title How Flawed Is ECE? An Analysis via Logit Smoothing
topic Machine Learning
Probability
68T37 (Primary) 62-08, 60E05 (Secondary)
url https://arxiv.org/abs/2402.10046