Conformal prediction for multi-dimensional time series by ellipsoidal sets

Chen Xu, Hanyang Jiang, Yao Xie
3/6/2024
stat.MLcs.LG

No official code found. Best implementation: hamrel-cxu/MultiDimSPCI (21 stars, MIT, last commit 2024-07-16).

Abstract

Conformal prediction (CP) has been a popular method for uncertainty quantification because it is distribution-free, model-agnostic, and theoretically sound. For forecasting problems in supervised learning, most CP methods focus on building prediction intervals for univariate responses. In this work, we develop a sequential CP method called $\texttt{MultiDimSPCI}$ that builds prediction $\textit{regions}$ for a multivariate response, especially in the context of multivariate time series, which are not exchangeable. Theoretically, we estimate $\textit{finite-sample}$ high-probability bounds on the conditional coverage gap. Empirically, we demonstrate that $\texttt{MultiDimSPCI}$ maintains valid coverage on a wide range of multivariate time series while producing smaller prediction regions than CP and non-CP baselines.

View on arXivView PDF

Code Implementations(1)

MIT

Official implementation of "Conformal prediction for multi-dimensional time series by ellipsoidal sets" (ICML 2024 spotlight)

218Python, Jupyter NotebookMar 6, 20242 years agoMIT

Similar Papers

Cite this paper

@article{xu2024conformal,
  title  = {Conformal prediction for multi-dimensional time series by ellipsoidal sets},
  author = {Chen Xu and Hanyang Jiang and Yao Xie},
  year   = {2024},
  eprint = {2403.03850},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2403.03850v2},
  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/2403.03850.svg)](https://codeofpaper.com/paper/2403.03850)

Discussion