arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2103.06126cs.LG
收藏

时空张量图卷积网络在流量预测中的应用

Spatial-Temporal Tensor Graph Convolutional Network for Traffic Prediction

School of Computer Science and Engineering, Nanjing University of Science and Technology(南京理工大学计算机科学与工程学院)

Xuran Xu, Tong Zhang, Chunyan Xu, Zhen Cui, Jian Yang

摘要:

准确的交通预测是城市交通引导和管理的关键。然而,现有的交通量预测模型在捕捉交通量数据的时空相关性时,大多没有考虑计算负担和存储空间。在这项工作中,我们提出了一个分解时空张量图卷积网络来处理交通速度预测。交通网络建模和统一到一个图,同时集成了空间和时间的信息。我们进一步扩展图卷积到张量空间,并提出了一个张量图卷积网络,以提取更多的区别特征的时空图数据。为了减少计算负担,我们采用Tucker张量分解并导出因式分解的张量卷积,该卷积在小尺度空间,时间和特征模式中执行单独的滤波。此外,我们可以受益于噪声抑制的交通数据时,丢弃那些平凡的成分在张量分解的过程中。在两个真实的交通速度数据集上的大量实验表明,该方法比传统的交通预测方法更有效,同时达到了最先进的性能。

英文摘要:

Accurate traffic prediction is crucial to the guidance and management of urban traffics. However, most of the existing traffic prediction models do not consider the computational burden and memory space when they capture spatial-temporal dependence among traffic data. In this work, we propose a factorized Spatial-Temporal Tensor Graph Convolutional Network to deal with traffic speed prediction. Traffic networks are modeled and unified into a graph that integrates spatial and temporal information simultaneously. We further extend graph convolution into tensor space and propose a tensor graph convolution network to extract more discriminating features from spatial-temporal graph data. To reduce the computational burden, we take Tucker tensor decomposition and derive factorized a tensor convolution, which performs separate filtering in small-scale space, time, and feature modes. Besides, we can benefit from noise suppression of traffic data when discarding those trivial components in the process of tensor decomposition. Extensive experiments on two real-world traffic speed datasets demonstrate our method is more effective than those traditional traffic prediction methods, and meantime achieves state-of-the-art performance.

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

↑