Individual Calibration with Randomized Forecasting
Shengjia Zhao, Tengyu Ma, Stefano Ermon
Abstract
Machine learning applications often require calibrated predictions, e.g. a 90% credible interval should contain the true outcome 90% of the times. However, typical definitions of calibration only require this to hold on average, and offer no guarantees on predictions made on individual samples. Thus, predictions can be systematically over or under confident on certain subgroups, leading to issues of fairness and potential vulnerabilities. We show that calibration for individual samples is possible in the regression setup if the predictions are randomized, i.e. outputting randomized credible intervals. Randomization removes systematic bias by trading off bias with variance. We design a training objective to enforce individual calibration and use it to train randomized regression functions. The resulting models are more calibrated for arbitrarily chosen subgroups of the data, and can achieve higher utility in decision making against adversaries that exploit miscalibrated predictions.
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 e825f044-39e2-461b-9ddf-b1c5a64b775cCited by top-tier papers26
- Beyond Pinball Loss: Quantile Methods for Calibrated Uncertainty QuantificationYoungseog Chung, Willie Neiswanger, Ian Char, Jeff SchneiderNeurIPS 2021 · 137 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
- When and How Mixup Improves CalibrationLinjun Zhang, Zhun Deng, Kenji Kawaguchi, James ZouICML 2022 · 79 citations
- Learning Barrier Certificates: Towards Safe Reinforcement Learning with Zero Training-time ViolationsYuping Luo, Tengyu MaNeurIPS 2021 · 58 citations
- Training-Free Uncertainty Estimation for Dense Regression: Sensitivity as a SurrogateLu Mi, Hao Wang, Yonglong Tian, Hao He et al.AAAI 2022 · 36 citations
Related papers
- When is Multicalibration Post-Processing Necessary?Dutch Hansen, Siddartha Devic, Preetum Nakkiran, Vatsal SharanNeurIPS 2024 · 21 citations
- Fair Conformal Classification via Learning Representation-Based GroupsSenrong Xu, Yanke Zhou, Yuhao Tan, Zenan Li et al.ICLR 2026 · 1 citation
- Calibrated Reliable Regression using Maximum Mean DiscrepancyPeng Cui, Wenbo Hu, Jun ZhuNeurIPS 2020 · 71 citations
- Distribution-Free Model-Agnostic Regression Calibration via Nonparametric MethodsShang Liu, Zhongze Cai, Xiaocheng LiNeurIPS 2023 · 5 citations
- Robust Decision-Making with Partially Calibrated ForecastersShayan Kiyani, Hamed Hassani, George J. Pappas, Aaron RothICLR 2026 · 1 citation
