Valid Selection among Conformal Sets

Liviu Aolaritei (UC Berkeley) · Michael Jordan (University of California, Berkeley) · Mahmoud Hegazy (École Polytechnique) · Aymeric Dieuleveut (Ecole Polytechnique, IPParis)
algorithmic stabilityconformal predictioncoverage guaranteesdistribution-free frameworkempirical evaluationmethodology comparisonmodel selectiononline conformal settingprediction set refinementprediction setsstability-based approachstatistical inferencestructural constraintsuncertainty quantificationvalidity of coverage

Conformal prediction offers a distribution-free framework for constructing prediction sets with coverage guarantees. In practice, multiple valid conformal prediction sets may be available, arising from different models or methodologies. However, selecting the most desirable set, such as the smallest, can invalidate the coverage guarantees. To address this challenge, we propose a stability-based approach that ensures coverage for the selected prediction set. We extend our results to the online conformal setting, propose several refinements in settings where additional structure is available, and demonstrate its effectiveness through experiments.