# Group-Weighted Conformal Prediction

arXiv: https://arxiv.org/abs/2401.17452 · HTML: https://mlsanity.com/paper/2401.17452/

## Key facts

- Cited by 14 papers in the map (data through 14 Oct 2026).
- First posted on arXiv 30 Jan 2024.

## Built on by (most cited first, 14 of 14)

- [Conformal Prediction: A Data Perspective](https://mlsanity.com/paper/2410.06494/) (2024)
- [Conformal Inference under High-Dimensional Covariate Shifts via Likelihood-Ratio Regularization](https://mlsanity.com/paper/2502.13030/) (2025)
- [Conditional Conformal Risk Adaptation](https://mlsanity.com/paper/2504.07611/) (2025)
- [Weight Clipping for Robust Conformal Inference under Unbounded Covariate Shifts](https://mlsanity.com/paper/2605.02072/) (2026)
- [Conformal Prediction with Corrupted Labels: Uncertain Imputation and Robust Re-weighting](https://mlsanity.com/paper/2505.04733/) (2025)
- [Uncertainty Quantification With Multiple Sources](https://mlsanity.com/paper/2405.06479/) (2024)
- [GeoConformal prediction: a model-agnostic framework of measuring the uncertainty of spatial prediction](https://mlsanity.com/paper/2412.08661/) (2024)
- [Distribution-Free Predictive Inference under Unknown Temporal Drift](https://mlsanity.com/paper/2406.06516/) (2024)
- [Attention-Based Feature Online Conformal Prediction for Time Series](https://mlsanity.com/paper/2511.15838/) (2025)
- [Conformal Prediction with Macro-Coverage Guarantees](https://mlsanity.com/paper/2606.28598/) (2026)
- [Self-Organized Conformal Prediction: Reducing Regional Coverage Gaps with Unsupervised Group Discovery](https://mlsanity.com/paper/2606.29403/) (2026)
- [The Exceedance Design Effect: Effective Sample Size for Thresholds under Clustering](https://mlsanity.com/paper/2608.21262/) (2026)
- [CHOIR: heterogeneity-aware conformal prediction for crash injury severity across driver safety strata](https://mlsanity.com/paper/2609.11592/) (2026)
- [Multi-source conformal prediction: leveraging heterogeneity via localization](https://mlsanity.com/paper/2609.14531/) (2026)

## Abstract (arXiv)

Conformal prediction (CP) is a method for constructing a prediction interval around the output of a fitted model, whose validity does not rely on the model being correct--the CP interval offers a coverage guarantee that is distribution-free, but relies on the training data being drawn from the same distribution as the test data. A recent variant, weighted conformal prediction (WCP), reweights the method to allow for covariate shift between the training and test distributions. However, WCP requires knowledge of the nature of the covariate shift-specifically,the likelihood ratio between the test and training covariate distributions. In practice, since this likelihood ratio is estimated rather than known exactly, the coverage guarantee may degrade due to the estimation error. In this paper, we consider a special scenario where observations belong to a finite number of groups, and these groups determine the covariate shift between the training and test distributions-for instance, this may arise if the training set is collected via stratified sampling. Our results demonstrate that in this special case, the predictive coverage guarantees of WCP can be drastically improved beyond the bounds given by existing estimation error bounds.
