Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Not All Retrievals are Useful: Cross-Attention for Input-Aware RAG in Time Series Forecasting

About

Retrieval-augmented generation (RAG) enhances zero-shot time series (TS) forecasting by leveraging external knowledge bases, yet existing approaches overlook input-level relevance when fusing retrieved samples with the query. We argue that not all retrievals are equally useful, and irrelevant ones can degrade performance. To this end, we propose Cross-RAG, a zero-shot RAG-based forecasting framework that selectively attends to query-relevant retrieved samples via query--retrieval cross-attention. By modeling input-level relevance between the query and retrieved samples, Cross-RAG jointly incorporates three sources of information: 1) the query itself, 2) the retrieved samples, and 3) their relational interactions. In particular, this input-aware design enables Cross-RAG to remain stable as the number of retrieved samples $k$ grows, whereas prior methods without cross-attention require careful $k$ tuning to avoid degradation from irrelevant retrievals. Extensive experiments demonstrate that Cross-RAG consistently improves zero-shot forecasting performance across multiple TSFM backbones and various RAG methods, with additional analyses confirming its effectiveness across various retrieval scenarios. Code is available at https://github.com/seunghan96/cross-rag/.

Seunghan Lee, Jaehoon Lee, Jun Seo, Sungdong Yoo, Minjae Kim, Tae Yoon Lim, Dongwan Kang, Hwanil Choi, SoonYoung Lee, Wonbin Ahn• 2026

Related benchmarks

TaskDatasetResultRank
Time Series ForecastingETTh1 (test)
MSE0.341
482
Time Series ForecastingETTm1 (test)
MSE0.29
427
Long-term forecastingETTh2
MSE0.243
376
Time Series ForecastingWeather (test)
MSE0.144
355
Time Series ForecastingETTh2 (test)
MSE0.243
314
Time Series ForecastingETTm2 (test)
MSE0.143
298
Time Series ForecastingElectricity (test)
MSE0.112
237
ForecastingWeather
MAE0.181
54
ForecastingETTm1
MAE0.322
49
Univariate Time Series ForecastingExchange (test)
MSE0.064
46
Showing 10 of 14 rows

Other info

Follow for update