IDEAS home Printed from https://ideas.repec.org/a/gam/jsusta/v16y2024i5p1986-d1347701.html
   My bibliography  Save this article

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

Author

Listed:
  • Song Liu

    (Institute for Key Laboratory of Traffic System, Chongqing Jiaotong University, Chongqing 400074, China
    School of Traffic and Transportation, Chongqing Jiaotong University, Chongqing 400074, China
    Institute for Intelligent Optimization of Comprehensive Transportation Systems, Chongqing Jiaotong University, Chongqing 400074, China
    College of Civil Aviation, Nanjing University of Aeronautics and Astronautics, Nanjing 210016, China)

  • Wenting Lin

    (School of Traffic and Transportation, Chongqing Jiaotong University, Chongqing 400074, China)

  • Yue Wang

    (Highway Service Center of Yongchuan District, Chongqing 402160, China)

  • Dennis Z. Yu

    (The David D. Reh School of Business, Clarkson University, Potsdam, NY 13699, USA)

  • Yong Peng

    (School of Traffic and Transportation, Chongqing Jiaotong University, Chongqing 400074, China
    Research Center for Transportation and International Supply Chain Management, Chongqing Jiaotong University, Chongqing 400074, China)

  • Xianting Ma

    (Institute for Key Laboratory of Traffic System, Chongqing Jiaotong University, Chongqing 400074, China
    School of Traffic and Transportation, Chongqing Jiaotong University, Chongqing 400074, China)

Abstract

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 ( R 2 ) 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.

Suggested Citation

  • Song Liu & Wenting Lin & Yue Wang & Dennis Z. Yu & Yong Peng & Xianting Ma, 2024. "Convolutional Neural Network-Based Bidirectional Gated Recurrent Unit–Additive Attention Mechanism Hybrid Deep Neural Networks for Short-Term Traffic Flow Prediction," Sustainability, MDPI, vol. 16(5), pages 1-15, February.
  • Handle: RePEc:gam:jsusta:v:16:y:2024:i:5:p:1986-:d:1347701
    as

    Download full text from publisher

    File URL: https://www.mdpi.com/2071-1050/16/5/1986/pdf
    Download Restriction: no

    File URL: https://www.mdpi.com/2071-1050/16/5/1986/
    Download Restriction: no
    ---><---

    References listed on IDEAS

    as
    1. Zhou, Guangzhao & Guo, Zanquan & Sun, Simin & Jin, Qingsheng, 2023. "A CNN-BiGRU-AM neural network for AI applications in shale oil production prediction," Applied Energy, Elsevier, vol. 344(C).
    2. Gyeongjae Lee & Sangho Choo & Sungtaek Choi & Hyangsook Lee, 2022. "Does the Inclusion of Spatio-Temporal Features Improve Bus Travel Time Predictions? A Deep Learning-Based Modelling Approach," Sustainability, MDPI, vol. 14(12), pages 1-20, June.
    Full references (including those not matched with items on IDEAS)

    Most related items

    These are the items that most often cite the same works as this one and are cited by the same works as this one.
    1. Zhou, Wei & Li, Xiangchengzhen & Qi, ZhongLi & Zhao, HaiHang & Yi, Jun, 2024. "A shale gas production prediction model based on masked convolutional neural network," Applied Energy, Elsevier, vol. 353(PA).

    Corrections

    All material on this site has been provided by the respective publishers and authors. You can help correct errors and omissions. When requesting a correction, please mention this item's handle: RePEc:gam:jsusta:v:16:y:2024:i:5:p:1986-:d:1347701. See general information about how to correct material in RePEc.

    If you have authored this item and are not yet registered with RePEc, we encourage you to do it here. This allows to link your profile to this item. It also allows you to accept potential citations to this item that we are uncertain about.

    If CitEc recognized a bibliographic reference but did not link an item in RePEc to it, you can help with this form .

    If you know of missing items citing this one, you can help us creating those links by adding the relevant references in the same way as above, for each refering item. If you are a registered author of this item, you may also want to check the "citations" tab in your RePEc Author Service profile, as there may be some citations waiting for confirmation.

    For technical questions regarding this item, or to correct its authors, title, abstract, bibliographic or download information, contact: MDPI Indexing Manager (email available below). General contact details of provider: https://www.mdpi.com .

    Please note that corrections may take a couple of weeks to filter through the various RePEc services.

    IDEAS is a RePEc service. RePEc uses bibliographic data supplied by the respective publishers.