【NeurIPS2024】用于时间序列预测的检索增强扩散模型

文摘   2024-11-04 19:03   山西  

尽管时间序列扩散模型在许多最新的研究中受到了广泛关注,但现有模型的性能仍然非常不稳定。限制时间序列扩散模型的因素包括时间序列数据集不足以及缺乏指导。为了解决这些限制,我们提出了一种检索增强时间序列扩散模型(RATD)。RATD 框架由两个部分组成:基于嵌入的检索过程和参考引导的扩散模型。在第一部分中,RATD 从数据库中检索与历史时间序列最相关的时间序列作为参考。这些参考在第二部分中被用于指导去噪过程。我们的方法通过利用数据库中的有意义样本来辅助采样,从而最大化数据集的利用率。同时,这种参考引导机制也弥补了现有时间序列扩散模型在指导方面的不足。在多个数据集上的实验和可视化结果展示了我们方法的有效性,尤其是在复杂的预测任务中。我们的代码已发布在:https://github.com/stanliu96/RATD

想要了解更多资讯,请扫描下方二维码,关注机器学习研究会

                                          


转自:专知

机器学习研究组订阅
机器学习研究会由百度七剑客雷鸣先生创办,旨在推动AI的技术发展和产业落地。参与组织北大、清华”AI前沿与产业趋势“公开课,广泛的和高校、企业、创业、VC开展合作,自身也参与优秀AI项目的投资和孵化。
 最新文章