Authors: Hyunsuk Kim, Isaac Gibbs, Michael W. Mahoney, Ryan J. Tibshirani
Organizations: Department of Statistics, University of California, Berkeley · Department of Statistics, University of Toronto · ICSI, LBNL, and Dept. of Statistics, University of California, Berkeley
The task of constructing a predictor that is simultaneously accurate under multiple evaluation metrics has been popularized recently under the name omniprediction. Prior work in this area has largely focused on binary prediction tasks, and existing results for multiclass and continuous outcomes typically suffer from slow error rates. In this paper, we consider forecasts expressed as a set of quantiles over a discrete but otherwise arbitrary set of probability levels. We develop sample-efficient algorithms for constructing forecasts that are accurate simultaneously across all proper losses that satisfy mild regularity conditions, where proper losses are those minimized by predicting the true quantiles. Our algorithms are online and can operate in a dynamic environment, while simultaneously taking multiple losses into account, and their forecasts offer the potential to provide diverse end-users with rich information for evaluating risk and informing a range of practical decision-making tasks. As an example, we demonstrate how our procedure can be used to ensemble quantile forecasts of COVID-19 hospitalizations made by experts during the pandemic.
Figures & tables
s(a,o)
o=1
o=0
a=1
C
C
a=0
L
0
Table 1: Payoff table
Figure 1: Visualization of the discretization on the real line. Outcomes Yt and forecasts fbn(t) take values in the grid {0,1/m,...,1} that is interleaved by the values of θi .
Algorithm 1 Omniprediction under all proper losses for multiple quantile levels
Figure 2: Omni-error (Eq. ( 11 )) over time for 1 - to 4 -week-ahead forecasting horizons (left to right panels) on the US-agg data for *Our algorithm (blue), *Hedge-QL (green) and the COVIDhub-Ens and JHUAPL-SLPEns ensembles (both black). The values of cH,cO shown in the legend of each panel correspond to the learning rate which achieved the lowest total omni-error. The red line shows the results of our algorithm with a fixed learning rate constant cO=1 . Lower values indicate better performance.
Figure 3: Left: State-level comparison of performance relative to *Our algorithm . Rows show comparisons with three baseline ensembles. Each curve shows a Gaussian kernel density estimate of the difference in total omni-error (left column) or total quantile error (right column) across the 51 state-level time series for a given forecasting horizon. Values to the right of zero (vertical black dotted line) indicate better performance of our algorithm. Right: Comparison of instantaneous quantile loss realized by *Our algorithm (x-axis) and *Hedge-QL (y-axis). See text for detailed descriptions.
Figure 4: Sensitivity of our algorithm to the choice of learning rate. Each panel shows the performance of our algorithm as a function of the learning rate (blue curve) as measured by the total omni-error (top row) or total quantile loss relative to best base forecaster (bottom row) on the US-agg data. Panels from left to right show results at 1 - to 4 -week time horizons, while horizontal lines indicate the performance of three baseline methods. Lower values indicate better performance.
Algorithm 5 Efficient computation of Vn−1∗−Vn∗ for all n∈[N−1]
Figure 5: Runtime of Algorithm 1 with cO=1 on U.S.-aggregate, 4-week-ahead prediction data versus grid size m ( Left ), number of quantile levels N ( Middle ), and number of base forecasters B ( Right ). Middle and right plots use grid size 500 .
Algorithm 7 Two-player game based omniprediction for weighted quantile losses
Figure 6: Quantile forecasts for US-agg weekly COVID-19 hospitalizations at 1 -week-ahead and 4 -week-ahead horizons. The learning rate for *Hedge-QL and *Our algorithm is selected to achieve the best total omni-error. The black curve shows the observed hospitalizations. The blue shaded regions show nested quantile prediction bands: darker regions correspond to more central quantile intervals, while lighter regions correspond to wider intervals.
Figure 7: Sensitivity of Hedge-QL to the choice of learning rate. Each panel shows the performance of Hedge-QL as a function of the learning rate (green curve) as measured by the total omni-error (top row) or total quantile loss relative to best base forecaster (middle row) on the US-agg data. Panels from left to right shows results at one to four week time horizons, while horizontal lines indicate the performance of *Our algorithm and the two baseline ensembles. Lower values indicate better performance. The bottom row shows the percentage of time steps at which the quantile estimates of Hedge-QL exhibit quantile crossing.
Figure 8: Sensitivity of our algorithm to the grid resolution. Each panel shows the total quantile loss relative to the best base forecaster on the US-agg data as a function of the grid unit {1000,500,250,100,50,25} (a smaller unit yields a larger grid size m ). Lower values indicate better performance.
Figure 9: Distribution of standardized ranks of instantaneous quantile loss across all (horizon, location, time) tuples for the 51 state-level series. For each tuple, we rank the available forecasts by quantile loss and normalize the ranks to [0,1] . The colored segments represent the empirical distribution of ranks grouped into quartiles.