Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

TARFVAE: Efficient One-Step Generative Time Series Forecasting via TARFLOW based VAE

About

Time series data is ubiquitous, with forecasting applications spanning from finance to healthcare. Beyond popular deterministic methods, generative models are gaining attention due to advancements in areas like image synthesis and video generation, as well as their inherent ability to provide probabilistic predictions. However, existing generative approaches mostly involve recurrent generative operations or repeated denoising steps, making the prediction laborious, particularly for long-term forecasting. Most of them only conduct experiments for relatively short-term forecasting, with limited comparison to deterministic methods in long-term forecasting, leaving their practical advantages unclear. This paper presents TARFVAE, a novel generative framework that combines the Transformer-based autoregressive flow (TARFLOW) and variational autoencoder (VAE) for efficient one-step generative time series forecasting. Inspired by the rethinking that complex architectures for extracting time series representations might not be necessary, we add a flow module, TARFLOW, to VAE to promote spontaneous learning of latent variables that benefit predictions. TARFLOW enhances VAE's posterior estimation by breaking the Gaussian assumption, thereby enabling a more informative latent space. TARFVAE uses only the forward process of TARFLOW, avoiding autoregressive inverse operations and thus ensuring fast generation. During generation, it samples from the prior latent space and directly generates full-horizon forecasts via the VAE decoder. With simple MLP modules, TARFVAE achieves superior performance over state-of-the-art deterministic and generative models across different forecast horizons on benchmark datasets while maintaining efficient prediction speed, demonstrating its effectiveness as an efficient and powerful solution for generative time series forecasting.

Jiawen Wei, Lan Jiang, Pengbo Wei, Ziwen Ye, Teng Song, Chen Chen, Guangrui Ma• 2025

Related benchmarks

TaskDatasetResultRank
Multivariate ForecastingETTh1
MSE0.361
645
Multivariate Time-series ForecastingETTm1
MSE0.38
433
Multivariate ForecastingETTh2
MSE0.273
341
Multivariate Time-series ForecastingETTm2
MSE0.271
334
Multivariate Time-series ForecastingWeather
MSE0.239
276
Multivariate Time-series ForecastingExchange
MAE0.4
165
Multivariate Time-series ForecastingElectricity
MSE0.168
150
Multivariate long-term series forecastingExchange (test)
MSE0.016
145
Multivariate Time-series ForecastingETTh1 (test)
MSE0.405
134
Multivariate Time-series ForecastingWeather (test)
MSE0.309
124
Showing 10 of 20 rows

Other info

Follow for update