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

Retrieval Augmented Time Series Forecasting

About

Time series forecasting uses historical data to predict future trends, leveraging the relationships between past observations and available features. In this paper, we propose RAFT, a retrieval-augmented time series forecasting method to provide sufficient inductive biases and complement the model's learning capacity. When forecasting the subsequent time frames, we directly retrieve historical data candidates from the training dataset with patterns most similar to the input, and utilize the future values of these candidates alongside the inputs to obtain predictions. This simple approach augments the model's capacity by externally providing information about past patterns via retrieval modules. Our empirical evaluations on ten benchmark datasets show that RAFT consistently outperforms contemporary baselines with an average win ratio of 86%.

Sungwon Han, Seungeon Lee, Meeyoung Cha, Sercan O Arik, Jinsung Yoon• 2025

Related benchmarks

TaskDatasetResultRank
Multivariate ForecastingETTh1
MSE0.42
645
Long-term time-series forecastingWeather
MSE0.27
348
Long-term time-series forecastingETTh2
MSE0.382
327
Long-term time-series forecastingTraffic
MSE0.414
278
Long-term forecastingETTm1
MSE0.381
184
Long-term forecastingETTh1
MSE0.428
179
Long-term forecastingETTm2
MSE0.281
174
Long-term forecastingETTh2
MSE0.357
163
Long-term time-series forecastingElectricity
MSE0.175
103
Long-term forecastingElectricity
MSE0.156
50
Showing 10 of 19 rows

Other info

Follow for update