A Novel Attention-Based Dynamic Multi-Graph Spatial-Temporal Graph Neural Network Model for Traffic Prediction

作者：Chunyan Diao, Dafang Zhang, Wei Liang, Man Jiang, Kuanching Li · 发表于：IEEE Transactions on Emerging Topics in Computational Intelligence · 年份：2024 · DOI：10.1109/tetci.2024.3462513 · 被引用次数：11 · 研究领域：Traffic Prediction and Management Techniques、Advanced Clustering Algorithms Research、Advanced Graph Neural Networks

Traffic flow prediction is a non-negligible part of intelligent transportation and mobility. Unfortunately, the unique non-linearity and complex spatial-ST-correlation of transport flow data suggest considerable challenges in prediction. The dynamic interaction of multiple spatial relations greatly influences traffic flow prediction. However, the existing spatial-temporal prediction algorithms are based on graph convolution to capture global or heterogeneous relationships, and simpler graph convolution models cannot accurately capture complex dynamic spatial relationships. To address the issues as mentioned above, this study proposes an attention-based multi-graph dynamic spatial-temporal prediction model ADMSTGCN to capture a variety of dynamic interaction relationships in traffic flow. First, we use a distance graph to explore the relationships between adjacent distances and use a semantic graph to mine spatial relationships between nodes that are far apart but have similar relationships, then fuse these two graphs to obtain a fusion graph with multiple spatial interaction relationships. The correlations between different neighbors are then further learned through a dynamic multi-graph spatial-temporal learning module that aggregates the features of different neighbors through gated graph convolution and attention mechanisms to capture various dynamic and complex spatial-temporal interactions. Experimental evaluations show that the framework proposed outperforms existing me...