论文标题

多空间 - 周期性融合图循环网络用于流量预测

Multi-Spatio-temporal Fusion Graph Recurrent Network for Traffic forecasting

论文作者

Zhao, Wei, Zhang, Shiqi, Zhou, Bing, Wang, Bei

论文摘要

交通预测对于新时代智能城市的交通建设至关重要。但是,流量数据的复杂空间和时间依赖性使流量预测极具挑战性。大多数现有的流量预测方法都依赖于预定义的邻接矩阵来对时空依赖性建模。但是,道路交通状态是高度实时的,因此邻接矩阵应随着时间的流逝而动态变化。本文介绍了一个新的多空间融合图复发网络(MSTFGRN),以解决上述问题。该网络提出了一种数据驱动的加权邻接矩阵生成方法,以补偿预定义的邻接矩阵未反映的实时空间依赖性。它还通过在不同时刻的平行时空关系上执行新的双向时空融合操作来有效地学习隐藏的时空依赖性。最后,通过将全局注意机制集成到时空融合模块中,同时捕获了整体时空依赖性。对四个大型现实世界流量数据集进行的广泛试验表明,与替代基线相比,我们的方法实现了最先进的性能。

Traffic forecasting is essential for the traffic construction of smart cities in the new era. However, traffic data's complex spatial and temporal dependencies make traffic forecasting extremely challenging. Most existing traffic forecasting methods rely on the predefined adjacency matrix to model the Spatio-temporal dependencies. Nevertheless, the road traffic state is highly real-time, so the adjacency matrix should change dynamically with time. This article presents a new Multi-Spatio-temporal Fusion Graph Recurrent Network (MSTFGRN) to address the issues above. The network proposes a data-driven weighted adjacency matrix generation method to compensate for real-time spatial dependencies not reflected by the predefined adjacency matrix. It also efficiently learns hidden Spatio-temporal dependencies by performing a new two-way Spatio-temporal fusion operation on parallel Spatio-temporal relations at different moments. Finally, global Spatio-temporal dependencies are captured simultaneously by integrating a global attention mechanism into the Spatio-temporal fusion module. Extensive trials on four large-scale, real-world traffic datasets demonstrate that our method achieves state-of-the-art performance compared to alternative baselines.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源