Adaptive Conformal Inference Under Distribution Shift
About
We develop methods for forming prediction sets in an online setting where the data generating distribution is allowed to vary over time in an unknown fashion. Our framework builds on ideas from conformal inference to provide a general wrapper that can be combined with any black box method that produces point predictions of the unseen label or estimated quantiles of its distribution. While previous conformal inference methods rely on the assumption that the data points are exchangeable, our adaptive approach provably achieves the desired coverage frequency over long-time intervals irrespective of the true data generating process. We accomplish this by modelling the distribution shift as a learning problem in a single parameter whose optimal value is varying over time and must be continuously re-estimated. We test our method, adaptive conformal inference, on two real world datasets and find that its predictions are robust to visible and significant distribution shifts.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Online Conformal Prediction | Variance Changepoint | Coverage0.91 | 24 | |
| Post-shift coverage recovery | Changepoint Shift 2 (Time 2) | Recovery Time0.00e+0 | 24 | |
| Time-series interval forecasting | Temperature | Coverage91 | 24 | |
| Time-series interval forecasting | Amazon Stock | Coverage90.2 | 24 | |
| Time-series interval forecasting | Electricity Demand | Coverage90.2 | 24 | |
| Time-series interval forecasting | Google Stock | Coverage90.5 | 24 | |
| Online Conformal Prediction | Heavy-tailed | Coverage90.3 | 24 | |
| Conformal Prediction | Changepoint simulated (test) | Coverage89.9 | 24 | |
| Post-shift coverage recovery | Changepoint Shift 1 (Time 1) | Recovery Time35 | 24 | |
| Conformal Prediction | Distribution Drift simulated (test) | Coverage89.9 | 24 |