Organizations: Electrical Engineering and Computer Science, University of California, Berkeley, Berkeley, CA, USA · School of Electrical and Computer Engineering, Cornell University, Ithaca, NY, USA
Financial time-series forecasting is a high-stakes problem where regime shifts and shocks make point-accurate yet overconfident models dangerous. We propose Uncertainty-Gated Generative Modeling (UGGM), which treats uncertainty as an internal control signal that gates (i) representation via gated reparameterization, (ii) propagation via similarity and confidence routing, and (iii) generation via uncertainty-controlled predictive distributions, together with uncertainty-driven regularization and calibration to curb miscalibration. Instantiated on Weak Innovation AutoEncoder (WIAE-GPF), our UG-WIAE-GPF significantly improves risk-sensitive forecasting, delivering a 63.5% MSE reduction on NYISO (0.3508 → 0.1281), with improved robustness under shock intervals (mSE: 0.2739 → 0.1748).
Figures & tables
Pointwise
Normalized
Robust (Med.)
Scale-free
Model
MSE ↓
MAE ↓
NMSE ↓
NMAE ↓
mSE ↓
mAE ↓
MAPE ↓
MASE ↓
Proposed method (WIAE-based)
UG-WIAE-GPF (Ours)
0.1281
0.2550
0.0119
0.0280
0.1748
0.2145
0.2515
0.0220
WIAE-GPF (2024)
0.3508
0.3835
0.0220
0.0399
0.2739
0.3466
0.3958
0.3990
Transformer-/decomposition-based baselines
PatchTST (2022)
1.7508
1.0866
0.0387
0.0320
10.5493
3.2480
3.3637
0.1648
Table 1: Comparison with state-of-the-art forecasters on NYISO (2018–2024). Lower is better. Best is bold; second best is underlined (ties share the same rank).
Appendix figures & tables2 assets
Supplementary material from the paper’s appendix.
Appendix
Figure 1: NYISO data characteristics. Top: Long-term day-ahead LMP trend (2018–2024) showing strong non-stationarity and extreme spikes. Bottom: Daily profile with hourly averages and 95% confidence intervals, highlighting pronounced intra-day heteroskedasticity.
Figure 2: Hourly probability density evolution of NYISO load/price. Distributions vary substantially across hours in location, scale, and tail mass, revealing strong intra-day heteroskedasticity and heavy-tailed extremes.
Time series forecasting predicts future values from ordered observations using statistical, neural, and probabilistic models to support decisions under uncertainty.
Real-world time series exhibit temporally structured uncertainty: volatility clusters in turbulent regimes, dissipates in stable periods, and shifts abruptly around structural breaks. Yet many probabilistic forecasting methods estimate predictive uncertainty as an independent per-step quantity, leaving the evolution and persistence of volatility regimes under-modeled. We formalize this missing dimension as temporal uncertainty dynamics and instantiate it in the Volatility Dynamics Variational Autoencoder (VolDy-VAE), a non-autoregressive generative forecaster with a location-scale decoder. VolDy-VAE combines a location path for mean prediction with a recurrent scale path that transfers and evolves a volatility hidden state from the look-back window to the forecasting horizon, enabling temporally coherent predictive variances. This design yields an adaptive attenuation mechanism: high-variance observations receive lower influence on the location estimate while their uncertainty is preserved through explicit scale predictions. We further provide a simplified regime-switching analysis showing that, when variances are known or consistently estimated, the volatility-aware objective reduces to inverse-variance weighting, whereas MSE-based estimators remain unbiased but statistically inefficient. Experiments on nine benchmarks show that VolDy-VAE improves forecasting accuracy and uncertainty calibration over competitive probabilistic and point-forecasting baselines while maintaining low inference latency; plug-in studies further indicate that the VolDy principle can benefit GAN, Koopman VAE, and Transformer backbones. The source code is publicly available at https://github.com/wangyijunlyy/VolDy-VAE.
Yijun Wang, Qiyuan Zhuang, Larysa Marchanka +1
Department of Computer Science, Southeast University, Nanjing, China · Francisk Skorina Gomel State University, Belarus
We propose ABF-T-GLCP, a model-agnostic framework for forecasting and uncertainty quantification in nonstationary multivariate time series. The central idea is to learn an adaptive predictive state representation for point forecasting and reuse it for conformal calibration. The forecasting module combines horizon-specific temporal experts through a learned gate and refines predictions using sparse predictive transfer across related series. The uncertainty module, Gate-Localized Conformal Prediction (GLCP), uses the learned gate state, together with temporal recency, to select locally relevant calibration residuals, thereby coupling uncertainty calibration to the predictive regimes used by the forecasting model. This shared representation allows point forecasts and prediction intervals to adapt consistently under evolving temporal dynamics while retaining the model-agnostic nature of conformal prediction and yielding approximate local coverage under mild stability conditions. Experiments on a large-scale high-frequency commodity forecasting benchmark show consistent gains in point forecasting accuracy and substantially narrower prediction intervals with empirical coverage close to the nominal level. Additional results indicate that the framework extends beyond the motivating financial application.
Ziling Ma, Junshu Jiang, Ángel López-Oriona +2
Statistics Department, Computer, Electrical and Mathematical Sciences and Engineering Division, King Abdullah University of Science and Technology, Thuwal 23955-6900, Saudi Arabia · NeuroAI Laboratory, Computer, Electrical and Mathematical Sciences and Engineering Division, King Abdullah University of Science and Technology, Thuwal 23955-6900, Saudi Arabia
The recent emergence of Time Series Foundation Models (TSFMs) has significantly advanced multi-step forecasting performance, enabling accurate predictions over extended future horizons. However, existing TSFMs often suffer from significantly inherent uncertainty, which typically manifests as derived forecast branches emerging at each time step and spreading to subsequent steps; different forecast branches often exhibit varying forecasting performance, thereby undermining the credibility of TSFM forecasts. In this paper, we propose the Slicing-Graphing-Alignment (SGA) method to quantify the uncertainty of multi-step TSFM forecasts. The proposed SGA first characterizes the topology of all potential forecast branches using a directed acyclic graph, such that the graph complexity bounds the uncertainty of multi-step forecasts, and then precisely measures the graph complexity by integrating both topological information and TSFM-inherent stochasticity. Experimental results conducted on 11 TSFMs and 27 datasets demonstrate that (i) SGA achieves the best performance when ranking predictive errors with uncertainty estimates; (ii) SGA works with a more extensive and more precise sampling coverage than those of existing UQ methods, deriving a quantification mechanism fundamentally different from those of established ones; and (iii) larger model scales of TSFMs correlate with lower uncertainty estimates of multi-step forecasts, suggesting another empirical scaling law for uncertainty quantification of multi-step TSFM forecasts.
Xin-Yu Hu, Shuang Liang, Cheng Feng +1
National Key Laboratory for Novel Software Technology, Nanjing University, China. · School of Intelligent Science and Technology, Nanjing University, China. · Siemens Data and AI Research, Beijing, China. +1