Decision Theoretic Foundations for Conformal Prediction: Optimal Uncertainty Quantification for Risk-Averse Agents
Shayan Kiyani, George J. Pappas, Aaron Roth, Hamed Hassani
Abstract
A fundamental question in data-driven decision making is how to quantify the uncertainty of predictions in ways that can usefully inform downstream action. This interface between prediction uncertainty and decision-making is especially important in risk-sensitive domains, such as medicine. In this paper, we develop decision-theoretic foundations that connect uncertainty quantification using prediction sets with risk-averse decision-making. Specifically, we answer three fundamental questions: (1) What is the correct notion of uncertainty quantification for risk-averse decision makers? We prove that prediction sets are optimal for decision makers who wish to optimize their value at risk. ( 2 ) What is the optimal policy that a risk averse decision maker should use to map prediction sets to actions? We show that a simple max-min decision policy is optimal for risk-averse decision makers. Finally, (3) How can we derive prediction sets that are optimal for such decision makers? We provide an exact characterization in the population regime and a distribution free finite-sample construction. Answering these questions naturally leads to an algorithm, Risk-Averse Calibration (RAC), which follows a provably optimal design for deriving action policies from predictions. RAC is designed to be both practical -capable of leveraging the quality of predictions in a black-box manner to enhance downstream utility-and safe-adhering to a user-defined risk threshold and optimizing the corresponding risk quantile of the user's downstream utility. Finally, we experimentally demonstrate the significant advantages of RAC in applications such as medical diagnosis and recommendation systems. Specifically, we show that RAC achieves a substantially improved trade-off between safety and utility, offering higher utility compared to existing methods while maintaining the safety guarantee.
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.
Cited by top-tier papers16
- Conformal Risk Training: End-to-End Optimization of Conformal Risk ControlChristopher Yeh, Nicolas Christianson, Adam Wierman, Yisong YueNeurIPS 2025 · 14 citations
- Conformal Prediction Beyond the Seen: A Missing Mass Perspective for Uncertainty Quantification in Generative ModelsSima Noorani, Shayan Kiyani, George J. Pappas, Hamed HassaniNeurIPS 2025 · 10 citations
- Human-AI Collaborative Uncertainty QuantificationSima Noorani, Shayan Kiyani, George Pappas, Hamed HassaniICML 2026 · 8 citations
- Conformalized Decision Risk AssessmentWenbin Zhou, Agni Orfanoudaki, Shixiang ZhuICLR 2026 · 6 citations
- Multi-Condition Conformal SelectionQingyang Hao, Wenbo Liao, Bingyi Jing, Hongxin WeiICLR 2026 · 5 citations
Builds on16
- Classification with Valid and Adaptive CoverageYaniv Romano, Matteo Sesia, Emmanuel J. CandèsNeurIPS 2020 · 586 citations
- Conformal Risk ControlAnastasios Nikolas Angelopoulos, Stephen Bates, Adam Fisch, Lihua Lei et al.ICLR 2024 · 242 citations
- Calibrating Predictions to Decisions: A Novel Approach to Multi-Class CalibrationShengjia Zhao, Michael P. Kim, Roshni Sahoo, Tengyu Ma et al.NeurIPS 2021 · 96 citations
- Practical Adversarial Multivalid Conformal PredictionOsbert Bastani, Varun Gupta, Christopher Jung, Georgy Noarov et al.NeurIPS 2022 · 82 citations
- Conformal Prediction for Uncertainty-Aware Planning with Diffusion Dynamics ModelJiankai Sun, Yiqi Jiang, Jianing Qiu, Parth Nobel et al.NeurIPS 2023 · 79 citations
Related papers
- Optimal Decision-Making Based on Prediction SetsTao Wang, Edgar DobribanICML 2026
- Conformal Risk-Averse Decision Making with Action Conditional GuaranteeZihan Zhu, Shayan Kiyani, George Pappas, Hamed HassaniICML 2026 · 1 citation
- PAC Prediction Sets for Meta-LearningSangdon Park, Edgar Dobriban, Insup Lee, Osbert BastaniNeurIPS 2022 · 21 citations
- Utility-Directed Conformal Prediction: A Decision-Aware Framework for Actionable Uncertainty QuantificationSantiago Cortes-Gomez, Carlos Miguel Patiño, Yewon Byun, Steven Wu et al.ICLR 2025
- Robust Decision-Making with Partially Calibrated ForecastersShayan Kiyani, Hamed Hassani, George J. Pappas, Aaron RothICLR 2026 · 1 citation
