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

Autoregressive Denoising Diffusion Models for Multivariate Probabilistic Time Series Forecasting

About

In this work, we propose \texttt{TimeGrad}, an autoregressive model for multivariate probabilistic time series forecasting which samples from the data distribution at each time step by estimating its gradient. To this end, we use diffusion probabilistic models, a class of latent variable models closely connected to score matching and energy-based methods. Our model learns gradients by optimizing a variational bound on the data likelihood and at inference time converts white noise into a sample of the distribution of interest through a Markov chain using Langevin sampling. We demonstrate experimentally that the proposed autoregressive denoising diffusion model is the new state-of-the-art multivariate probabilistic forecasting method on real-world data sets with thousands of correlated dimensions. We hope that this method is a useful tool for practitioners and lays the foundation for future research in this area.

Kashif Rasul, Calvin Seward, Ingmar Schuster, Roland Vollgraf• 2021

Related benchmarks

TaskDatasetResultRank
Time Series ForecastingETTh1
MSE1.15
601
Time Series ForecastingETTm2
MSE1.36
382
Time Series ForecastingETTm1
MSE1.251
334
Time Series ForecastingETTh1 (test)
MSE1.332
262
Time Series ForecastingETTm1 (test)
MSE1.877
196
Time Series ForecastingTraffic (test)
MSE0.19
192
Time Series ForecastingTraffic
MSE3.495
145
Time Series ForecastingWeather (test)
MSE1.11
110
Time Series ForecastingElectricity
MSE0.69
77
Time Series ForecastingElectricity (test)
MSE2.703
72
Showing 10 of 71 rows
...

Other info

Follow for update