PAC Confidence Predictions for Deep Neural Network Classifiers
Sangdon Park, Shuo Li, Insup Lee, Osbert Bastani
Abstract
A key challenge for deploying deep neural networks (DNNs) in safety critical settings is the need to provide rigorous ways to quantify their uncertainty. In this paper, we propose a novel algorithm for constructing predicted classification confidences for DNNs that comes with provable correctness guarantees. Our approach uses Clopper-Pearson confidence intervals for the Binomial distribution in conjunction with the histogram binning approach to calibrated prediction. In addition, we demonstrate how our predicted confidences can be used to enable downstream guarantees in two settings: (i) fast DNN inference, where we demonstrate how to compose a fast but inaccurate DNN with an accurate but slow DNN in a rigorous way to improve performance without sacrificing accuracy, and (ii) safe planning, where we guarantee safety when using a DNN to predict whether a given action is safe based on visual observations. In our experiments, we demonstrate that our approach can be used to provide guarantees for state-of-the-art DNNs.
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 c346d0a9-86a2-4be9-b651-1b2f8d74171cCited by top-tier papers5
- PAC Prediction Sets Under Covariate ShiftSangdon Park, Edgar Dobriban, Insup Lee, Osbert BastaniICLR 2022 · 54 citations
- Consistent Accelerated Inference via Confident Adaptive TransformersTal Schuster, Adam Fisch, Tommi S. Jaakkola, Regina BarzilayEMNLP 2021 · 30 citations
- Experts Don't Cheat: Learning What You Don't Know By Predicting PairsDaniel D. Johnson, Daniel Tarlow, David Duvenaud, Chris J. MaddisonICML 2024 · 18 citations
- PAC Prediction Sets Under Label ShiftWenwen Si, Sangdon Park, Insup Lee, Edgar Dobriban et al.ICLR 2024 · 15 citations
- Certified Error Control of Candidate Set Pruning for Two-Stage Relevance RankingMinghan Li, Xinyu Zhang, Ji Xin, Hongyang Zhang et al.EMNLP 2022 · 3 citations
Builds on3
- Habitat: A Platform for Embodied AI ResearchManolis Savva, Jitendra Malik, Devi Parikh, Dhruv Batra et al.ICCV 2019 · 1,863 citations
- Personalized HeartSteps: A Reinforcement Learning Algorithm for Optimizing Physical ActivityPeng Liao, Kristjan H. Greenewald, Predrag V. Klasnja, Susan A. MurphyUbiComp 2020 · 163 citations
- PAC Confidence Sets for Deep Neural Networks via Calibrated PredictionSangdon Park, Osbert Bastani, Nikolai Matni, Insup LeeICLR 2020 · 77 citations
Related papers
- Evaluating Deep Neural Networks in Deployment: A Comparative Study (Replicability Study)Eduard Pinconschi, Divya Gopinath, Rui Abreu, Corina S. PasareanuISSTA 2024
- Confidence-Aware Learning for Deep Neural NetworksJooyoung Moon, Jihyo Kim, Younghak Shin, Sangheum HwangICML 2020 · 184 citations
- Provably Adversarially Robust Detection of Out-of-Distribution Data (Almost) for FreeAlexander Meinke, Julian Bitterwolf, Matthias HeinNeurIPS 2022 · 23 citations
- Balancing Two Classifiers via A Simplex ETF Structure for Model CalibrationJiani Ni, He Zhao, Jintong Gao, Dandan Guo et al.CVPR 2025
- Certifiably Adversarially Robust Detection of Out-of-Distribution DataJulian Bitterwolf, Alexander Meinke, Matthias HeinNeurIPS 2020 · 91 citations
