590
v1v2v3v4 (latest)

Conformal Risk Control

International Conference on Learning Representations (ICLR), 2022
Main:11 Pages
4 Figures
Bibliography:3 Pages
Appendix:5 Pages
Abstract

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 O(1/n)\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.

View on arXiv
Comments on this paper