Convolutional Neural Network-Based Bidirectional Gated Recurrent Unit-Additive Attention Mechanism Hybrid Deep Neural Networks for Short-Term Traffic Flow Prediction

被引:3
|
作者
Liu, Song [1 ,2 ,3 ,4 ,5 ]
Lin, Wenting [2 ]
Wang, Yue [6 ]
Yu, Dennis Z. [7 ]
Peng, Yong [2 ,5 ]
Ma, Xianting [1 ,2 ]
机构
[1] Chongqing Jiaotong Univ, Inst Key Lab Traff Syst, Chongqing 400074, Peoples R China
[2] Chongqing Jiaotong Univ, Sch Traff & Transportat, Chongqing 400074, Peoples R China
[3] Chongqing Jiaotong Univ, Inst Intelligent Optimizat Comprehens Transportat, Chongqing 400074, Peoples R China
[4] Nanjing Univ Aeronaut & Astronaut, Coll Civil Aviat, Nanjing 210016, Peoples R China
[5] Chongqing Jiaotong Univ, Res Ctr Transportat & Int Supply Chain Management, Chongqing 400074, Peoples R China
[6] Highway Serv Ctr Yongchuan Dist, Chongqing 402160, Peoples R China
[7] Clarkson Univ, David D Reh Sch Business, Potsdam, NY 13699 USA
关键词
sustainable transportation; short-term traffic flow; convolutional neural network; bidirectional gated recurrent unit; additive attention mechanism; combinatorial predictive model; MODEL;
D O I
10.3390/su16051986
中图分类号
X [环境科学、安全科学];
学科分类号
08 ; 0830 ;
摘要
To more accurately predict short-term traffic flow, this study posits a sophisticated integrated prediction model, CNN-BiGRU-AAM, based on the additive attention mechanism of a convolutional bidirectional gated recurrent unit neural network. This model seeks to enhance the precision of traffic flow prediction by integrating both historical and prospective data. Specifically, the model achieves prediction through two steps: encoding and decoding. In the encoding phase, convolutional neural networks are used to extract spatial correlations between weather and traffic flow in the input sequence, while the BiGRU model captures temporal correlations in the time series. In the decoding phase, an additive attention mechanism is introduced to weigh and fuse the encoded features. The experimental results demonstrate that the CNN-BiGRU model, coupled with the additive attention mechanism, is capable of dynamically capturing the temporal patterns of traffic flow, and the introduction of isolation forests can effectively handle data anomalies and missing values, improving prediction accuracy. Compared to benchmark models such as GRU, the CNN-BiGRU-AAM model shows significant improvement on the test set, with a 47.49 reduction in the Root Mean Square Error (RMSE), a 30.72 decrease in the Mean Absolute Error (MAE), and a 5.27% reduction in the Mean Absolute Percentage Error (MAPE). The coefficient of determination (R2) reaches 0.97, indicating the high accuracy of the CNN-BiGRU-AAM model in traffic flow prediction. It provides a good solution for short-term traffic flow with spatio-temporal features, thereby enhancing the efficiency of traffic management and planning and promoting the sustainable development of transportation.
引用
收藏
页数:15
相关论文
共 50 条
  • [41] Short-term traffic flow prediction based on improved wavelet neural network
    Chen, Qiuxia
    Song, Ying
    Zhao, Jianfeng
    NEURAL COMPUTING & APPLICATIONS, 2021, 33 (14): : 8181 - 8190
  • [42] Combination prediction for short-term traffic flow based on artificial neural network
    Liu, Jiansheng
    Fu, Hui
    Liao, Xinxing
    WCICA 2006: SIXTH WORLD CONGRESS ON INTELLIGENT CONTROL AND AUTOMATION, VOLS 1-12, CONFERENCE PROCEEDINGS, 2006, : 8659 - +
  • [43] Short-term traffic flow prediction based on improved wavelet neural network
    Qiuxia Chen
    Ying Song
    Jianfeng Zhao
    Neural Computing and Applications, 2021, 33 : 8181 - 8190
  • [44] Attention-based bidirectional gated recurrent unit neural networks for well logs prediction and lithology identification
    Zeng, Lili
    Ren, Weijian
    Shan, Liqun
    NEUROCOMPUTING, 2020, 414 : 153 - 171
  • [45] Attention-Based Convolutional Recurrent Neural Network for Traffic Index Prediction
    Liu, Xiaolei
    Duan, Zhengyu
    Luo, Songwen
    Zhao, Yixin
    CICTP 2023: INNOVATION-EMPOWERED TECHNOLOGY FOR SUSTAINABLE, INTELLIGENT, DECARBONIZED, AND CONNECTED TRANSPORTATION, 2023, : 1140 - 1149
  • [46] Vehicle speed prediction using a convolutional neural network combined with a gated recurrent unit with attention
    Zhang, Dongxue
    Wang, Zhennan
    Jiao, Xiaohong
    Zhang, Zhao
    PROCEEDINGS OF THE INSTITUTION OF MECHANICAL ENGINEERS PART D-JOURNAL OF AUTOMOBILE ENGINEERING, 2024,
  • [47] Attention-based Recurrent Neural Network for Traffic Flow Prediction
    Chen, Qi
    Wang, Wei
    Huang, Xin
    Liang, Hai-ning
    JOURNAL OF INTERNET TECHNOLOGY, 2020, 21 (03): : 831 - 839
  • [48] Hybrid Deep Learning Network Intrusion Detection System Based on Convolutional Neural Network and Bidirectional Long Short-Term Memory
    Jihado, Anindra Ageng
    Girsang, Abba Suganda
    JOURNAL OF ADVANCES IN INFORMATION TECHNOLOGY, 2024, 15 (02) : 219 - 232
  • [49] Predicting Short-term Traffic Flow by Long Short-Term Memory Recurrent Neural Network
    Tian, Yongxue
    Pan, Li
    2015 IEEE INTERNATIONAL CONFERENCE ON SMART CITY/SOCIALCOM/SUSTAINCOM (SMARTCITY), 2015, : 153 - 158
  • [50] Hybrid Elman neural network model for short-term traffic prediction
    Weng Xiao-xiong
    Du Gao-li
    PROCEEDINGS OF THE EIGHTH IASTED INTERNATIONAL CONFERENCE ON CONTROL AND APPLICATIONS, 2006, : 281 - +