arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2104.05914cs.LGcs.AI
收藏

GSA-Forecaster:基于图序列注意力机制的图结构时变数据预测

GSA-Forecaster: Forecasting Graph-Based Time-Dependent Data with Graph Sequence Attention

Carnegie Mellon University(卡内基梅隆大学);Iowa State University(爱荷华州立大学);Microsoft(微软公司)

Yang Li, Di Wang, José M. F. Moura

AI总结:

本文提出GSA-Forecaster,一种基于图序列注意力机制的新型深度学习模型,用于预测图结构时变数据,同时捕捉空间与时间依赖并整合辅助信息,在真实数据集上优于现有最先进模型。

摘要:

基于图结构的时变数据预测具有广泛的实际应用,但也面临诸多挑战。有效的模型必须同时捕捉数据中的空间和时间依赖关系,同时整合辅助信息以提高预测准确性。在本文中,我们指出了当前最先进模型在处理时间依赖方面的局限性。为克服这些局限,我们引入了GSA-Forecaster,一种专为基于图结构的时变场景设计的新型深度学习模型。GSA-Forecaster利用本文提出的新注意力机制——图序列注意力,来有效管理时间依赖。GSA-Forecaster将数据的图结构直接集成到其架构中,以处理空间依赖。此外,它整合辅助信息以进一步优化预测。我们使用真实的基于图结构的时变数据集验证其性能,结果表明其相比现有最先进模型具有更优的有效性。

英文摘要:

Forecasting graph-based, time-dependent data has broad practical applications but presents challenges. Effective models must capture both spatial and temporal dependencies in the data, while also incorporating auxiliary information to enhance prediction accuracy. In this paper, we identify limitations in current state-of-the-art models regarding temporal dependency handling. To overcome this, we introduce GSA-Forecaster, a new deep learning model designed for forecasting in graph-based, time-dependent contexts. GSA-Forecaster utilizes graph sequence attention, a new attention mechanism proposed in this paper, to effectively manage temporal dependencies. GSA-Forecaster integrates the data's graph structure directly into its architecture, addressing spatial dependencies. Additionally, it incorporates auxiliary information to refine its predictions further. We validate its performance using real-world graph-based, time-dependent datasets, where it demonstrates superior effectiveness compared to existing state-of-the-art models.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑