Length Optimization in Conformal Prediction
This work addresses a key gap in conformal prediction for researchers and practitioners needing reliable uncertainty quantification with efficient prediction sets, representing a novel integration rather than an incremental improvement.
The paper tackles the problem of reconciling conditional validity and length efficiency in conformal prediction by developing the CPL framework, which constructs prediction sets with near-optimal length while ensuring conditional validity under covariate shifts, achieving superior performance in empirical evaluations across diverse datasets.
Conditional validity and length efficiency are two crucial aspects of conformal prediction (CP). Conditional validity ensures accurate uncertainty quantification for data subpopulations, while proper length efficiency ensures that the prediction sets remain informative. Despite significant efforts to address each of these issues individually, a principled framework that reconciles these two objectives has been missing in the CP literature. In this paper, we develop Conformal Prediction with Length-Optimization (CPL) - a novel and practical framework that constructs prediction sets with (near-) optimal length while ensuring conditional validity under various classes of covariate shifts, including the key cases of marginal and group-conditional coverage. In the infinite sample regime, we provide strong duality results which indicate that CPL achieves conditional validity and length optimality. In the finite sample regime, we show that CPL constructs conditionally valid prediction sets. Our extensive empirical evaluations demonstrate the superior prediction set size performance of CPL compared to state-of-the-art methods across diverse real-world and synthetic datasets in classification, regression, and large language model-based multiple choice question answering. An Implementation of our algorithm can be accessed at the following link: https://github.com/shayankiyani98/CP.