Conformal Risk-Averse Decision Making with Action Conditional Guarantee
Zihan Zhu, Shayan Kiyani, George Pappas, Hamed Hassani
Abstract
Reliable decision making pipelines powered by machine learning models require uncertainty quantification (UQ) methods that come with explicit safety guarantees. Conformal prediction provides such UQ by wrapping ML predictions into prediction sets, and recent work by Kiyani et al. (2025b) established that these sets can be translated into optimal risk-averse decision policies -- yet only inheriting marginal safety guarantees. We generalize and strengthen their results by (i) introducing action-conditional conformal prediction, which yields safety guarantees conditioned explicitly on each action taken by the decision maker, (ii) showing that action-conditional prediction sets serve as a proxy for the feasible decision space for risk-averse decision makers aiming to optimize action-conditional value-at-risk, and (iii) proposing a principled finite-sample algorithm based on pinball-loss minimization, connecting the framework of Gibbs et al. (2025) to action-conditional guarantees. Experiments on two real-world datasets confirm that our approach significantly improves action-conditional performance over conformal baselines.
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 67d6d8f0-86a4-41bf-b80f-4754199965beBuilds on28
- Conformal Risk ControlAnastasios Nikolas Angelopoulos, Stephen Bates, Adam Fisch, Lihua Lei et al.ICLR 2024 · 242 citations
- Class-Conditional Conformal Prediction with Many ClassesTiffany Ding, Anastasios Angelopoulos, Stephen Bates, Michael I. Jordan et al.NeurIPS 2023 · 160 citations
- Conformal Language ModelingVictor Quach, Adam Fisch, Tal Schuster, Adam Yala et al.ICLR 2024 · 132 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
- 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
- Spectral Conformal Risk Control: Distribution-Free Tail Guarantees via Bayesian QuadratureMohammad Mahdi Kazemi Esfeh, Qi Yan, Yongxing Zhang, Zahra Gholami et al.CVPR 2026
- Conformalized Credal Set PredictorsAlireza Javanmardi, David Stutz, Eyke HüllermeierNeurIPS 2024 · 28 citations
- Decision Theoretic Foundations for Conformal Prediction: Optimal Uncertainty Quantification for Risk-Averse AgentsShayan Kiyani, George J. Pappas, Aaron Roth, Hamed HassaniICML 2025
- Conformal Validity Guarantees Exist for Any Data Distribution (and How to Find Them)Drew Prinster, Samuel Don Stanton, Anqi Liu, Suchi SariaICML 2024 · 20 citations
