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/.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Time Series Forecasting | ETTh1 (test) | MSE0.341 | 482 | |
| Time Series Forecasting | ETTm1 (test) | MSE0.29 | 427 | |
| Long-term forecasting | ETTh2 | MSE0.243 | 376 | |
| Time Series Forecasting | Weather (test) | MSE0.144 | 355 | |
| Time Series Forecasting | ETTh2 (test) | MSE0.243 | 314 | |
| Time Series Forecasting | ETTm2 (test) | MSE0.143 | 298 | |
| Time Series Forecasting | Electricity (test) | MSE0.112 | 237 | |
| Forecasting | Weather | MAE0.181 | 54 | |
| Forecasting | ETTm1 | MAE0.322 | 49 | |
| Univariate Time Series Forecasting | Exchange (test) | MSE0.064 | 46 |