Share your thoughts, 1 month free Claude Pro on usSee more
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
909
Time Series ForecastingETTh1--
869
Multivariate Time-series ForecastingETTm1
MSE0.348
742
Multivariate Time-series ForecastingETTm2
MSE0.254
593
Long-term time-series forecastingWeather
MSE0.27
527
Long-term forecastingETTh1
MSE0.421
500
Time Series ForecastingETTh1 (test)
MSE0.37
482
Long-term time-series forecastingETTh2
MSE0.382
474
Multivariate Time-series ForecastingWeather
MSE0.241
466
Long-term forecastingETTm1
MSE0.348
452
Showing 10 of 53 rows

Other info

Follow for update