A Spatiotemporal Multiscale Graph Convolutional Network for Traffic Flow Prediction

被引:0
|
作者
Cao, Shuqin [1 ]
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 Ec, 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
关键词
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.
引用
下载
收藏
页码:1 / 14
页数:14
相关论文
共 50 条
  • [41] STGMN: A gated multi-graph convolutional network framework for traffic flow prediction
    Qingjian Ni
    Meng Zhang
    Applied Intelligence, 2022, 52 : 15026 - 15039
  • [42] Hierarchical Traffic Flow Prediction Based on Spatial-Temporal Graph Convolutional Network
    Wang, Hanqiu
    Zhang, Rongqing
    Cheng, Xiang
    Yang, Liuqing
    IEEE TRANSACTIONS ON INTELLIGENT TRANSPORTATION SYSTEMS, 2022, 23 (09) : 16137 - 16147
  • [43] Federated Spatio-Temporal Traffic Flow Prediction Based on Graph Convolutional Network
    Wang, Hanqiu
    Zhang, Rongqing
    Cheng, Xiang
    Yang, Liuqing
    2022 14TH INTERNATIONAL CONFERENCE ON WIRELESS COMMUNICATIONS AND SIGNAL PROCESSING, WCSP, 2022, : 221 - 225
  • [44] GECRAN: Graph embedding based convolutional recurrent attention network for traffic flow prediction
    Yan, Jianqiang
    Zhang, Lin
    Gao, Yuan
    Qu, Boting
    EXPERT SYSTEMS WITH APPLICATIONS, 2024, 256
  • [45] Dynamic Graph Convolutional Network for Long Short-Term Traffic Flow Prediction
    Wang, Yan
    Ren, Qianqian
    SSRN, 2022,
  • [46] SPATIOTEMPORAL GRAPH CONVOLUTIONAL NEURAL NETWORKS FOR METRO FLOW PREDICTION
    Jin, Shiyuan
    Jing, Changfeng
    Wang, Yi
    Lv, Xinxin
    XXIV ISPRS CONGRESS IMAGING TODAY, FORESEEING TOMORROW, COMMISSION IV, 2022, 43-B4 : 403 - 409
  • [47] RT-GCN: Gaussian-based spatiotemporal graph convolutional network for robust traffic prediction
    Liu, Yutian
    Rasouli, Soora
    Wong, Melvin
    Feng, Tao
    Huang, Tianjin
    INFORMATION FUSION, 2024, 102
  • [48] IGAGCN: Information geometry and attention-based spatiotemporal graph convolutional networks for traffic flow prediction
    An, Jiyao
    Guo, Liang
    Liu, Wei
    Fu, Zhiqiang
    Ren, Ping
    Liu, Xinzhi
    Li, Tao
    NEURAL NETWORKS, 2021, 143 : 355 - 367
  • [49] Spatiotemporal Graph Convolutional Network for Multi-Scale Traffic Forecasting
    Wang, Yi
    Jing, Changfeng
    ISPRS INTERNATIONAL JOURNAL OF GEO-INFORMATION, 2022, 11 (02)
  • [50] Koopman theory meets graph convolutional network: Learning the complex dynamics of non-stationary highway traffic flow for spatiotemporal prediction
    Wang, Ting
    Ngoduy, Dong
    Li, Ye
    Lyu, Hao
    Zou, Guojian
    Dantsuji, Takao
    CHAOS SOLITONS & FRACTALS, 2024, 187