arXiv:2608.06223cs.AIcs.LG2026-08被引 1

用相似时间序列增强预测,提升准确率

TS-RAG: Retrieval Augmented Generation for Time Series Forecasting

论文配图:TS-RAG: Retrieval Augmented Generation for Time Series Forecasting
图 1 · 摘自论文原文
  • 设计专用引用标记,融合输入与检索序列信息
  • 在多个真实数据集上达到当前最佳性能
  • 适合需要高精度时序预测的研究与应用

尽管基于Transformer的深度学习模型在时间序列预测中表现优异,但检索增强生成(RAG)在该领域的应用仍有限。由于RAG通过引入外部相关信息提升了大语言模型的能力,检索相似的时间序列作为参考也可能改善预测准确性。然而,多数时间序列模型受限于训练数据少、参数规模小,且缺乏大语言模型的强生成能力,简单拼接参考序列作为提示无法达到预期效果。为此,我们提出新方法TS-RAG,利用RAG提升预测性能。该框架引入专门设计的参考标记,有效融合输入序列与检索到的相似序列信息,实现对复杂时序动态更稳健的捕捉。实验表明,TS-RAG在多个真实世界预测基准上均实现一致的领先性能。

原文摘要 · Abstract (English)

While deep learning models, particularly transformer-based architectures, have shown impressive performance in time series forecasting, the application of retrieval-augmented generation (RAG) in this domain remains limited. Since RAG has proven effective in enhancing the capabilities of large language models by incorporating relevant external information, retrieving similar time series sequences as references might also improve accuracy in time series forecasting tasks. However, most time series models are constrained by limited training data, smaller parameter scales, and a lack of the extensive generative capabilities found in large language models. Simply concatenating reference sequences into the prompt, as done in language models, may not yield the expected results. To address these challenges, we propose a novel approach, TS-RAG, which leverages RAG to enhance forecasting performance. The framework introduces specially designed reference tokens to effectively fuse information from the input sequence with that from retrieved similar sequences, enabling a more robust capture of complex temporal dynamics. Experimental results demonstrate that TS-RAG achieves consistent state-of-the-art performance across several real-world forecasting benchmarks.

时间序列RAG预测

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。