Conformal Prediction under Levy-Prokhorov Distribution Shifts: Robustness to Local and Global Perturbations

Liviu Aolaritei, Zheyu Oliver Wang, Julie Zhu, Michael I. Jordan, Youssef Marzouk
2/19/2025
stat.MLcs.LGmath.ST

No official code found. Best implementation: olivrw/LP-robust-conformal (1 star, MIT, last commit 2025-05-17).

Abstract

Conformal prediction provides a powerful framework for constructing prediction intervals with finite-sample guarantees, yet its robustness under distribution shifts remains a significant challenge. This paper addresses this limitation by modeling distribution shifts using Levy-Prokhorov (LP) ambiguity sets, which capture both local and global perturbations. We provide a self-contained overview of LP ambiguity sets and their connections to popular metrics such as Wasserstein and Total Variation. We show that the link between conformal prediction and LP ambiguity sets is a natural one: by propagating the LP ambiguity set through the scoring function, we reduce complex high-dimensional distribution shifts to manageable one-dimensional distribution shifts, enabling exact quantification of worst-case quantiles and coverage. Building on this analysis, we construct robust conformal prediction intervals that remain valid under distribution shifts, explicitly linking LP parameters to interval width and confidence levels. Experimental results on real-world datasets demonstrate the effectiveness of the proposed approach.

View on arXivView PDF

Code Implementations(1)

MIT

Robust conformal prediction under Levy-Prokhorov Distribution Shifts

11Cython, PythonJan 20, 20251 years agoMIT

Similar Papers

Cite this paper

@article{aolaritei2025conformal,
  title  = {Conformal Prediction under Levy-Prokhorov Distribution Shifts: Robustness to Local and Global Perturbations},
  author = {Liviu Aolaritei and Zheyu Oliver Wang and Julie Zhu and Michael I. Jordan and Youssef Marzouk},
  year   = {2025},
  eprint = {2502.14105},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2502.14105v2}
}

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/2502.14105.svg)](https://codeofpaper.com/paper/2502.14105)

Discussion