VDGCNeT: A novel network-wide Virtual Dynamic Graph Convolution Neural network and Transformer-based traffic prediction model

被引:22
作者
Zheng, Ge [1 ]
Chai, Wei Koong [2 ]
Zhang, Jiankang [2 ]
Katos, Vasilis [2 ]
机构
[1] Univ Cambridge, Inst Mfg IfM, Dept Engn, 17 Charles Babbage Rd, Cambridge CB3 0FS, Cambs, England
[2] Bournemouth Univ, Dept Comp & Informat, Poole BH12 5BB, Dorset, England
关键词
Traffic prediction; Deep learning; Virtual road graph; Intelligent transportation system; Large-scale road networks; FLOW PREDICTION; ARIMA;
D O I
10.1016/j.knosys.2023.110676
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
We address the problem of traffic prediction on large-scale road networks. We propose a novel deep learning model, Virtual Dynamic Graph Convolution Neural Network and Transformer with Gate and Attention mechanisms (VDGCNeT), to comprehensively extract complex, dynamic and hidden spatial dependencies of road networks for achieving high prediction accuracy. For this purpose, we advocate the use of a virtual dynamic road graph that captures the dynamic and hidden spatial dependencies of road segments in real road networks instead of purely relying on the physical road connectivity. We further design a novel framework based on Graph Convolution Neural Network (GCN) and Transformer to analyze dynamic and hidden spatial-temporal features. The gate mechanism is utilized for concatenating learned spatial and temporal features from Spatial and Temporal Transformers, respectively, while the Attention-based Similarity is used to update dynamic road graph. Two realworld traffic datasets from large-scale road networks with different properties are used for training and testing our model. We compare our VDGCNeT against nine other well-known models in the literature. Our results demonstrate that the proposed VDGCNeT is capable of achieving highly accurate predictions - on average 96.77% and 91.68% accuracy on PEMS-BAY and METR-LA datasets respectively. Overall, our VDGCNeT performs the best when compared against other existing models.& COPY; 2023 Elsevier B.V. All rights reserved.
引用
收藏
页数:16
相关论文
共 50 条
  • [1] Spatial Network-Wide Traffic Flow Imputation with Graph Neural Network
    Sabzekar, Sina
    Bahmani, Rezvan
    Ghasemi, Masoud
    Amini, Zahra
    INTERNATIONAL JOURNAL OF INTELLIGENT TRANSPORTATION SYSTEMS RESEARCH, 2024, : 385 - 403
  • [2] Short-term Network-wide Traffic Prediction Based on Graph Convolutional Network
    Chen X.-Q.
    Zhou L.-X.
    Cao Z.
    Jiaotong Yunshu Xitong Gongcheng Yu Xinxi/Journal of Transportation Systems Engineering and Information Technology, 2020, 20 (04): : 49 - 55
  • [3] A deep learning based multitask model for network-wide traffic speed prediction
    Zhang, Kunpeng
    Zheng, Liang
    Liu, Zijian
    Jia, Ning
    NEUROCOMPUTING, 2020, 396 : 438 - 450
  • [4] TAGTN: Traffic Prediction Model based on Adaptive Graph Transformer Network
    Zheng, Zhedian
    Sun, Wei
    PROCEEDINGS OF 2024 3RD INTERNATIONAL CONFERENCE ON CYBER SECURITY, ARTIFICIAL INTELLIGENCE AND DIGITAL ECONOMY, CSAIDE 2024, 2024, : 352 - 357
  • [5] Optimized Graph Convolution Recurrent Neural Network for Traffic Prediction
    Guo, Kan
    Hu, Yongli
    Qian, Zhen
    Liu, Hao
    Zhang, Ke
    Sun, Yanfeng
    Gao, Junbin
    Yin, Baocai
    IEEE TRANSACTIONS ON INTELLIGENT TRANSPORTATION SYSTEMS, 2021, 22 (02) : 1138 - 1149
  • [6] Survey On Traffic Flow Prediction Methods Based On Graph Convolution Neural Network
    Gao, Yi
    Wang, Xiaomeng
    SEVENTH INTERNATIONAL CONFERENCE ON TRAFFIC ENGINEERING AND TRANSPORTATION SYSTEM, ICTETS 2023, 2024, 13064
  • [7] Traffic Prediction and Congestion Control Based on Directed Graph Convolution Neural Network
    Zeng Y.-C.
    Shao M.-H.
    Sun L.-J.
    Lu C.
    Zhongguo Gonglu Xuebao/China Journal of Highway and Transport, 2021, 34 (12): : 239 - 248
  • [8] A graph convolution network based latency prediction algorithm for convolution neural network
    Li Z.
    Zhang R.
    Tan W.
    Ren Y.
    Lei M.
    Wu H.
    Beijing Hangkong Hangtian Daxue Xuebao/Journal of Beijing University of Aeronautics and Astronautics, 2022, 48 (12): : 2450 - 2459
  • [9] A Dynamic Heterogeneous Graph Convolution Network For Traffic Flow Prediction
    Li, He
    Jin, Duo
    Li, XueJiao
    Qiao, Shaojie
    COMPUTER JOURNAL, 2024, 67 (01) : 31 - 44
  • [10] Graph Transformer-Based Dynamic Edge Interaction Encoding for Traffic Prediction
    Ouyang, Nan
    Ao, Lei
    Cai, Qing
    Wan, Wenkang
    Ren, Xiaojiang
    He, Xin
    Sheng, Kai
    IEEE TRANSACTIONS ON INTELLIGENT TRANSPORTATION SYSTEMS, 2025, 26 (03) : 4066 - 4079