RSA-CP: Efficient Conformal Prediction in Small-Sample Regimes via Random Score Alignment
Pankaj Bhagwat, Zhixian Yang, yihao wang, Bei Jiang, Linglong Kong
Abstract
Conformal Prediction (CP) provides rigorous finite-sample coverage guarantees, yet its statistical efficiency hinges critically on the size of the calibration set. In data-scarce regimes, CP often suffers from volatile quantile estimation, leading to overly conservative and wide prediction intervals. To address this, we propose Random Score Alignment-Conformal Prediction (RSA-CP), a simple framework designed to improve sample efficiency in small-sample CP. Instead of requiring the computationally intensive generation of full synthetic datasets, RSA-CP enhances calibration by directly aligning real scores with a high-resolution reference score distribution. By employing an optimal transport mapping, our framework refines "step-like" quantile increments through a globally optimal use of reference information. We provide theoretical guarantees establishing that RSA-CP maintains robust coverage without any distributional assumptions on the reference scores. Empirical evaluations demonstrate that RSA-CP consistently produces shorter and more precise prediction intervals while maintaining finite-sample coverage guarantees. Overall, RSA-CP offers a computationally efficient and theoretically grounded solution for robust uncertainty quantification under limited data.
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 on9
- Adaptive Conformal Predictions for Time SeriesMargaux Zaffran, Olivier Féron, Yannig Goude, Julie Josse et al.ICML 2022 · 209 citations
- Class-Conditional Conformal Prediction with Many ClassesTiffany Ding, Anastasios Angelopoulos, Stephen Bates, Michael I. Jordan et al.NeurIPS 2023 · 160 citations
- PAC Confidence Sets for Deep Neural Networks via Calibrated PredictionSangdon Park, Osbert Bastani, Nikolai Matni, Insup LeeICLR 2020 · 77 citations
- Stratified Prediction-Powered Inference for Effective Hybrid Evaluation of Language ModelsAdam Fisch, Joshua Maynez, R. Alex Hofer, Bhuwan Dhingra et al.NeurIPS 2024 · 27 citations
- Synthetic-powered predictive inferenceMeshi Bashari, Roy Maor Lotan, Yonghoon Lee, Edgar Dobriban et al.NeurIPS 2025 · 12 citations
Related papers
- Distribution-informed Online Conformal PredictionDongjian Hu, Junxi Wu, Shu-Tao Xia, Changliang ZouICLR 2026 · 2 citations
- Non-exchangeable Conformal Prediction with Optimal Transport: Tackling Distribution Shift with Unlabeled DataAlvaro H. C. Correia, Christos LouizosNeurIPS 2025 · 5 citations
- Robust Bayes-Assisted Conformal PredictionKianoosh Ashouritaklimi, Stefano Cortinovis, Francois CaronICML 2026
- Calibrating Decision Robustness via Inverse Conformal Risk ControlWenbin Zhou, Shixiang ZhuICML 2026
- Optimal transport-based conformal predictionGauthier Thurin, Kimia Nadjahi, Claire BoyerICML 2025
