Achieving Risk Control in Online Learning Settings

Shai Feldman, Liran Ringel, Stephen Bates, Yaniv Romano

Abstract:

To provide rigorous uncertainty quantification for online learning models, we develop a framework for constructing uncertainty sets that provably control risk---such as coverage of confidence intervals, false negative rate, or F1 score---in the online setting. This extends conformal prediction to apply to a larger class of online learning problems. Our method guarantees risk control at any user-specified level even when the underlying data distribution shifts drastically, even adversarially, over time in an unknown fashion.
The technique we propose is highly flexible as it can be applied with any base online learning algorithm (e.g., a deep neural network trained online), requiring minimal implementation effort and essentially zero additional computational cost.
We further extend our approach to control multiple risks simultaneously, so the prediction sets we generate are valid for all given risks.
To demonstrate the utility of our method, we conduct experiments on real-world tabular time-series data sets showing that the proposed method rigorously controls various natural risks.
Furthermore, we show how to construct valid intervals for an online image-depth estimation problem that previous sequential calibration schemes cannot handle.

Submission Length: Regular submission (no more than 12 pages of main content)

Code: https://github.com/Shai128/rrc

Assigned Action Editor: ~Ruoyu_Sun1

License: Creative Commons Attribution 4.0 International (CC BY 4.0)

Submission Number: 1204