Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

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.

Xudong Zhang, Jierui Lei, Jiacheng Li, Lingdong Shen, Jian Cui, Haina Tang• 2026

Related benchmarks

TaskDatasetResultRank
Multivariate ForecastingPeMS04
MSE0.087
43
Multivariate Time-series ForecastingPEMS03 In-Distribution
MAE0.191
13
Multivariate Time-series ForecastingPEMS07 In-Distribution
MAE0.167
13
Multivariate Time-series ForecastingPEMS08 In-Distribution
Mean Absolute Error (MAE)0.186
13
Multivariate Time-series ForecastingWeather In-Distribution
MAE0.225
13
Multivariate Time-series ForecastingECL In-Distribution
MAE0.24
13
Multivariate Time-series ForecastingSolar In-Distribution
MAE0.194
13
Multivariate Time-series ForecastingFlight In-Distribution
MAE0.255
13
Time Series ForecastingPSM anomalous intervals (test)
MAE0.243
13
Time Series ForecastingCHP-LCS-Flow anomalous intervals (test)
MAE0.155
13
Showing 10 of 12 rows

Other info

Follow for update