SpeedCP: Fast Kernel-based Conditional Conformal Prediction
Yating Liu, Yeo Jin Jung, Zixuan Wu, So Won Jeong, Claire Donnat
Abstract
Conformal prediction provides distribution-free prediction sets with finite-sample conditional guarantees. RKHS-based frameworks—while promising for complex covariate shifts—suffer from prohibitive computational costs. To guarantee conditional validity under such shifts while ensuring feasibility, we build upon the framework of Gibbs et al. (2025) by introducing a stable and efficient algorithm that computes the full solution path of the regularized RKHS conformal optimization problem, at essentially the same cost as a single kernel quantile fit. Our approach provides simultaneous hyperparameter tuning which provides smoothness control and data-adaptive calibration. To extend the method to high-dimensional settings, we further integrate our approach with low-rank latent embeddings that capture conditional validity in a data-driven latent space. Empirically, our method provides reliable conditional coverage across a variety of modern black-box predictors, improving the interval length of Gibbs et al. (2025) by 30%, while achieving a 40-fold speedup.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Builds on4
- Conformal Prediction for Uncertainty-Aware Planning with Diffusion Dynamics ModelJiankai Sun, Yiqi Jiang, Jianing Qiu, Parth Nobel et al.NeurIPS 2023 · 79 citations
- Conformal prediction for multi-dimensional time series by ellipsoidal setsChen Xu, Hanyang Jiang, Yao XieICML 2024 · 43 citations
- Uncertainty Sets for Image Classifiers using Conformal PredictionAnastasios Nikolas Angelopoulos, Stephen Bates, Michael I. Jordan, Jitendra MalikICLR 2021 · 31 citations
- Batch Multivalid Conformal PredictionChristopher Jung, Georgy Noarov, Ramya Ramalingam, Aaron RothICLR 2023 · 1 citation
Related papers
- Adaptive Conformal Inference Under Distribution ShiftIsaac Gibbs, Emmanuel J. CandèsNeurIPS 2021 · 665 citations
- Scalable and adaptive prediction bands with kernel sum-of-squaresLouis Allain, Sébastien Da Veiga, Brian StaberNeurIPS 2025 · 1 citation
- Conformal Prediction using Conditional HistogramsMatteo Sesia, Yaniv RomanoNeurIPS 2021 · 106 citations
- Sample-Conditional Coverage in Split-Conformal PredictionJohn C. DuchiNeurIPS 2025 · 2 citations
- Probabilistic Conformal Prediction with Approximate Conditional ValidityVincent Plassier, Alexander Fishkov, Mohsen Guizani, Maxim Panov et al.ICLR 2025
