Skip to main content
Article
Node Connection Strength Matrix-Based Graph Convolution Network for Traffic Flow Prediction
IEEE Transactions on Vehicular Technology
  • Jian Chen, Sun Yat-Sen University
  • Wei Wang, Beijing Institute of Technology
  • Keping Yu, Hosei University
  • Xiping Hu, Beijing Institute of Technology
  • Ming Cai, Sun Yat-Sen University
  • Mohsen Guizani, Mohamed Bin Zayed University of Artificial Intelligence
Document Type
Article
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.

DOI
10.1109/TVT.2023.3265300
Publication Date
4-6-2023
Keywords
  • Convolution,
  • Deep learning,
  • Feature extraction,
  • Graph Convolution Network,
  • Indexes,
  • Network Representation Learning,
  • Node Connection Strength,
  • Predictive models,
  • Representation learning,
  • Task analysis,
  • Traffic Flow Prediction
Comments

IR Deposit conditions:

OA version (pathway a) Accepted version

No embargo

When accepted for publication, set statement to accompany deposit (see policy)

Must link to publisher version with DOI

Publisher copyright and source must be acknowledged

Citation Information
J. Chen, W. Wang, K. Yu, X. Hu, M. Cai and M. Guizani, "Node Connection Strength Matrix-Based Graph Convolution Network for Traffic Flow Prediction," in IEEE Transactions on Vehicular Technology, April 2023, doi: 10.1109/TVT.2023.3265300.