Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

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.

Aadyot Bhatnagar, Huan Wang, Caiming Xiong, Yu Bai• 2023

Related benchmarks

TaskDatasetResultRank
Time-series interval forecastingElectricity Demand
Coverage90.3
24
Conformal PredictionChangepoint simulated (test)
Coverage90
24
Post-shift coverage recoveryChangepoint Shift 1 (Time 1)
Recovery Time12
24
Time-series interval forecastingAmazon Stock
Coverage90
24
Post-shift coverage recoveryChangepoint Shift 2 (Time 2)
Recovery Time0.00e+0
24
Online Conformal PredictionVariance Changepoint
Coverage0.9
24
Time-series interval forecastingGoogle Stock
Coverage90
24
Time-series interval forecastingTemperature
Coverage90.1
24
Online Conformal PredictionHeavy-tailed
Coverage90
24
Online Conformal PredictionDistribution Drift
Coverage90
24
Showing 10 of 16 rows

Other info

Follow for update