Improved Online Conformal Prediction via Strongly Adaptive Online Learning
About
We study the problem of uncertainty quantification via prediction sets, in an online setting where the data distribution may vary arbitrarily over time. Recent work develops online conformal prediction techniques that leverage regret minimization algorithms from the online learning literature to learn prediction sets with approximately valid coverage and small regret. However, standard regret minimization could be insufficient for handling changing environments, where performance guarantees may be desired not only over the full time horizon but also in all (sub-)intervals of time. We develop new online conformal prediction methods that minimize the strongly adaptive regret, which measures the worst-case regret over all intervals of a fixed length. We prove that our methods achieve near-optimal strongly adaptive regret for all interval lengths simultaneously, and approximately valid coverage. Experiments show that our methods consistently obtain better coverage and smaller prediction sets than existing methods on real-world tasks, such as time series forecasting and image classification under distribution shift.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Time-series interval forecasting | Electricity Demand | Coverage90.3 | 24 | |
| Conformal Prediction | Changepoint simulated (test) | Coverage90 | 24 | |
| Post-shift coverage recovery | Changepoint Shift 1 (Time 1) | Recovery Time12 | 24 | |
| Time-series interval forecasting | Amazon Stock | Coverage90 | 24 | |
| Post-shift coverage recovery | Changepoint Shift 2 (Time 2) | Recovery Time0.00e+0 | 24 | |
| Online Conformal Prediction | Variance Changepoint | Coverage0.9 | 24 | |
| Time-series interval forecasting | Google Stock | Coverage90 | 24 | |
| Time-series interval forecasting | Temperature | Coverage90.1 | 24 | |
| Online Conformal Prediction | Heavy-tailed | Coverage90 | 24 | |
| Online Conformal Prediction | Distribution Drift | Coverage90 | 24 |