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
796
Time Series ForecastingETTm2
MSE0.286
536
Time Series ForecastingETTm2
MSE0.268
300
Time Series ForecastingECL
MSE0.879
294
Time Series ForecastingElectricity
MSE0.27
237
Multivariate Time-series ForecastingETTh1 (test)
MSE0.407
160
Multivariate long-term series forecastingExchange (test)
MSE0.018
159
Multivariate Time-series ForecastingWeather (test)
MSE0.311
145
Time Series ForecastingILI
MAE1.169
141
Time Series Forecastingsolar
MSE0.821
106
Showing 10 of 63 rows

Other info

Follow for update