Online correction method improves multi-channel forecasts with less data storage

Local and Global Recursive Ridge Regression for Online Residual Correction of Frozen Multivariate Forecasters under Label Delay

Machine Learning

Summary

Forecast models predict many future points at once but can be off, especially when real results arrive late. The authors designed a method called DRiC that corrects these predictions online using a compressed representation, needing much less memory than previous methods. Their technique works either per data channel or globally across channels and uses statistical tricks to ignore large mistakes and keep corrections stable. Tested on different datasets, DRiC often reduces prediction errors more efficiently while using less memory.

What this means in practice

  • For energy demand forecasters: Improve electricity consumption forecasts by online correcting multi-channel errors with small memory overhead.
  • For traffic monitoring teams: Enhance traffic flow predictions by applying recursive correction efficiently for many data streams with label delays.

Authors

Takumi Fujimoto, Hiroaki Nishi

Abstract

A frozen multi-horizon forecaster can be corrected online from its own residuals. However, under label delay the target of a block arrives only after the whole horizon has passed, and the corrector's retained state grows with the number of channels. Aiming to combine a small retained state with a large error reduction, we propose DRiC (DCT Ridge Corrector), which fits a ridge regression of the leading discrete cosine transform (DCT) coefficients of the next residual block on those of the frozen forecast and of the latest input block, with sufficient statistics updated recursively. Its local variant, DRiC-L, keeps one such regression per channel; its global variant, DRiC-G, pools all channels into one regression, whose statistics do not depend on the channel count. A guard tracks a running scale of the frozen forecast's residual for each channel; it skips updates on blocks far outside that scale and clips the issued correction. We evaluate on seven datasets (four ETT series, Weather with its missing-value codes filled, Electricity, and Traffic) with DLinear, PatchTST, and iTransformer backbones, against 10 naive, gradient-trained, and closed-form methods. DRiC-L with eight DCT coefficients (DRiC-L8) reduces the mean squared error (MSE) by 14.55% on average over 126 conditions, against 12.98% for ELF, the most accurate comparator, while retaining 15-17 times less state, although ELF is more accurate on Traffic. DRiC-G retains 14.6 kB on the 321-channel Electricity data and 18.9 kB on the 862-channel Traffic data and is the most accurate method at or below its state on both. However, 5 to 7 of the 10 comparators, all with at least 10 times its state, are more accurate there. On the unfilled Weather file, DRiC-L8 without the guard has on average 8.0 times the frozen forecast's MSE, whereas with the guard it reduces the MSE by 7.94%. Code: https://github.com/keiotakmin/dric.