VLBM: Variational Latent Basis Modeling for OOD Robust Multivariate Time Series Forecasting
About
Out of distribution (OOD) events in multivariate time series forecasting are rare but often dominate real world risk, making average case forecasting insufficient for reliable deployment. Under standard average risk training on mixed ID/OOD distributions, optimization signals from rare OOD events can be overwhelmed by frequent in distribution (ID) patterns, so strong benchmark accuracy may not translate into reliability under high impact shifts. To address this issue, we propose VLBM (Variational Latent Basis Model), a theory guided latent forecasting framework that separates stable dynamics from OOD induced deviations. VLBM learns a shared latent basis that defines a low rank subspace for stable ID dynamics, explicitly decomposes inputs into basis subspace components and orthogonal residual components, and aligns a future aware posterior with a future blind prior so that test time latent inference depends only on historical input. Across 12 benchmark tasks spanning transportation, weather, power systems, and other real world domains, including newly constructed real world OOD traffic datasets, VLBM achieves state of the art OOD robustness and ID accuracy, with average MAE and MSE gains of 15.08\% and 7.74\% over the strongest baseline. On a synthetic simulation dataset, VLBM also consistently achieves the best performance and better tracks OOD pulse recovery. These results support latent structured forecasting as a principled route to robust prediction under mixed ID and OOD conditions. The code is available at https://github.com/leijieruilq/VLBM_OOD_forecast.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Multivariate Forecasting | PeMS04 | MSE0.087 | 43 | |
| Multivariate Time-series Forecasting | PEMS03 In-Distribution | MAE0.191 | 13 | |
| Multivariate Time-series Forecasting | PEMS07 In-Distribution | MAE0.167 | 13 | |
| Multivariate Time-series Forecasting | PEMS08 In-Distribution | Mean Absolute Error (MAE)0.186 | 13 | |
| Multivariate Time-series Forecasting | Weather In-Distribution | MAE0.225 | 13 | |
| Multivariate Time-series Forecasting | ECL In-Distribution | MAE0.24 | 13 | |
| Multivariate Time-series Forecasting | Solar In-Distribution | MAE0.194 | 13 | |
| Multivariate Time-series Forecasting | Flight In-Distribution | MAE0.255 | 13 | |
| Time Series Forecasting | PSM anomalous intervals (test) | MAE0.243 | 13 | |
| Time Series Forecasting | CHP-LCS-Flow anomalous intervals (test) | MAE0.155 | 13 |