Auto-Regressive Moving Diffusion Models for Time Series Forecasting

Jiaxin Gao, Qinglong Cao, Yuntian Chen
12/12/2024
cs.LGcs.AIstat.AP

Official code: daxin007/ARMD (118 stars, MIT, last commit 2025-02-09).

Abstract

Time series forecasting (TSF) is essential in various domains, and recent advancements in diffusion-based TSF models have shown considerable promise. However, these models typically adopt traditional diffusion patterns, treating TSF as a noise-based conditional generation task. This approach neglects the inherent continuous sequential nature of time series, leading to a fundamental misalignment between diffusion mechanisms and the TSF objective, thereby severely impairing performance. To bridge this misalignment, and inspired by the classic Auto-Regressive Moving Average (ARMA) theory, which views time series as continuous sequential progressions evolving from previous data points, we propose a novel Auto-Regressive Moving Diffusion (ARMD) model to first achieve the continuous sequential diffusion-based TSF. Unlike previous methods that start from white Gaussian noise, our model employs chain-based diffusion with priors, accurately modeling the evolution of time series and leveraging intermediate state information to improve forecasting accuracy and stability. Specifically, our approach reinterprets the diffusion process by considering future series as the initial state and historical series as the final state, with intermediate series generated using a sliding-based technique during the forward process. This design aligns the diffusion model's sampling procedure with the forecasting objective, resulting in an unconditional, continuous sequential diffusion TSF model. Extensive experiments conducted on seven widely used datasets demonstrate that our model achieves state-of-the-art performance, significantly outperforming existing diffusion-based TSF models. Our code is available on GitHub: https://github.com/daxin007/ARMD.

View on arXivView PDF

Code Implementations(1)

daxin007/ARMDOfficial100%
MIT

[AAAI 2025] Official Implementation of "Auto-Regressive Moving Diffusion Models for Time Series Forecasting"

11812PythonDec 12, 20241 years agoMIT

Similar Papers

Cite this paper

@article{gao2024autoregressive,
  title  = {Auto-Regressive Moving Diffusion Models for Time Series Forecasting},
  author = {Jiaxin Gao and Qinglong Cao and Yuntian Chen},
  year   = {2024},
  eprint = {2412.09328},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2412.09328v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2412.09328.svg)](https://codeofpaper.com/paper/2412.09328)

Discussion