CAOS: Conformal Aggregation of One-Shot Predictors
Maja Waldron
Abstract
One-shot prediction enables rapid adaptation of pretrained foundation models to new tasks using only one labeled example, but lacks principled uncertainty quantification. While conformal prediction provides finite-sample coverage guarantees, standard split conformal methods are inefficient in the one-shot setting due to data splitting and reliance on a single predictor. We propose Conformal Aggregation of One-Shot Predictors (CAOS), a conformal framework that adaptively aggregates multiple one-shot predictors and uses a leave-one-out calibration scheme to fully exploit scarce labeled data. Despite violating classical exchangeability assumptions, we prove that CAOS achieves valid marginal coverage using a monotonicity-based argument. Experiments on one-shot facial landmarking and RAFT text classification tasks show that CAOS produces substantially smaller prediction sets than split conformal baselines while maintaining reliable coverage.
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.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext c7ddfc2e-7009-4b2c-bf27-ebfa5624ea38Builds on12
- Language Models are Few-Shot LearnersTom B. Brown, Benjamin Mann, Nick Ryder, Melanie Subbiah et al.NeurIPS 2020 · 64,255 citations
- Learning Transferable Visual Models From Natural Language SupervisionAlec Radford, Jong Wook Kim, Chris Hallacy, Aditya Ramesh et al.ICML 2021 · 47,906 citations
- Conformal Language ModelingVictor Quach, Adam Fisch, Tal Schuster, Adam Yala et al.ICLR 2024 · 132 citations
- Few-Shot Conformal Prediction with Auxiliary TasksAdam Fisch, Tal Schuster, Tommi S. Jaakkola, Regina BarzilayICML 2021 · 66 citations
- Synthetic-powered predictive inferenceMeshi Bashari, Roy Maor Lotan, Yonghoon Lee, Edgar Dobriban et al.NeurIPS 2025 · 12 citations
Related papers
- Split conformal classification with unsupervised calibrationSantiago MazuelasNeurIPS 2025 · 1 citation
- Multi-model Ensemble Conformal Prediction in Dynamic EnvironmentsErfan Hajihashemi, Yanning ShenNeurIPS 2024 · 13 citations
- Coverage-Guaranteed Prediction Sets for Out-of-Distribution DataXin Zou, Weiwei LiuAAAI 2024 · 5 citations
- Model Uncertainty Quantification by Conformal Prediction in Continual LearningRui Gao, Weiwei LiuICML 2025
- Conformal Prediction for Zero-Shot ModelsJulio Silva-Rodríguez, Ismail Ben Ayed, Jose DolzCVPR 2025
