Share your thoughts, 1 month free Claude Pro on usSee more
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
836
Time Series ForecastingETTh2
MSE3.462
796
Time Series ForecastingETTm2
MSE1.101
536
Time Series ForecastingETTh1 (test)
MSE1.332
398
Time Series ForecastingETTm1
MSE1.251
363
Time Series ForecastingETTm1 (test)
MSE1.877
315
Time Series ForecastingETTm2
MSE1.69
300
Time Series ForecastingECL
MSE0.505
294
Time Series ForecastingTraffic (test)
MSE0.19
272
Time Series ForecastingWeather (test)
MSE1.11
248
Showing 10 of 122 rows
...

Other info

Follow for update