Flow-based Conformal Prediction for Multi-dimensional Time Series
About
Time series prediction underpins a broad range of downstream tasks across many scientific domains. Recent advances and increasing adoption of black-box machine learning models for time series prediction highlight the critical need for uncertainty quantification. While conformal prediction has gained attention as a reliable uncertainty quantification method, conformal prediction for time series faces two key challenges: (1) \textbf{leveraging correlations in observations and non-conformity scores to overcome the exchangeability assumption}, and (2) \textbf{constructing prediction sets for multi-dimensional outcomes}. To address these challenges, we propose a novel conformal prediction method for time series using flow with classifier-free guidance. We provide coverage guarantees by establishing exact non-asymptotic marginal coverage and a finite-sample bound on conditional coverage for the proposed method. Evaluations on real-world time series datasets demonstrate that our method constructs significantly smaller prediction sets than existing conformal prediction methods, maintaining target coverage.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Multivariate Conformal Prediction | Traffic dy=2 (test) | Coverage96.8 | 20 | |
| Multivariate Conformal Prediction | Traffic dy=4 (test) | Coverage96.6 | 20 | |
| Multivariate Conformal Prediction | Traffic dy=8 (test) | Coverage96.5 | 20 | |
| Multivariate Conformal Prediction | Wind dy=8 (test) | Coverage95.6 | 20 | |
| Multivariate Conformal Prediction | Wind dy=4 (test) | Coverage95.7 | 20 | |
| Multivariate Conformal Prediction | Solar dy=4 (test) | Coverage96.9 | 20 | |
| Multivariate Conformal Prediction | Wind dy=2 (test) | Coverage95.2 | 20 | |
| Multivariate Conformal Prediction | Solar dy=2 (test) | Coverage96.8 | 20 |