Few-shot Conformal Prediction with Auxiliary Tasks
Not on the map. Why?
Abstract (arXiv)
We develop a novel approach to conformal prediction when the target task has limited data available for training. Conformal prediction identifies a small set of promising output candidates in place of a single prediction, with guarantees that the set contains the correct answer with high probability. When training data is limited, however, the predicted set can easily become unusably large. In this work, we obtain substantially tighter prediction sets while maintaining desirable marginal guarantees by casting conformal prediction as a meta-learning paradigm over exchangeable collections of auxiliary tasks. Our conformalization algorithm is simple, fast, and agnostic to the choice of underlying model, learning algorithm, or dataset. We demonstrate the effectiveness of this approach across a number of few-shot classification and regression tasks in natural language processing, computer vision, and computational chemistry for drug discovery.
Key facts. Cited by 53 papers in the map (data through 7 Oct 2026). Builds on 28 of its 48 references that are in the map. First posted on arXiv 17 Feb 2021.
Builds on
28 of 48 references.
References · 28
- 2017Model-Agnostic Meta-Learning for Fast Adaptation of Deep Networkscited by 4,861
- 2017On Calibration of Modern Neural Networkscited by 4,860
- 2015Dropout as a Bayesian Approximation: Representing Model Uncertainty in Deep Learningcited by 4,416
- 2016Simple and Scalable Predictive Uncertainty Estimation using Deep Ensemblescited by 3,888
- 2017Prototypical Networks for Few-shot Learningcited by 3,001
- 2016Matching Networks for One Shot Learningcited by 2,246
- 2016Concrete Problems in AI Safetycited by 1,582
- 2017Deep Setscited by 1,394
- 2016Distribution-Free Predictive Inference For Regressioncited by 657
- 2019Conformalized Quantile Regressioncited by 620
- 2019Conformal Prediction Under Covariate Shiftcited by 511
- 2019Generalizing from a Few Examples: A Survey on Few-Shot Learningcited by 462
16 more
- 2018Meta-learning with differentiable closed-form solverscited by 395
- 2017Training Confidence-calibrated Classifiers for Detecting Out-of-Distribution Samplescited by 389
- 2019Analyzing Learned Molecular Representations for Property Predictioncited by 373
- 2020Uncertainty Sets for Image Classifiers using Conformal Predictioncited by 365
- 2015Probabilistic Backpropagation for Scalable Learning of Bayesian Neural Networkscited by 295
- 2021Distribution-Free, Risk-Controlling Prediction Setscited by 278
- 2018FewRel: A Large-Scale Supervised Few-Shot Relation Classification Dataset with State-of-the-Art Evaluationcited by 254
- 2020Pitfalls of In-Domain Uncertainty Estimation and Ensembling in Deep Learningcited by 208
- 2018To Trust Or Not To Trust A Classifiercited by 205
- 2016Towards a Neural Statisticiancited by 119
- 2019Distributional conformal predictioncited by 104
- 2020Uncertainty Quantification Using Neural Networks for Molecular Property Predictioncited by 66
- 2019Few-shot Text Classification with Distributional Signaturescited by 58
- 2019Adaptive, Distribution-Free Prediction Intervals for Deep Networkscited by 56
- 2019With Malice Towards None: Assessing Uncertainty via Equalized Coveragecited by 43
- 2020Knowing what you know: valid and validated confidence sets in multiclass and multilabel predictioncited by 9
Built on by
Citing papers per year: 2021: 7; 2022: 12; 2023: 13; 2024: 8; 2025: 10; 2026: 3.
53 papers.
Sort:
By area:
- 2021A Gentle Introduction to Conformal Prediction and Distribution-Free Uncertainty Quantificationcited by 777
- 2021Learn then Test: Calibrating Predictive Algorithms to Achieve Risk Controlcited by 256
- 2023Conformal Language Modelingcited by 163
- 2021Consistent Accelerated Inference via Confident Adaptive Transformerscited by 73
- 2023Conformal Prediction for Deep Classifier via Label Rankingcited by 52
- 2024Conformal Prediction for Natural Language Processing: A Surveycited by 52
- 2023Federated Conformal Predictors for Distributed Uncertainty Quantificationcited by 46
- 2021PAC Prediction Sets Under Covariate Shiftcited by 42
- 2022Conformal Prediction Sets with Limited False Positivescited by 28
- 2022Conformal Prediction with Temporal Quantile Adjustmentscited by 25
- 2024Conformal Prediction: A Data Perspectivecited by 25
- 2021Conformal prediction for text infilling and part-of-speech predictioncited by 22
38 more
- 2022Semantic uncertainty intervals for disentangled latent spacescited by 22
- 2024Non-Exchangeable Conformal Language Generation with Nearest Neighborscited by 21
- 2025Multivariate Conformal Prediction using Optimal Transportcited by 21
- 2022Split Localized Conformal Predictioncited by 20
- 2021Stable Conformal Prediction Setscited by 19
- 2024Conformal Prediction for Class-wise Coverage via Augmented Label Rank Calibrationcited by 18
- 2022Few-Shot Calibration of Set Predictors via Meta-Learned Cross-Validation-Based Conformal Predictioncited by 17
- 2022PAC Prediction Sets for Meta-Learningcited by 16
- 2023Conformal Autoregressive Generation: Beam Search with Coverage Guaranteescited by 16
- 2023Selective Generation for Controllable Language Modelscited by 15
- 2022Recommendation Systems with Distribution-Free Reliability Guaranteescited by 13
- 2023On the Expected Size of Conformal Prediction Setscited by 13
- 2024Semi-Supervised Risk Control via Prediction-Powered Inferencecited by 13
- 2023Estimating Uncertainty in Multimodal Foundation Models using Public Internet Datacited by 10
- 2025Synthetic-Powered Predictive Inferencecited by 9
- 2021Root-finding Approaches for Computing Conformal Prediction Setcited by 8
- 2023Knowing When to Stop: Delay-Adaptive Spiking Neural Network Classifiers with Reliability Guaranteescited by 8
- 2023SymmPI: Predictive Inference for Data with Group Symmetriescited by 8
- 2025Direct Prediction Set Minimization via Bilevel Conformal Classifier Trainingcited by 8
- 2023Empirically Validating Conformal Prediction on Modern Vision Architectures Under Distribution Shift and Long-tailed Datacited by 7
- 2022Calibration of Natural Language Understanding Models with Venn--ABERS Predictorscited by 6
- 2023Conformalizing Machine Translation Evaluationcited by 5
- 2024Coverage-Guaranteed Prediction Sets for Out-of-Distribution Datacited by 5
- 2022Distribution-Free Finite-Sample Guarantees and Split Conformal Predictioncited by 4
- 2023Conformalization of Sparse Generalized Linear Modelscited by 4
- 2024Uncertainty Quantification for Neurosymbolic Programs via Compositional Conformal Predictioncited by 4
- 2024CHAMP: Conformalized 3D Human Multi-Hypothesis Pose Estimatorscited by 4
- 2025Adaptive Coverage Policies in Conformal Predictioncited by 4
- 2022A novel Deep Learning approach for one-step Conformal Prediction approximationcited by 3
- 2023Conformal Loss-Controlling Predictioncited by 3
- 2022ACon2: Adaptive Conformal Consensus for Provable Blockchain Oraclescited by 2
- 2025MetaSTNet: Multimodal Meta-learning for Cellular Traffic Conformal Predictioncited by 2
- 2025Another Fit Bites the Dust: Conformal Prediction as a Calibration Standard for Machine Learning in High-Energy Physicscited by 2
- 2022A Confidence Machine for Sparse High-Order Interaction Modelcited by 1
- 2025Semi-Supervised Conformal Prediction With Unlabeled Nonconformity Scorecited by 1
- 2025Optimizing In-Context Learning for Efficient Full Conformal Predictioncited by 1
- 2026CAOS: Conformal Aggregation of One-Shot Predictorscited by 1
- 2025Adept: Annotation-Denoising Auxiliary Tasks with Discrete Cosine Transform Map and Keypoint for Human-Centric Pretrainingcited by 0
50 most cited of 53 All 53 (JSON)
- 2026Stable Localized Conformal Prediction via Transduction2 May 2026
- 2026Softmax is not Enough (for Adaptive Conformal Classification)23 Feb 2026
- 2026CAOS: Conformal Aggregation of One-Shot Predictors8 Jan 2026
- 2025Another Fit Bites the Dust: Conformal Prediction as a Calibration Standard for Machine Learning in High-Energy Physics18 Dec 2025
- 2025Cost-Sensitive Conformal Training with Provably Controllable Learning Bounds22 Nov 2025
- 2025Adaptive Coverage Policies in Conformal Prediction5 Oct 2025
- 2025Optimizing In-Context Learning for Efficient Full Conformal Prediction1 Sep 2025
- 2025Direct Prediction Set Minimization via Bilevel Conformal Classifier Training7 Jun 2025
- 2025Semi-Supervised Conformal Prediction With Unlabeled Nonconformity Score27 May 2025
- 2025MetaSTNet: Multimodal Meta-learning for Cellular Traffic Conformal Prediction26 May 2025
- 2025Synthetic-Powered Predictive Inference19 May 2025
- 2025Adept: Annotation-Denoising Auxiliary Tasks with Discrete Cosine Transform Map and Keypoint for Human-Centric Pretraining29 Apr 2025
38 more
- 2025Multivariate Conformal Prediction using Optimal Transport5 Feb 2025
- 2024Semi-Supervised Risk Control via Prediction-Powered Inference15 Dec 2024
- 2024Conformal Prediction: A Data Perspective9 Oct 2024
- 2024Conformal Prediction for Class-wise Coverage via Augmented Label Rank Calibration10 Jun 2024
- 2024CHAMP: Conformalized 3D Human Multi-Hypothesis Pose Estimators27 May 2024
- 2024Uncertainty Quantification for Neurosymbolic Programs via Compositional Conformal Prediction24 May 2024
- 2024Conformal Prediction for Natural Language Processing: A Survey3 May 2024
- 2024Coverage-Guaranteed Prediction Sets for Out-of-Distribution Data29 Mar 2024
- 2024Non-Exchangeable Conformal Language Generation with Nearest Neighbors1 Feb 2024
- 2023SymmPI: Predictive Inference for Data with Group Symmetries26 Dec 2023
- 2023Estimating Uncertainty in Multimodal Foundation Models using Public Internet Data15 Oct 2023
- 2023Conformal Prediction for Deep Classifier via Label Ranking10 Oct 2023
- 2023Conformal Autoregressive Generation: Beam Search with Coverage Guarantees7 Sep 2023
- 2023Selective Generation for Controllable Language Models18 Jul 2023
- 2023Conformalization of Sparse Generalized Linear Models11 Jul 2023
- 2023Empirically Validating Conformal Prediction on Modern Vision Architectures Under Distribution Shift and Long-tailed Data3 Jul 2023
- 2023Conformal Language Modeling16 Jun 2023
- 2023On the Expected Size of Conformal Prediction Sets12 Jun 2023
- 2023Conformalizing Machine Translation Evaluation9 Jun 2023
- 2023Federated Conformal Predictors for Distributed Uncertainty Quantification27 May 2023
- 2023Knowing When to Stop: Delay-Adaptive Spiking Neural Network Classifiers with Reliability Guarantees18 May 2023
- 2023Conformal Loss-Controlling Prediction6 Jan 2023
- 2022ACon2: Adaptive Conformal Consensus for Provable Blockchain Oracles17 Nov 2022
- 2022Distribution-Free Finite-Sample Guarantees and Split Conformal Prediction26 Oct 2022
- 2022Few-Shot Calibration of Set Predictors via Meta-Learned Cross-Validation-Based Conformal Prediction6 Oct 2022
- 2022A novel Deep Learning approach for one-step Conformal Prediction approximation25 Jul 2022
- 2022Semantic uncertainty intervals for disentangled latent spaces20 Jul 2022
- 2022PAC Prediction Sets for Meta-Learning6 Jul 2022
- 2022Recommendation Systems with Distribution-Free Reliability Guarantees4 Jul 2022
- 2022Split Localized Conformal Prediction27 Jun 2022
- 2022A Confidence Machine for Sparse High-Order Interaction Model28 May 2022
- 2022Calibration of Natural Language Understanding Models with Venn--ABERS Predictors21 May 2022
- 2022Conformal Prediction with Temporal Quantile Adjustments20 May 2022
- 2022Conformal Prediction Sets with Limited False Positives15 Feb 2022
- 2021Stable Conformal Prediction Sets19 Dec 2021
- 2021Conformal prediction for text infilling and part-of-speech prediction4 Nov 2021
- 2021Learn then Test: Calibrating Predictive Algorithms to Achieve Risk Control3 Oct 2021
- 2021A Gentle Introduction to Conformal Prediction and Distribution-Free Uncertainty Quantification15 Jul 2021
50 newest of 53 All 53 (JSON)
- 2021A Gentle Introduction to Conformal Prediction and Distribution-Free Uncertainty Quantificationcited by 777
- 2021Learn then Test: Calibrating Predictive Algorithms to Achieve Risk Controlcited by 256
- 2023Conformal Language Modelingcited by 163
- 2021Consistent Accelerated Inference via Confident Adaptive Transformerscited by 73
- 2023Conformal Prediction for Deep Classifier via Label Rankingcited by 52
- 2024Conformal Prediction for Natural Language Processing: A Surveycited by 52
- 2023Federated Conformal Predictors for Distributed Uncertainty Quantificationcited by 46
- 2021PAC Prediction Sets Under Covariate Shiftcited by 42
- 2022Conformal Prediction Sets with Limited False Positivescited by 28
- 2022Conformal Prediction with Temporal Quantile Adjustmentscited by 25
- 2024Conformal Prediction: A Data Perspectivecited by 25
- 2021Conformal prediction for text infilling and part-of-speech predictioncited by 22
12 most cited of 52 All 52 (JSON)
- 2026Stable Localized Conformal Prediction via Transduction2 May 2026
- 2026Softmax is not Enough (for Adaptive Conformal Classification)23 Feb 2026
- 2026CAOS: Conformal Aggregation of One-Shot Predictors8 Jan 2026
- 2025Another Fit Bites the Dust: Conformal Prediction as a Calibration Standard for Machine Learning in High-Energy Physics18 Dec 2025
- 2025Cost-Sensitive Conformal Training with Provably Controllable Learning Bounds22 Nov 2025
- 2025Adaptive Coverage Policies in Conformal Prediction5 Oct 2025
- 2025Optimizing In-Context Learning for Efficient Full Conformal Prediction1 Sep 2025
- 2025Direct Prediction Set Minimization via Bilevel Conformal Classifier Training7 Jun 2025
- 2025Semi-Supervised Conformal Prediction With Unlabeled Nonconformity Score27 May 2025
- 2025MetaSTNet: Multimodal Meta-learning for Cellular Traffic Conformal Prediction26 May 2025
- 2025Synthetic-Powered Predictive Inference19 May 2025
- 2025Adept: Annotation-Denoising Auxiliary Tasks with Discrete Cosine Transform Map and Keypoint for Human-Centric Pretraining29 Apr 2025
12 newest of 52 All 52 (JSON)