Multi-View Spatial-Temporal Graph Convolutional Network for Traffic Prediction

• Published in: IEEE transactions on intelligent transportation systems Vol. 25; no. 8; pp. 9572 - 9586
• Main Authors: Wei, Shuqing, Feng, Siyuan, Yang, Hai
• Format: Journal Article
• Language: English
• Published: IEEE 01.08.2024
• Subjects: Attention mechanism; Computational modeling; Convolutional neural networks; Correlation; Data models; multi-view learning; Predictive models; Public transportation; Roads; spatial–temporal graph convolutional network; traffic prediction
• ISSN: 1524-9050; 1558-0016
• DOI: 10.1109/TITS.2024.3364759

Multi-step traffic speed prediction is a challenging issue due to the multiple spatial-temporal dependencies among roads. Some spatial dependencies, especially those formed by different traffic modes, are not fully exploited, and how to simultaneously consider spatial and temporal dependencies and effectively integrate them within a single prediction framework needs further exploration. To tackle the above issues, we propose a multi-view spatial-temporal graph convolutional framework MVSTG, which adequately exploits the multi-view spatial-temporal dependencies and their interactions to improve the accuracy of traffic prediction. Multi-view temporal learning captures the multiple temporal trends by temporal convolution from multi-granularity historical data, and multi-view spatial learning handles the multiple spatial correlations by graph convolution from multiple graphs. In addition, view-wise attention-based fusion is proposed to adaptively identify the importance of each upstream view, fuse the multi-view information, and generate integrated results for downstream views. The experiments on two real-world urban traffic datasets demonstrate that the multi-view data and the proposed model framework enhance performance on the accuracy of speed prediction, especially in mid-term and long-term prediction.