Adapting Prediction Sets to Distribution Shifts Without Labels
Not on the map. Why?
Abstract (arXiv)
Recently there has been a surge of interest to deploy confidence set predictions rather than point predictions in machine learning. Unfortunately, the effectiveness of such prediction sets is frequently impaired by distribution shifts in practice, and the challenge is often compounded by the lack of ground truth labels at test time. Focusing on a standard set-valued prediction framework called conformal prediction (CP), this paper studies how to improve its practical performance using only unlabeled data from the shifted test domain. This is achieved by two new methods called ECP and EACP, whose main idea is to adjust the score function in CP according to its base model's own uncertainty evaluation. Through extensive experiments on a number of large-scale datasets and neural network architectures, we show that our methods provide consistent improvement over existing baselines and nearly match the performance of fully supervised methods.
Key facts. Cited by 6 papers in the map (data through 7 Oct 2026). Builds on 47 of its 63 references that are in the map. First posted on arXiv 3 Jun 2024.
Builds on
47 of 63 references.
References · 47
- 2016Densely Connected Convolutional Networkscited by 6,691
- 2017On Calibration of Modern Neural Networkscited by 4,860
- 2019Benchmarking Neural Network Robustness to Common Corruptions and Perturbationscited by 2,857
- 2016A Baseline for Detecting Misclassified and Out-of-Distribution Examples in Neural Networkscited by 2,362
- 2020The Many Faces of Robustness: A Critical Analysis of Out-of-Distribution Generalizationcited by 1,842
- 2019Do ImageNet Classifiers Generalize to ImageNet?cited by 1,507
- 2019Natural Adversarial Examplescited by 1,401
- 2020WILDS: A Benchmark of in-the-Wild Distribution Shiftscited by 1,366
- 2016Distribution-Free Predictive Inference For Regressioncited by 657
- 2019Conformalized Quantile Regressioncited by 620
- 2019Conformal Prediction Under Covariate Shiftcited by 511
- 2022Efficient Test-Time Model Adaptation without Forgettingcited by 467
35 more
- 2020Improving robustness against common corruptions by covariate shift adaptationcited by 427
- 2020Classification with Valid and Adaptive Coveragecited by 390
- 2021Revisiting the Calibration of Modern Neural Networkscited by 387
- 2021MEMO: Test Time Robustness via Adaptation and Augmentationcited by 376
- 2020Uncertainty Sets for Image Classifiers using Conformal Predictioncited by 365
- 2021Adaptive Conformal Inference Under Distribution Shiftcited by 360
- 2022Conformal prediction beyond exchangeabilitycited by 348
- 2017Functional Map of the Worldcited by 338
- 2023Robots That Ask For Help: Uncertainty Alignment for Large Language Model Plannerscited by 298
- 2016Least Ambiguous Set-Valued Classifiers with Bounded Error Levelscited by 270
- 2020Evaluating Prediction-Time Batch Normalization for Robustness under Covariate Shiftcited by 235
- 2023Conformal Language Modelingcited by 163
- 2023Conformal Prediction with Large Language Models for Multi-Choice Question Answeringcited by 131
- 2024Language Models with Conformal Factuality Guaranteescited by 128
- 2022Conformal Inference for Online Prediction with Arbitrary Distribution Shiftscited by 123
- 2023EcoTTA: Memory-Efficient Continual Test-time Adaptation via Self-distilled Regularizationcited by 123
- 2022Safe Planning in Dynamic Environments using Conformal Predictioncited by 112
- 2020The iWildCam 2020 Competition Datasetcited by 86
- 2023Conformal PID Control for Time Series Predictioncited by 85
- 2023Improved Online Conformal Prediction via Strongly Adaptive Online Learningcited by 76
- 2022Practical Adversarial Multivalid Conformal Predictioncited by 65
- 2019Flexible distribution-free conditional predictive bands using density estimatorscited by 64
- 2021SITA: Single Image Test-time Adaptationcited by 57
- 2023RDumb: A simple approach that questions our progress in continual test-time adaptationcited by 50
- 2024Conformal Alignment: Knowing When to Trust Foundation Models with Guaranteescited by 50
- 2023Conformal Decision Theory: Safe Autonomous Decisions from Imperfect Predictionscited by 46
- 2023Object Pose Estimation with Statistical Guarantees: Conformal Keypoint Detection and Geometric Uncertainty Propagationcited by 33
- 2021Set-valued classification -- overview via a unified frameworkcited by 27
- 2024The Entropy Enigma: Success and Failure of Entropy Minimizationcited by 24
- 2023Improving Adaptive Conformal Prediction Using Self-Supervised Learningcited by 17
- 2023Integrating Uncertainty Awareness into Conformalized Quantile Regressioncited by 17
- 2024Not all distributional shifts are equal: Fine-grained robust conformal inferencecited by 15
- 2023Empirically Validating Conformal Prediction on Modern Vision Architectures Under Distribution Shift and Long-tailed Datacited by 7
- 2024CLOSURE: Fast Quantification of Pose Uncertainty Setscited by 6
- 2024Optimal Aggregation of Prediction Intervals under Unsupervised Domain Shiftcited by 4
Built on by
Citing papers per year: 2026: 6.
6 papers.
- 2026Coverage Guarantees for Pseudo-Calibrated Conformal Prediction under Distribution Shiftcited by 1
- 2026Retrospective Counterfactual Prediction by Conditioning on the Factual Outcome: A Cross-World Approachcited by 1
- 2026Diagnosing Conformal Prediction Failures Under Distribution Shift: A COVID-19 Case Studycited by 0
- 2026DistMatch: Adaptive Binning via Distribution Matching for Robust Sequential Conformal Predictioncited by 0
- 2026The Label Complexity of Useful Class-Conditional Prediction Sets under Distribution Shiftcited by 0
- 2026Does Marginal Coverage Guarantee Class-Conditional Safety for Zero-Shot VLMs Under Shift?cited by 0
- 2026Does Marginal Coverage Guarantee Class-Conditional Safety for Zero-Shot VLMs Under Shift?19 Aug 2026
- 2026The Label Complexity of Useful Class-Conditional Prediction Sets under Distribution Shift20 Jul 2026
- 2026DistMatch: Adaptive Binning via Distribution Matching for Robust Sequential Conformal Prediction30 May 2026
- 2026Retrospective Counterfactual Prediction by Conditioning on the Factual Outcome: A Cross-World Approach28 Mar 2026
- 2026Coverage Guarantees for Pseudo-Calibrated Conformal Prediction under Distribution Shift16 Feb 2026
- 2026Diagnosing Conformal Prediction Failures Under Distribution Shift: A COVID-19 Case Study1 Jan 2026