Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Conformal Risk Control

About

We extend conformal prediction to control the expected value of any monotone loss function. The algorithm generalizes split conformal prediction together with its coverage guarantee. Like conformal prediction, the conformal risk control procedure is tight up to an $\mathcal{O}(1/n)$ factor. We also introduce extensions of the idea to distribution shift, quantile risk control, multiple and adversarial risk control, and expectations of U-statistics. Worked examples from computer vision and natural language processing demonstrate the usage of our algorithm to bound the false negative rate, graph distance, and token-level F1-score.

Anastasios N. Angelopoulos, Stephen Bates, Adam Fisch, Lihua Lei, Tal Schuster• 2022

Related benchmarks

TaskDatasetResultRank
Conformal PredictionCIFAR-100
Avg Prediction Set Size2.7219
32
Reinforcement Learning from Verifiable RewardsHEAD-QA
AR16.5
30
Medical Image SegmentationMSD Pancreas (test)
DSC45.19
30
Medical Image SegmentationCAMUS (test)
DSC81.07
22
Medical Image SegmentationACDC-LV (test)
Coverage99.8
10
Medical Image SegmentationACDC-RV (test)
Coverage99.8
10
Distribution Shift RobustnessSixteen Adversarial Cells MedQA + GSM8K (eval)
Violations4
10
Expert-Iteration RLVRMedQA, HEAD-QA, ARC-C, and CaseHOLD
Pathwise Clean Score4
10
Mathematical ReasoningGSM8K
AR (%)28.9
10
Natural Language InferencemedNLI
AR (%)28.9
10
Showing 10 of 20 rows

Other info

Follow for update