A Spatiotemporal Multiscale Graph Convolutional Network for Traffic Flow Prediction

被引:11
作者
Cao, Shuqin
Wu, Libing [1 ,2 ,3 ]
Zhang, Rui [4 ]
Wu, Dan [5 ]
Cui, Jianqun [6 ]
Chang, Yanan [6 ]
机构
[1] Wuhan Univ, Sch Comp Sci, Wuhan 430072, Peoples R China
[2] Wuhan Univ, Sch Cyber Sci & Engn, Wuhan 430072, Peoples R China
[3] Guangdong Lab Artificial Intelligence & Digital E, Guangzhou 510335, Peoples R China
[4] Nanjing Univ Sci & Technol, Sch Cyber Sci & Engn, Nanjing 210094, Peoples R China
[5] Univ Windsor, Sch Comp Sci, Windsor, ON N9B 3P4, Canada
[6] Cent China Normal Univ, Sch Comp Sci, Wuhan 430079, Peoples R China
基金
中国国家自然科学基金;
关键词
Correlation; Roads; Spatiotemporal phenomena; Feature extraction; Convolutional neural networks; Predictive models; Adaptive systems; Traffic prediction; spatiotemporal correlations; multiscale graph; graph convolutional networks; cross-scale fusion;
D O I
10.1109/TITS.2024.3354802
中图分类号
TU [建筑科学];
学科分类号
0813 ;
摘要
Traffic prediction is vital to traffic planning, control, and optimization, which is necessary for intelligent traffic management. Existing methods mostly capture spatiotemporal correlations on a fine-grained traffic graph, which cannot make full use of cluster information in coarse-grained traffic graph. However, the flow variation of clusters in the coarse-grained traffic graph is more stable compared with nodes in the fine-grained traffic graph. And the flow variation of a fine-grained node is generally consistent with the trend of the cluster to which the node belongs. Thus information in the coarse-grained traffic graph can guide feature learning in the fine-grained traffic graph. To this end, we propose a Spatiotemporal Multiscale Graph Convolutional Network (SMGCN) that explores spatiotemporal correlations on a multiscale graph. Specifically, given a fine-grained traffic graph, we first generate a coarse-grained traffic graph by graph clustering, and extract spatiotemporal correlations on both fine-grained and coarse-grained traffic graphs. Then we propose a cross-scale fusion (CF) to implement information diffusion between the fine-grained and coarse-grained traffic graphs. Moreover, we employ an adaptive dynamic graph convolution network to mine both static and dynamic spatial features. We evaluate SMGCN on real-world datasets and obtain a 1.18% - 3.32% improvement over state-of-the-arts.
引用
收藏
页码:8705 / 8718
页数:14
相关论文
共 48 条
[41]  
Yao HX, 2018, AAAI CONF ARTIF INTE, P2588
[42]  
Yin XY, 2022, Arxiv, DOI arXiv:2207.01301
[43]  
Yu B, 2018, PROCEEDINGS OF THE TWENTY-SEVENTH INTERNATIONAL JOINT CONFERENCE ON ARTIFICIAL INTELLIGENCE, P3634
[44]  
Yu R., 2017, P SIAM INT C DAT MIN, P777, DOI DOI 10.1137/1.9781611974973.87
[45]   Spatial-Temporal Convolutional Graph Attention Networks for Citywide Traffic Flow Forecasting [J].
Zhang, Xiyue ;
Huang, Chao ;
Xu, Yong ;
Xia, Lianghao .
CIKM '20: PROCEEDINGS OF THE 29TH ACM INTERNATIONAL CONFERENCE ON INFORMATION & KNOWLEDGE MANAGEMENT, 2020, :1853-1862
[46]   T-GCN: A Temporal Graph Convolutional Network for Traffic Prediction [J].
Zhao, Ling ;
Song, Yujiao ;
Zhang, Chao ;
Liu, Yu ;
Wang, Pu ;
Lin, Tao ;
Deng, Min ;
Li, Haifeng .
IEEE TRANSACTIONS ON INTELLIGENT TRANSPORTATION SYSTEMS, 2020, 21 (09) :3848-3858
[47]  
Zheng CAP, 2020, AAAI CONF ARTIF INTE, V34, P1234
[48]   DSTAGCN: Dynamic Spatial-Temporal Adjacent Graph Convolutional Network for Traffic Forecasting [J].
Zheng, Qi ;
Zhang, Yaying .
IEEE TRANSACTIONS ON BIG DATA, 2023, 9 (01) :241-253