English

Distribution-free binary classification: prediction sets, confidence intervals and calibration

Machine Learning 2022-02-17 v4 Artificial Intelligence Machine Learning Statistics Theory Methodology Statistics Theory

Abstract

We study three notions of uncertainty quantification -- calibration, confidence intervals and prediction sets -- for binary classification in the distribution-free setting, that is without making any distributional assumptions on the data. With a focus towards calibration, we establish a 'tripod' of theorems that connect these three notions for score-based classifiers. A direct implication is that distribution-free calibration is only possible, even asymptotically, using a scoring function whose level sets partition the feature space into at most countably many sets. Parametric calibration schemes such as variants of Platt scaling do not satisfy this requirement, while nonparametric schemes based on binning do. To close the loop, we derive distribution-free confidence intervals for binned probabilities for both fixed-width and uniform-mass binning. As a consequence of our 'tripod' theorems, these confidence intervals for binned probabilities lead to distribution-free calibration. We also derive extensions to settings with streaming data and covariate shift.

Keywords

Cite

@article{arxiv.2006.10564,
  title  = {Distribution-free binary classification: prediction sets, confidence intervals and calibration},
  author = {Chirag Gupta and Aleksandr Podkopaev and Aaditya Ramdas},
  journal= {arXiv preprint arXiv:2006.10564},
  year   = {2022}
}

Comments

34 pages; significant updates from previous version (unambiguous notation, better exposition, and cleaner results); originally appeared as a spotlight at Neural Information Processing Systems (NeurIPS) '20

R2 v1 2026-06-23T16:26:10.195Z