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 条
  • [1] Dynamic Spatiotemporal Graph Wavelet Network for Traffic Flow Prediction
    Xu, Weijian
    Liu, Jingjin
    Yan, Jingwen
    Yang, Juan
    Liu, Huifen
    Zhou, Teng
    IEEE INTERNET OF THINGS JOURNAL, 2024, 11 (05): : 8019 - 8029
  • [2] Spatiotemporal Data Fusion in Graph Convolutional Networks for Traffic Prediction
    Zhao, Baoxin
    Gao, Xitong
    Liu, Jianqi
    Zhao, Juanjuan
    Xu, Chengzhong
    IEEE ACCESS, 2020, 8 : 76632 - 76641
  • [3] Extended Multi-Component Gated Recurrent Graph Convolutional Network for Traffic Flow Prediction
    Zhao, Junhui
    Xiong, Xincheng
    Zhang, Qingmiao
    Wang, Dongming
    IEEE TRANSACTIONS ON INTELLIGENT TRANSPORTATION SYSTEMS, 2024, 25 (05) : 4634 - 4644
  • [4] Dynamic Graph Convolutional Recurrent Network With Spatiotemporal Category Information Embedding for Traffic Flow Prediction
    Zhu, Guodong
    Niu, Yunyun
    Du, Songzhi
    Wang, Pengcheng
    IEEE INTERNET OF THINGS JOURNAL, 2024, 11 (24): : 39473 - 39486
  • [5] Multi-View SpatialTemporal Graph Convolutional Network for Traffic Prediction
    Wei, Shuqing
    Feng, Siyuan
    Yang, Hai
    IEEE TRANSACTIONS ON INTELLIGENT TRANSPORTATION SYSTEMS, 2024, 25 (08) : 9572 - 9586
  • [6] ADGCN: An Asynchronous Dilation Graph Convolutional Network for Traffic Flow Prediction
    Qi, Tao
    Li, Guanghui
    Chen, Lingqiang
    Xue, Yanming
    IEEE INTERNET OF THINGS JOURNAL, 2022, 9 (05) : 4001 - 4014
  • [7] Temporal Multi-Graph Convolutional Network for Traffic Flow Prediction
    Lv, Mingqi
    Hong, Zhaoxiong
    Chen, Ling
    Chen, Tieming
    Zhu, Tiantian
    Ji, Shouling
    IEEE TRANSACTIONS ON INTELLIGENT TRANSPORTATION SYSTEMS, 2021, 22 (06) : 3337 - 3348
  • [8] Spatial-Temporal Dynamic Graph Convolutional Neural Network for Traffic Prediction
    Xiao, Wenjuan
    Wang, Xiaoming
    IEEE ACCESS, 2023, 11 : 97920 - 97929
  • [9] Traffic Message Channel Prediction Based on Graph Convolutional Network
    Li, Ning
    Jia, Shuangcheng
    Li, Qian
    IEEE ACCESS, 2021, 9 : 135423 - 135431
  • [10] Spatial-Temporal Traffic Prediction With an Interactive Spatial-Enhanced Graph Convolutional Network Model
    Li, Qin
    Xu, Pai
    Yang, Xuan
    Wu, Yuankai
    He, Hongwen
    He, Deqiang
    IEEE TRANSACTIONS ON INTELLIGENT TRANSPORTATION SYSTEMS, 2024, : 20767 - 20778