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 条
  • [41] Traffic Demand Prediction Based on Dynamic Transition Convolutional Neural Network
    Du, Bowen
    Hu, Xiao
    Sun, Leilei
    Liu, Junming
    Qiao, Yanan
    Lv, Weifeng
    IEEE TRANSACTIONS ON INTELLIGENT TRANSPORTATION SYSTEMS, 2021, 22 (02) : 1237 - 1247
  • [42] Toward Directed Spatiotemporal Graph: A New Idea for Heterogeneous Traffic Prediction
    Ku, Yixuan
    Guo, Chen
    Zhang, Kangshuai
    Cui, Yunduan
    Shu, Hongfeng
    Yang, Yang
    Peng, Lei
    IEEE INTELLIGENT TRANSPORTATION SYSTEMS MAGAZINE, 2024, 16 (02) : 70 - 87
  • [43] A Network Traffic Prediction Model Based on Layered Training Graph Convolutional Network
    Li, Yulian
    Su, Yang
    IEEE ACCESS, 2025, 13 : 24398 - 24410
  • [44] Spatiotemporal Meteorological Prediction Based on Fully Convolutional Neural Network
    Zhang, Jiaqi
    Wang, Bin
    Hua, Mingyang
    Chen, Zekun
    Liang, Shili
    Kang, Xinyu
    IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2024, 62
  • [45] Self-Supervised Spatiotemporal Graph Neural Networks With Self-Distillation for Traffic Prediction
    Ji, Junzhong
    Yu, Fan
    Lei, Minglong
    IEEE TRANSACTIONS ON INTELLIGENT TRANSPORTATION SYSTEMS, 2023, 24 (02) : 1580 - 1593
  • [46] A Resource-Aware Multi-Graph Neural Network for Urban Traffic Flow Prediction in Multi-Access Edge Computing Systems
    Ali, Ahmad
    Ullah, Inam
    Shabaz, Mohammad
    Sharafian, Amin
    Khan, Muhammad Attique
    Bai, Xiaoshan
    Qiu, Li
    IEEE TRANSACTIONS ON CONSUMER ELECTRONICS, 2024, 70 (04) : 7252 - 7265
  • [47] Operating Status Prediction Model at EV Charging Stations With Fusing Spatiotemporal Graph Convolutional Network
    Su, Su
    Li, Yujing
    Chen, Qifang
    Xia, Mingchao
    Yamashita, Koji
    Jurasz, Jakub
    IEEE TRANSACTIONS ON TRANSPORTATION ELECTRIFICATION, 2023, 9 (01) : 114 - 129
  • [48] Heterogeneous Modular Traffic Prediction Based on Multilayer Graph Convolutional Network
    Chang, Mengmeng
    Ding, Zhiming
    Zhao, Zilin
    Cai, Zhi
    IEEE TRANSACTIONS ON INTELLIGENT TRANSPORTATION SYSTEMS, 2024, 25 (07) : 7805 - 7817
  • [49] Spatio-Temporal Residual Graph Convolutional Network for Short-Term Traffic Flow Prediction
    Zhang, Qingyong
    Tan, Meifang
    Li, Changwu
    Xia, Huiwen
    Chang, Wanfeng
    Li, Minglong
    IEEE ACCESS, 2023, 11 : 84187 - 84199
  • [50] A Graph and Attentive Multi-Path Convolutional Network for Traffic Prediction
    Qi, Jianzhong
    Zhao, Zhuowei
    Tanin, Egemen
    Cui, Tingru
    Nassir, Neema
    Sarvi, Majid
    IEEE TRANSACTIONS ON KNOWLEDGE AND DATA ENGINEERING, 2023, 35 (07) : 6548 - 6560