TS-RAG: Retrieval Augmented Generation for Time Series Forecasting
TS-RAG:用于时间序列预测的检索增强生成
🧭 检索增强的时间序列预测
Baidu, Inc.
提出 TS-RAG,用参考令牌融合检索序列,提升时间序列预测精度
📌 工作参考价值(4 点)
- 在电商销量、流量等时序预测中,可借鉴 TS-RAG 的检索增强思路,检索相似历史窗口并通过可学习的参考令牌融合,避免简单拼接带来的噪声。
- 对于推荐系统的行为序列预测(如用户下一步动作),可检索相似用户序列作为参考,用融合机制增强模型对长尾或冷启动场景的预测能力。
- 工程实现上,需预先对历史序列切片并计算表示向量,构建近似最近邻索引,在线推理时低延迟检索,同时注意定期更新索引。
- 参考令牌设计与 Transformer 主干解耦,可灵活插入现有预测模型,无需大幅改动架构。