Conformal Validity Guarantees Exist for Any Data Distribution (and How to Find Them)

Drew Prinster, Samuel Stanton, Anqi Liu, Suchi Saria
5/10/2024
cs.LGcs.AIstat.ML

No official code found. Best implementation: drewprinster/conformal-mfcs (29 stars, MIT, last commit 2024-06-28).

Abstract

As artificial intelligence (AI) / machine learning (ML) gain widespread adoption, practitioners are increasingly seeking means to quantify and control the risk these systems incur. This challenge is especially salient when such systems have autonomy to collect their own data, such as in black-box optimization and active learning, where their actions induce sequential feedback-loop shifts in the data distribution. Conformal prediction is a promising approach to uncertainty and risk quantification, but prior variants' validity guarantees have assumed some form of ``quasi-exchangeability'' on the data distribution, thereby excluding many types of sequential shifts. In this paper we prove that conformal prediction can theoretically be extended to \textit{any} joint data distribution, not just exchangeable or quasi-exchangeable ones. Although the most general case is exceedingly impractical to compute, for concrete practical applications we outline a procedure for deriving specific conformal algorithms for any data distribution, and we use this procedure to derive tractable algorithms for a series of AI/ML-agent-induced covariate shifts. We evaluate the proposed algorithms empirically on synthetic black-box optimization and active learning tasks.

View on arXivView PDF

Code Implementations(1)

MIT

Code for multistep feedback covariate shift conformal prediction experiments in "Conformal Validity Guarantees Exist for Any Data Distribution (and How to Find Them)" (ICML 2024)

293Shell, PythonMay 7, 20242 years agoMIT

Similar Papers

Cite this paper

@article{prinster2024conformal,
  title  = {Conformal Validity Guarantees Exist for Any Data Distribution (and How to Find Them)},
  author = {Drew Prinster and Samuel Stanton and Anqi Liu and Suchi Saria},
  year   = {2024},
  eprint = {2405.06627},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2405.06627v3},
  journal = {ICML 2024 2024}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2405.06627.svg)](https://codeofpaper.com/paper/2405.06627)

Discussion