Share your thoughts, 1 month free Claude Pro on usSee more
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 ForecastingETTh2
MSE0.456
561
Time Series ForecastingETTm2
MSE0.41
382
Time Series ForecastingECL
MSE0.879
211
Multivariate Time-series ForecastingETTh1 (test)
MSE0.407
150
Multivariate long-term series forecastingExchange (test)
MSE0.018
146
Multivariate Time-series ForecastingWeather (test)
MSE0.311
140
Time Series ForecastingElectricity
MSE0.27
114
Time Series ForecastingILI
MAE1.169
103
Multivariate Time-series ForecastingETTm1 (test)
MSE0.336
83
Time Series ForecastingETTm2
MSE0.268
53
Showing 10 of 52 rows

Other info

Follow for update