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

Embracing Heteroscedasticity for Probabilistic Time Series Forecasting

About

Probabilistic time series forecasting (PTSF) aims to model the full predictive distribution of future observations, enabling both accurate forecasting and principled uncertainty quantification. A central requirement of PTSF is to embrace heteroscedasticity, as real-world time series exhibit time-varying conditional variances induced by nonstationary dynamics, regime changes, and evolving external conditions. However, most existing non-autoregressive generative approaches to PTSF, such as TimeVAE and $K^2$VAE, rely on MSE-based training objectives that implicitly impose a homoscedastic assumption, thereby fundamentally limiting their ability to model temporal heteroscedasticity. To address this limitation, we propose the Location-Scale Gaussian VAE (LSG-VAE), a simple but effective framework that explicitly parameterizes both the predictive mean and time-dependent variance through a location-scale likelihood formulation. This design enables LSG-VAE to faithfully capture heteroscedastic aleatoric uncertainty and introduces an adaptive attenuation mechanism that automatically down-weights highly volatile observations during training, leading to improved robustness in trend prediction. Extensive experiments on nine benchmark datasets demonstrate that LSG-VAE consistently outperforms fifteen strong generative baselines while maintaining high computational efficiency suitable for real-time deployment.

Yijun Wang, Qiyuan Zhuang, Xiu-Shen Wei• 2026

Related benchmarks

TaskDatasetResultRank
Long-term time-series forecastingETTh1--
446
Long-term time-series forecastingILI--
102
Long-term time-series forecastingExchange--
79
Probabilistic ForecastingTraffic
CRPS0.18
48
Probabilistic ForecastingElectricity
CRPS0.069
44
Probabilistic time series forecastingETTm1
CRPS0.208
28
Probabilistic time series forecastingExchange
CRPS0.022
23
Probabilistic time series forecastingETTm2
CRPS0.107
22
Probabilistic time series forecastingETTh1
CRPS0.252
18
Probabilistic time series forecastingETTh2
CRPS0.148
17
Showing 10 of 30 rows

Other info

Follow for update