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 条
  • [1] Hybrid gated recurrent unit and convolutional neural network-based deep learning mechanism for efficient shilling attack detection in social networks
    Praveena, N.
    Juneja, Kapil
    Rashid, Mamoon
    Almagrabi, Alaa Omran
    Sekaran, Kaushik
    Ramalingam, Rajakumar
    Usman, Muhammad
    COMPUTERS & ELECTRICAL ENGINEERING, 2023, 108
  • [2] Deep Learning Wind Power Prediction Model Based on Attention Mechanism-Based Convolutional Neural Network and Gated Recurrent Unit Neural Network
    Hou, Zai-Hong
    Bai, Yu-Long
    Ding, Lin
    Yue, Xiao-Xin
    Huang, Yu-Ting
    Song, Wei
    Bi, Qi
    JOURNAL OF CIRCUITS SYSTEMS AND COMPUTERS, 2024, 33 (16)
  • [3] Short-term wind power forecasting based on dual attention mechanism and gated recurrent unit neural network
    Xu, Wu
    Liu, Yang
    Fan, Xinhao
    Shen, Zhifang
    Wu, Qingchang
    FRONTIERS IN ENERGY RESEARCH, 2024, 12
  • [4] A Short-Term Traffic Flow Prediction Model Based on an Improved Gate Recurrent Unit Neural Network
    Shu, Wanneng
    Cai, Ken
    Xiong, Neal Naixue
    IEEE TRANSACTIONS ON INTELLIGENT TRANSPORTATION SYSTEMS, 2022, 23 (09) : 16654 - 16665
  • [5] Prediction of Network Security Situation Based on Attention Mechanism and Convolutional Neural Network-Gated Recurrent Unit
    Feng, Yuan
    Zhao, Hongying
    Zhang, Jianwei
    Cai, Zengyu
    Zhu, Liang
    Zhang, Ran
    APPLIED SCIENCES-BASEL, 2024, 14 (15):
  • [6] Short-term Traffic flow Prediction based on Deep Circulation Neural Network
    Liu, RuRu
    Hong, Feng
    Lu, Changhua
    Jiang, WeiWei
    2018 INTERNATIONAL SEMINAR ON COMPUTER SCIENCE AND ENGINEERING TECHNOLOGY (SCSET 2018), 2019, 1176
  • [7] Short-term Traffic Flow Prediction with LSTM Recurrent Neural Network
    Kang, Danqing
    Lv, Yisheng
    Chen, Yuan-yuan
    2017 IEEE 20TH INTERNATIONAL CONFERENCE ON INTELLIGENT TRANSPORTATION SYSTEMS (ITSC), 2017,
  • [8] Short-Term Load Forecasting Based on Adabelief Optimized Temporal Convolutional Network and Gated Recurrent Unit Hybrid Neural Network
    Shi, Hanhong
    Wang, Lei
    Scherer, Rafal
    Wozniak, Marcin
    Zhang, Pengchao
    Wei, Wei
    IEEE ACCESS, 2021, 9 : 66965 - 66981
  • [9] Attention-Based Gated Recurrent Graph Convolutional Network for Short-Term Traffic Flow Forecasting
    Lou, Ping
    Wu, Zihao
    Hu, Jiwei
    Liu, Quan
    Wei, Qin
    JOURNAL OF MATHEMATICS, 2023, 2023
  • [10] A combined method for short-term traffic flow prediction based on recurrent neural network
    Lu, Saiqun
    Zhang, Qiyan
    Chen, Guangsen
    Seng, Dewen
    ALEXANDRIA ENGINEERING JOURNAL, 2021, 60 (01) : 87 - 94