Skip to yearly menu bar Skip to main content


Poster

Retrieval-Augmented Diffusion Models for Time Series Forecasting

Jingwei Liu · Ling Yang · Hongyan Li · Shenda Hong

East Exhibit Hall A-C #4202
[ ]
Fri 13 Dec 11 a.m. PST — 2 p.m. PST

Abstract:

While time series diffusion models have received considerable focus from many recent works, the performance of existing models remains highly unstable. Factors limiting time series diffusion models include insufficient time series datasets and the absence of guidance. To address these limitations, we propose a Retrieval-Augmented Time series Diffusion model (RATD). The framework of RATD consists of two parts: an embedding-based retrieval process and a reference-guided diffusion model. In the first part, RATD retrieves the time series that are most relevant to historical time series from the database as references. The references are utilized to guide the denoising process in the second part. Our approach allows leveraging meaningful samples within the database to aid in sampling, thus maximizing the utilization of datasets. Meanwhile, this reference-guided mechanism also compensates for the deficiencies of existing time series diffusion models in terms of guidance. Experiments and visualizations on multiple datasets demonstrate the effectiveness of our approach, particularly in complicated prediction tasks. Our code is available at https://github.com/stanliu96/RATD

Live content is unavailable. Log in and register to view live content