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 条
[1]   Two-Stage Intrusion Detection System in Intelligent Transportation Systems Using Rule Extraction Methods From Deep Neural Networks [J].
Almutlaq, Samah ;
Derhab, Abdelouahid ;
Hassan, Mohammad Mehedi ;
Kaur, Kuljeet .
IEEE TRANSACTIONS ON INTELLIGENT TRANSPORTATION SYSTEMS, 2023, 24 (12) :15687-15701
[2]  
Bai L, 2020, PROC ADVNEURAL INF P, P1
[3]  
Bai L, 2019, PROCEEDINGS OF THE TWENTY-EIGHTH INTERNATIONAL JOINT CONFERENCE ON ARTIFICIAL INTELLIGENCE, P1981
[4]   Geometric Multimodal Deep Learning With Multiscaled Graph Wavelet Convolutional Network [J].
Behmanesh, Maysam ;
Adibi, Peyman ;
Ehsani, Sayyed Mohammad Saeed ;
Chanussot, Jocelyn .
IEEE TRANSACTIONS ON NEURAL NETWORKS AND LEARNING SYSTEMS, 2024, 35 (05) :6991-7005
[5]   Traffic transformer: Capturing the continuity and periodicity of time series for traffic forecasting [J].
Cai, Ling ;
Janowicz, Krzysztof ;
Mai, Gengchen ;
Yan, Bo ;
Zhu, Rui .
TRANSACTIONS IN GIS, 2020, 24 (03) :736-755
[6]  
Chen WQ, 2020, AAAI CONF ARTIF INTE, V34, P3529
[7]  
Diao ZL, 2019, AAAI CONF ARTIF INTE, P890
[8]   Multitype Highway Mobility Analytics for Efficient Learning Model Design: A Case of Station Traffic Prediction [J].
Duan, Sijing ;
Lyu, Feng ;
Ren, Ju ;
Wang, Yifeng ;
Yang, Peng ;
Zhang, Desheng ;
Zhang, Yaoxue .
IEEE TRANSACTIONS ON INTELLIGENT TRANSPORTATION SYSTEMS, 2022, 23 (10) :19484-19496
[9]   Spatial-Temporal Graph ODE Networks for Traffic Flow Forecasting [J].
Fang, Zheng ;
Long, Qingqing ;
Song, Guojie ;
Xie, Kunqing .
KDD '21: PROCEEDINGS OF THE 27TH ACM SIGKDD CONFERENCE ON KNOWLEDGE DISCOVERY & DATA MINING, 2021, :364-373
[10]  
Geng X, 2019, AAAI CONF ARTIF INTE, P3656