Distribution-free binary classification: prediction sets, confidence intervals and calibration
Chirag Gupta, Aleksandr Podkopaev, Aaditya Ramdas
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. * equal contribution; paper appeared as a spotlight at NeurIPS 2020.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 7a5a6478-cfdc-4858-98d0-9193c4740222Cited by top-tier papers35
- On Calibration and Out-of-Domain GeneralizationYoav Wald, Amir Feder, Daniel Greenfeld, Uri ShalitNeurIPS 2021 · 184 citations
- Assessing Generalization of SGD via DisagreementYiding Jiang, Vaishnavh Nagarajan, Christina Baek, J. Zico KolterICLR 2022 · 134 citations
- Conformal Language ModelingVictor Quach, Adam Fisch, Tal Schuster, Adam Yala et al.ICLR 2024 · 132 citations
- Improving Expert Predictions with Conformal PredictionEleni Straitouri, Lequn Wang, Nastaran Okati, Manuel Gomez RodriguezICML 2023 · 56 citations
- Top-label calibration and multiclass-to-binary reductionsChirag Gupta, Aaditya RamdasICLR 2022 · 51 citations
Builds on2
- Mix-n-Match : Ensemble and Compositional Methods for Uncertainty Calibration in Deep LearningJize Zhang, Bhavya Kailkhura, Thomas Yong-Jin HanICML 2020 · 276 citations
- A Unified View of Label Shift EstimationSaurabh Garg, Yifan Wu, Sivaraman Balakrishnan, Zachary C. LiptonNeurIPS 2020 · 186 citations
Related papers
- Generalized Venn and Venn-Abers Calibration with Applications in Conformal PredictionLars van der Laan, Ahmed M. AlaaICML 2025
- Distribution-Free Model-Agnostic Regression Calibration via Nonparametric MethodsShang Liu, Zhongze Cai, Xiaocheng LiNeurIPS 2023 · 5 citations
- Calibration of Neural Networks using SplinesKartik Gupta, Amir Rahimi, Thalaiyasingam Ajanthan, Thomas Mensink et al.ICLR 2021 · 128 citations
- MBCT: Tree-Based Feature-Aware Binning for Individual Uncertainty CalibrationSiguang Huang, Yunli Wang, Lili Mou, Huayue Zhang et al.WWW 2022 · 18 citations
- Probability Calibration for Knowledge Graph Embedding ModelsPedro Tabacof, Luca CostabelloICLR 2020 · 49 citations
