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

被引:10
|
作者
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] A novel reinforced dynamic graph convolutional network model with data imputation for network-wide traffic flow prediction
    Chen, Yong
    Chen, Xiqun
    TRANSPORTATION RESEARCH PART C-EMERGING TECHNOLOGIES, 2022, 143
  • [2] Transformer-Based Spatiotemporal Graph Diffusion Convolution Network for Traffic Flow Forecasting
    Wei, Siwei
    Yang, Yang
    Liu, Donghua
    Deng, Ke
    Wang, Chunzhi
    ELECTRONICS, 2024, 13 (16)
  • [3] Spatial Network-Wide Traffic Flow Imputation with Graph Neural Network
    Sina Sabzekar
    Rezvan Bahmani
    Masoud Ghasemi
    Zahra Amini
    International Journal of Intelligent Transportation Systems Research, 2025, 23 (1) : 385 - 403
  • [4] 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
  • [5] Network-wide Traffic Flow Prediction Research Based on DTW Algorithm Spatial-temporal Graph Convolution
    Liu Y.-C.
    Li Z.-P.
    Lv C.-P.
    Zhang T.
    Liu Y.
    Jiaotong Yunshu Xitong Gongcheng Yu Xinxi/Journal of Transportation Systems Engineering and Information Technology, 2022, 22 (03): : 147 - 157and178
  • [6] Learning continuous dynamic network representation with transformer-based temporal graph neural network
    Li, Yingji
    Wu, Yue
    Sun, Mingchen
    Yang, Bo
    Wang, Ying
    INFORMATION SCIENCES, 2023, 649
  • [7] A novel hybrid framework based on temporal convolution network and transformer for network traffic prediction
    Zhang, Zhiwei
    Gong, Shuhui
    Liu, Zhaoyu
    Chen, Da
    PLOS ONE, 2023, 18 (09):
  • [8] 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
  • [9] 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
  • [10] Dynamic multi-graph convolution recurrent neural network for traffic speed prediction
    Ge, Liang
    Jia, Yixuan
    Li, Qinhong
    Ye, Xiaofeng
    JOURNAL OF INTELLIGENT & FUZZY SYSTEMS, 2023, 44 (05) : 7319 - 7332