TS-RAG: Retrieval-Augmented Generation based Time Series Foundation Models are Stronger Zero-Shot Forecaster
Kanghui Ning, Zijie Pan, Yu Liu, Yushan Jiang, James Y. Zhang, Kashif Rasul, Anderson Schneider, Lintao Ma, Yuriy Nevmyvaka, Dongjin Song
Abstract
Large Language Models (LLMs) and Foundation Models (FMs) have recently become prevalent for time series forecasting tasks. While fine-tuning LLMs enables domain adaptation, they often struggle to generalize across diverse and unseen datasets. Moreover, existing Time Series Foundation Models (TSFMs) still face challenges in handling non-stationary dynamics and distribution shifts, largely due to the lack of effective mechanisms for adaptation. To this end, we present TS-RAG, a retrieval-augmented generation framework for time series forecasting that enhances the generalization and interpretability of TSFMs. Specifically, TS-RAG leverages pre-trained time series encoders to retrieve semantically relevant segments from a dedicated knowledge base, enriching the contextual representation of the input query. Furthermore, we propose an Adaptive Retrieval Mixer (ARM) module that dynamically fuses the retrieved patterns with the TSFM's internal representation, improving forecasting accuracy without requiring task-specific fine-tuning. Thorough empirical studies on seven public benchmark datasets demonstrate that TS-RAG achieves state-of-the-art zero-shot forecasting performance, outperforming the existing TSFMs by up to 6.84% across diverse domains while also providing desirable interpretability. Our code and data are available at: https://github.com/UConn-DSIS/TS-RAG .
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 56d9f633-773d-4809-bda7-6bb115519239Cited by top-tier papers9
- Learning Pattern-Specific Experts for Time Series Forecasting Under Patch-level Distribution ShiftYanru Sun, Zongxia Xie, Emadeldeen Eldele, Dongyue Chen et al.NeurIPS 2025 · 31 citations
- Cross City Traffic Flow Generation via Retrieval Augmented Diffusion ModelYudong Li, Jingyuan Wang, Xie Yu, Peiyu Wang et al.NeurIPS 2025 · 9 citations
- TRACE: Grounding Time Series in Context for Multimodal Embedding and RetrievalJialin Chen, Ziyu Zhao, Gaukhar Nurbek, Aosong Feng et al.NeurIPS 2025 · 8 citations
- Enhancing Multivariate Time Series Forecasting with Global Temporal RetrievalFanpu Cao, Lu Dai, Jindong Han, Hui XiongICLR 2026 · 7 citations
- MEMTS: Internalizing Domain Knowledge via Parameterized Memory for Retrieval-Free Domain Adaptation of Time Series Foundation ModelsXiaoyun Yu, Li Fan, Xiangfei Qiu, Nanqing Dong et al.KDD 2026 · 2 citations
Builds on26
- Language Models are Few-Shot LearnersTom B. Brown, Benjamin Mann, Nick Ryder, Melanie Subbiah et al.NeurIPS 2020 · 64,255 citations
- Retrieval-Augmented Generation for Knowledge-Intensive NLP TasksPatrick Lewis, Ethan Perez, Aleksandra Piktus, Fabio Petroni et al.NeurIPS 2020 · 19,162 citations
- Informer: Beyond Efficient Transformer for Long Sequence Time-Series ForecastingHaoyi Zhou, Shanghang Zhang, Jieqi Peng, Shuai Zhang et al.AAAI 2021 · 7,289 citations
- One Fits All: Power General Time Series Analysis by Pretrained LMTian Zhou, Peisong Niu, Xue Wang, Liang Sun et al.NeurIPS 2023 · 1,178 citations
- Time-LLM: Time Series Forecasting by Reprogramming Large Language ModelsMing Jin, Shiyu Wang, Lintao Ma, Zhixuan Chu et al.ICLR 2024 · 915 citations
Related papers
- QuiZSF: A Retrieval-Augmented Framework for Zero-Shot Time Series ForecastingShichao Ma, Zhengyang Zhou, Qihe Huang, Binwu Wang et al.WWW 2026
- In-Context Fine-Tuning for Time-Series Foundation ModelsMatthew Faw, Rajat Sen, Yichen Zhou, Abhimanyu DasICML 2025
- Cross-city Time Series Forecasting with Retrieval-Augmented Large Language ModelsYue Jiang, Chenxi Liu, Yile Chen, Qin Chao et al.WWW 2026
- TimeMRA: LLM-Empowered Time Series Forecasting via Multi-Scale Retrieval-Augmented RepresentationsZongjiang Shang, Chengxi Jin, Binqing Wu, Dongliang Cui et al.ICML 2026
- Time-FFM: Towards LM-Empowered Federated Foundation Model for Time Series ForecastingQingxiang Liu, Xu Liu, Chenghao Liu, Qingsong Wen et al.NeurIPS 2024 · 43 citations
