Node Connection Strength Matrix-Based Graph Convolution Network for Traffic Flow Prediction
Jian Chen, Wei Wang, Keping Yu, Xiping Hu, Ming Cai, Mohsen Guizani
Abstract
Traffic flow prediction plays an integral role in intelligent transport systems, helping to manage and control urban traffic and improving the operational efficiency of road networks. Although the current mainstream traffic flow prediction models have achieved good accuracy, they cannot effectively utilize the unique characteristics of the traffic network where the importance of a node in the traffic network is positively correlated with the traffic flow through the node. Actually, the historical traffic properties of nodes will have a great influence on the future. With this background, in this paper, we propose a node connection strength index by network representation learning to utilize the historical traffic attributes of nodes. Then, we design a graph convolution network based on the node connection strength matrix to predict the traffic flow of the node. A novel Dynamics Extractor is designed to learn the various characteristics of the traffic flow. Experimental results demonstrate that the proposed scheme has a better performance by comparison with baseline methods.