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

Non-autoregressive Conditional Diffusion Models for Time Series Prediction

About

Recently, denoising diffusion models have led to significant breakthroughs in the generation of images, audio and text. However, it is still an open question on how to adapt their strong modeling ability to model time series. In this paper, we propose TimeDiff, a non-autoregressive diffusion model that achieves high-quality time series prediction with the introduction of two novel conditioning mechanisms: future mixup and autoregressive initialization. Similar to teacher forcing, future mixup allows parts of the ground-truth future predictions for conditioning, while autoregressive initialization helps better initialize the model with basic time series patterns such as short-term trends. Extensive experiments are performed on nine real-world datasets. Results show that TimeDiff consistently outperforms existing time series diffusion models, and also achieves the best overall performance across a variety of the existing strong baselines (including transformers and FiLM).

Lifeng Shen, James Kwok• 2023

Related benchmarks

TaskDatasetResultRank
Time Series ForecastingETTm2
MSE0.41
382
Multivariate long-term series forecastingExchange (test)
MSE0.018
145
Multivariate Time-series ForecastingETTh1 (test)
MSE0.407
134
Multivariate Time-series ForecastingWeather (test)
MSE0.311
124
Time Series ForecastingElectricity
MSE0.27
77
Multivariate Time-series ForecastingETTm1 (test)
MSE0.336
67
Multivariate Time-series ForecastingElectricity (test)--
36
Time Series ForecastingTraffic
MSE0.68
31
Showing 8 of 8 rows

Other info

Follow for update