A new ensemble deep graph reinforcement learning network for spatio-temporal traffic volume forecasting in a freeway network

被引:0
|
作者
Shang, Pan [1 ]
Liu, Xinwei [2 ]
Yu, Chengqing [3 ]
Yan, Guangxi [3 ]
Xiang, Qingqing [4 ]
Mi, Xiwei [1 ]
机构
[1] Beijing Jiaotong Univ, Sch Traff & Transportat, Beijing 100044, Peoples R China
[2] Beijing Jiaotong Univ, Sch Mech Elect & Control Engn, Beijing 100044, Peoples R China
[3] Cent South Univ, Sch Traff & Transportat Engn, Changsha 410075, Peoples R China
[4] East China Jiaotong Univ, Sch Traff & Transportat, Nanchang 330013, Jiangxi, Peoples R China
关键词
Traffic volume forecasting; Graph convolutional network; Graph attention network; Deep reinforcement learning; Spatio-temporal traffic volume series; CONVOLUTIONAL NEURAL-NETWORK; SWARM OPTIMIZATION; ATTENTION; PREDICTION; MODEL; CLASSIFICATION; NODES;
D O I
10.1016/j.dsp.2022.103419
中图分类号
TM [电工技术]; TN [电子技术、通信技术];
学科分类号
0808 ; 0809 ;
摘要
Spatio-temporal traffic volume forecasting technologies can effectively improve freeway traffic efficiency and the travel comfort of humans. To construct a high-precision traffic volume forecasting model, this study proposed a new ensemble deep graph reinforcement learning network. The modeling process of the spatio-temporal prediction model mainly included three steps. In step I, raw spatiotemporal traffic network datasets (traffic volumes, traffic speeds, weather, and holidays) were preprocessed and the adjacency matrix was constructed. In step II, a graph attention network (GAT) and graph convolution network (GCN) were used as the main predictors to build the spatio-temporal traffic volume forecasting model and obtain the forecasting results, respectively. In step III, deep reinforcement learning was used to effectively analyze the correlations between the forecasting results from these two neural networks and the final results, so as to optimize the weight coefficient. The final result of the proposed model was obtained by combining the forecasting results from the GAT and GCN with the weight coefficient. Based on summarizing and analyzing the experimental results, it can be concluded that: (1) deep reinforcement learning can effectively integrate the two different graph neural networks and achieve better results than traditional ensemble methods; and (2) the presented ensemble model performs better than twenty-one models proposed by other researchers for all studied cases. (C)& nbsp;2022 Elsevier Inc. All rights reserved.
引用
收藏
页数:13
相关论文
共 50 条
  • [1] A new ensemble deep graph reinforcement learning network for spatio-temporal traffic volume forecasting in a freeway network
    Shang, Pan
    Liu, Xinwei
    Yu, Chengqing
    Yan, Guangxi
    Xiang, Qingqing
    Mi, Xiwei
    [J]. Digital Signal Processing: A Review Journal, 2022, 123
  • [2] Adaptive Joint Spatio-Temporal Graph Learning Network for Traffic Data Forecasting
    Wang, Tianyi
    Chen, Shu-Ching
    [J]. ACM Transactions on Spatial Algorithms and Systems, 2024, 10 (03)
  • [3] Probabilistic spatio-temporal graph convolutional network for traffic forecasting
    Karim, Atkia Akila
    Nower, Naushin
    [J]. APPLIED INTELLIGENCE, 2024, : 7070 - 7085
  • [4] Adaptive Spatio-temporal Graph Neural Network for traffic forecasting
    Ta, Xuxiang
    Liu, Zihan
    Hu, Xiao
    Yu, Le
    Sun, Leilei
    Du, Bowen
    [J]. KNOWLEDGE-BASED SYSTEMS, 2022, 242
  • [5] Spatio-temporal fusion graph convolutional network for traffic flow forecasting
    Ma, Ying
    Lou, Haijie
    Yan, Ming
    Sun, Fanghui
    Li, Guoqi
    [J]. INFORMATION FUSION, 2024, 104
  • [6] Spatio-Temporal Graph Attention Convolution Network for Traffic Flow Forecasting
    Liu, Kun
    Zhu, Yifan
    Wang, Xiao
    Ji, Hongya
    Huang, Chengfei
    [J]. TRANSPORTATION RESEARCH RECORD, 2024, 2678 (09) : 136 - 149
  • [7] Spatio-Temporal Graph Structure Learning for Traffic Forecasting
    Zhang, Qi
    Chang, Jianlong
    Meng, Gaofeng
    Xiang, Shiming
    Pan, Chunhong
    [J]. THIRTY-FOURTH AAAI CONFERENCE ON ARTIFICIAL INTELLIGENCE, THE THIRTY-SECOND INNOVATIVE APPLICATIONS OF ARTIFICIAL INTELLIGENCE CONFERENCE AND THE TENTH AAAI SYMPOSIUM ON EDUCATIONAL ADVANCES IN ARTIFICIAL INTELLIGENCE, 2020, 34 : 1177 - 1185
  • [8] Deep spatio-temporal graph convolutional network for traffic accident prediction
    Yu, Le
    Du, Bowen
    Hu, Xiao
    Sun, Leilei
    Han, Liangzhe
    Lv, Weifeng
    [J]. NEUROCOMPUTING, 2021, 423 (423) : 135 - 147
  • [9] Spatio-Temporal Graph Convolutional Networks: A Deep Learning Framework for Traffic Forecasting
    Yu, Bing
    Yin, Haoteng
    Zhu, Zhanxing
    [J]. PROCEEDINGS OF THE TWENTY-SEVENTH INTERNATIONAL JOINT CONFERENCE ON ARTIFICIAL INTELLIGENCE, 2018, : 3634 - 3640
  • [10] Heterogeneous Spatio-Temporal Graph Convolution Network for Traffic Forecasting with Missing Values
    Zhong, Weida
    Suo, Qiuling
    Jia, Xiaowei
    Zhang, Aidong
    Su, Lu
    [J]. 2021 IEEE 41ST INTERNATIONAL CONFERENCE ON DISTRIBUTED COMPUTING SYSTEMS (ICDCS 2021), 2021, : 707 - 717