论文标题
通过张量分解重建的融合图,时空交通建模
Spatial-temporal traffic modeling with a fusion graph reconstructed by tensor decomposition
论文作者
论文摘要
准确的时空交通流量预测对于帮助交通经理采取控制措施和驱动程序来选择最佳旅行路线至关重要。最近,由于其强大的捕获空间依赖性的能力,图形卷积网络(GCN)已被广泛用于交通预测。时空图邻接矩阵的设计是GCN成功的关键,它仍然是一个空旷的问题。本文提议通过张量分解重建二进制邻接矩阵,并提出了交通流量预测方法。首先,我们将空间融合图邻接矩阵重新调整为三向邻接张量。然后,我们通过Tucker分解重建了邻接张量,其中编码更有信息和全局的空间依赖性。最后,为全球相关性学习组装了用于局部空间 - 周期相关性学习和扩张的卷积模块的空间同步卷积模块,以汇总和学习该道路网络的全面空间依赖性。四个开放访问数据集的实验结果表明,就预测性能和计算成本而言,所提出的模型优于最先进的方法。
Accurate spatial-temporal traffic flow forecasting is essential for helping traffic managers to take control measures and drivers to choose the optimal travel routes. Recently, graph convolutional networks (GCNs) have been widely used in traffic flow prediction owing to their powerful ability to capture spatial-temporal dependencies. The design of the spatial-temporal graph adjacency matrix is a key to the success of GCNs, and it is still an open question. This paper proposes reconstructing the binary adjacency matrix via tensor decomposition, and a traffic flow forecasting method is proposed. First, we reformulate the spatial-temporal fusion graph adjacency matrix into a three-way adjacency tensor. Then, we reconstructed the adjacency tensor via Tucker decomposition, wherein more informative and global spatial-temporal dependencies are encoded. Finally, a Spatial-temporal Synchronous Graph Convolutional module for localized spatial-temporal correlations learning and a Dilated Convolution module for global correlations learning are assembled to aggregate and learn the comprehensive spatial-temporal dependencies of the road network. Experimental results on four open-access datasets demonstrate that the proposed model outperforms state-of-the-art approaches in terms of the prediction performance and computational cost.