A Spatiotemporal Multiscale Graph Convolutional Network for Traffic Flow Prediction

被引:5
|
作者
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
相关论文
共 50 条
  • [21] GCN-GAN: Integrating Graph Convolutional Network and Generative Adversarial Network for Traffic Flow Prediction
    Zheng, Hongling
    Li, Xiang
    Li, Yongfeng
    Yan, Ziqin
    Li, Tinghong
    IEEE ACCESS, 2022, 10 : 94051 - 94062
  • [22] Bayesian graph convolutional network for traffic prediction
    Fu, Jun
    Zhou, Wei
    Chen, Zhibo
    NEUROCOMPUTING, 2024, 582
  • [23] AST-GCN: Attribute-Augmented Spatiotemporal Graph Convolutional Network for Traffic Forecasting
    Zhu, Jiawei
    Wang, Qiongjie
    Tao, Chao
    Deng, Hanhan
    Zhao, Ling
    Li, Haifeng
    IEEE ACCESS, 2021, 9 : 35973 - 35983
  • [24] Spatiotemporal Deep-Learning Model With Graph Convolutional Network for Well Logs Prediction
    Feng, Shuo
    Li, Xuegui
    Zeng, Fang
    Hu, Zhongrui
    Sun, Yuhang
    Wang, Zepeng
    Duan, Hanxu
    IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2023, 20
  • [25] Multi dynamic temporal representation graph convolutional network for traffic flow prediction
    Zuojun Wu
    Xiaojun Liu
    Xiaoling Zhang
    Scientific Reports, 15 (1)
  • [26] ST-DAGCN: A spatiotemporal dual adaptive graph convolutional network model for traffic prediction
    Liu, Yutian
    Feng, Tao
    Rasouli, Soora
    Wong, Melvin
    NEUROCOMPUTING, 2024, 601
  • [27] SAX-STGCN: Dynamic Spatio-Temporal Graph Convolutional Networks for Traffic Flow Prediction
    Lei, Bin
    Zhang, Peng
    Suo, Yifei
    Li, Na
    IEEE ACCESS, 2022, 10 : 107022 - 107031
  • [28] Synchronous Spatiotemporal Graph Transformer: A New Framework for Traffic Data Prediction
    Wang, Tian
    Chen, Jiahui
    Lu, Jinhu
    Liu, Kexin
    Zhu, Aichun
    Snoussi, Hichem
    Zhang, Baochang
    IEEE TRANSACTIONS ON NEURAL NETWORKS AND LEARNING SYSTEMS, 2023, 34 (12) : 10589 - 10599
  • [29] T-GCN: A Temporal Graph Convolutional Network for Traffic Prediction
    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
  • [30] Weather-Aware Fiber-Wireless Traffic Prediction Using Graph Convolutional Networks
    Abdullah, Mariam
    He, Jiayuan
    Wang, Ke
    IEEE ACCESS, 2022, 10 : 95908 - 95918