Is this model reliable for everyone? Testing for strong calibration
Not on the map. Why?
Abstract (arXiv)
In a well-calibrated risk prediction model, the average predicted probability is close to the true event rate for any given subgroup. Such models are reliable across heterogeneous populations and satisfy strong notions of algorithmic fairness. However, the task of auditing a model for strong calibration is well-known to be difficult -- particularly for machine learning (ML) algorithms -- due to the sheer number of potential subgroups. As such, common practice is to only assess calibration with respect to a few predefined subgroups. Recent developments in goodness-of-fit testing offer potential solutions but are not designed for settings with weak signal or where the poorly calibrated subgroup is small, as they either overly subdivide the data or fail to divide the data at all. We introduce a new testing procedure based on the following insight: if we can reorder observations by their expected residuals, there should be a change in the association between the predicted and observed residuals along this sequence if a poorly calibrated subgroup exists. This lets us reframe the problem of calibration testing into one of changepoint detection, for which powerful methods already exist. We begin with introducing a sample-splitting procedure where a portion of the data is used to train a suite of candidate models for predicting the residual, and the remaining data are used to perform a score-based cumulative sum (CUSUM) test. To further improve power, we then extend this adaptive CUSUM test to incorporate cross-validation, while maintaining Type I error control under minimal assumptions. Compared to existing methods, the proposed procedure consistently achieved higher power in simulation studies and more than doubled the power when auditing a mortality risk prediction model.
Key facts. Cited by 6 papers in the map (data through 14 Oct 2026). Builds on 27 of its 81 references that are in the map. First posted on arXiv 28 Jul 2023.
Builds on
27 of 81 references.
References · 27
- 2019Sentence-BERT: Sentence Embeddings using Siamese BERT-Networkscited by 9,170
- 2017A Unified Approach to Interpreting Model Predictionscited by 6,555
- 2016Equality of Opportunity in Supervised Learningcited by 2,007
- 2019Nuanced Metrics for Measuring Unintended Bias with Real Data for Text Classificationcited by 404
- 2018Learning Models with Uniform Performance via Distributionally Robust Optimizationcited by 335
- 2018Multiaccuracy: Black-Box Post-Processing for Fairness in Classificationcited by 222
- 2022Domino: Discovering Systematic Errors with Cross-Modal Embeddingscited by 156
- 2021Localized Conformal Prediction: A Generalized Inference Framework for Conformal Predictioncited by 114
- 2019Calibration tests in multi-class classification: A unifying frameworkcited by 79
- 2018Automated Data Slicing for Model Validation:A Big data - AI Integration Approachcited by 70
- 2020Learning Certified Individually Fair Representationscited by 69
- 2020Efficient nonparametric statistical inference on population feature importance using Shapley valuescited by 61
15 more
- 2020Distributionally Robust Losses for Latent Covariate Mixturescited by 58
- 2019Metric Learning for Individual Fairnesscited by 52
- 2020Individual Calibration with Randomized Forecastingcited by 50
- 2019With Malice Towards None: Assessing Uncertainty via Equalized Coveragecited by 43
- 2023Multicalibration as Boosting for Regressioncited by 33
- 2021Local Calibration: Metrics and Recalibrationcited by 24
- 2022T-Cal: An optimal test for the calibration of predictive modelscited by 24
- 2019Conformal calibratorscited by 22
- 2020Auditing ML Models for Individual Bias and Unfairnesscited by 18
- 2022Modular Conformal Calibrationcited by 15
- 2021Statistical inference for individual fairnesscited by 13
- 2019Is a Classification Procedure Good Enough? A Goodness-of-Fit Assessment Tool for Classification Learningcited by 9
- 2019Goodness-of-fit testing in high-dimensional generalized linear modelscited by 8
- 2021Inference on function-valued parameters using a restricted score testcited by 4
- 2022Monitoring machine learning (ML)-based risk prediction algorithms in the presence of confounding medical interventionscited by 2
Built on by
Citing papers per year: 2023: 1; 2024: 0; 2025: 4; 2026: 1.
6 papers.
- 2023Designing monitoring strategies for deployed machine learning algorithms: navigating performativity through a causal lenscited by 3
- 2026Adaptive auditing of AI systems with anytime-valid guaranteescited by 3
- 2025Algorithmic Accountability in Small Data: Sample-Size-Induced Bias Within Classification Metricscited by 1
- 2025"Who experiences large model decay and why?" A Hierarchical Framework for Diagnosing Heterogeneous Performance Driftcited by 1
- 2025On the Role of Calibration in Benchmarking Algorithmic Fairness for Skin Cancer Detectioncited by 1
- 2025Reliably Detecting Model Failures in Deployment Without Labelscited by 0
- 2026Adaptive auditing of AI systems with anytime-valid guarantees7 May 2026
- 2025On the Role of Calibration in Benchmarking Algorithmic Fairness for Skin Cancer Detection10 Nov 2025
- 2025Reliably Detecting Model Failures in Deployment Without Labels5 Jun 2025
- 2025"Who experiences large model decay and why?" A Hierarchical Framework for Diagnosing Heterogeneous Performance Drift31 May 2025
- 2025Algorithmic Accountability in Small Data: Sample-Size-Induced Bias Within Classification Metrics6 May 2025
- 2023Designing monitoring strategies for deployed machine learning algorithms: navigating performativity through a causal lens20 Nov 2023