Robust Decision-Making with Partially Calibrated Forecasters
Shayan Kiyani, Hamed Hassani, George J. Pappas, Aaron Roth
Abstract
Calibration has emerged as a foundational goal in "trustworthy machine learning", in part because of its strong decision theoretic semantics. Independent of the underlying distribution, and independent of the decision maker's utility function, calibration promises that amongst all policies mapping predictions to actions, the uniformly best policy is the one that "trusts the predictions" and acts as if they were correct. But this is true only of fully calibrated forecasts, which are tractable to guarantee only for very low dimensional prediction problems. For higher dimensional prediction problems (e.g. when outcomes are multiclass), weaker forms of calibration have been studied that lack these decision theoretic properties. In this paper we study how a conservative decision maker should map predictions endowed with these weaker ("partial") calibration guarantees to actions, in a way that is robust in a minimax sense: i.e. to maximize their expected utility in the worst case over distributions consistent with the calibration guarantees. We characterize their minimax optimal decision rule via a duality argument, and show that surprisingly, "trusting the predictions and acting accordingly" is recovered in this minimax sense by decision calibration (and any strictly stronger notion of calibration), a substantially weaker and more tractable condition than full calibration. For calibration guarantees that fall short of decision calibration, the minimax optimal decision rule is still efficiently computable, and we provide an empirical evaluation of a natural one that applies to any regression model solved to optimize squared error. E (X,Y )∼D [u(a(f (X)), Y )],
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 be1149fc-3f7d-437c-ae5f-fa29d247bfd6Builds on8
- 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
- Top-label calibration and multiclass-to-binary reductionsChirag Gupta, Aaditya RamdasICLR 2022 · 51 citations
- Near-Optimal Algorithms for OmnipredictionPrincewill Okoroafor, Robert Kleinberg, Michael P. KimFOCS 2025 · 37 citations
- Multicalibration as Boosting for RegressionIra Globus-Harris, Declan Harrison, Michael Kearns, Aaron Roth et al.ICML 2023 · 36 citations
- Outcome indistinguishabilityCynthia Dwork, Michael P. Kim, Omer Reingold, Guy N. Rothblum et al.STOC 2021 · 24 citations
Related papers
- Human-Aligned Calibration for AI-Assisted Decision MakingNina Corvelo Benz, Manuel Gomez RodriguezNeurIPS 2023 · 45 citations
- Reliable Decisions with Threshold CalibrationRoshni Sahoo, Shengjia Zhao, Alyssa Chen, Stefano ErmonNeurIPS 2021 · 35 citations
- Decision Theoretic Foundations for Conformal Prediction: Optimal Uncertainty Quantification for Risk-Averse AgentsShayan Kiyani, George J. Pappas, Aaron Roth, Hamed HassaniICML 2025
- Calibration by Distribution Matching: Trainable Kernel Calibration MetricsCharlie Marx, Sofian Zalouk, Stefano ErmonNeurIPS 2023 · 21 citations
- Individual Calibration with Randomized ForecastingShengjia Zhao, Tengyu Ma, Stefano ErmonICML 2020 · 69 citations
