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

NA-DGRU: A Dual-GRU Traffic Speed Prediction Model Based on Neighborhood Aggregation and Attention Mechanism

Author

Listed:
  • Xiaoping Tian

    (College of Information Engineering, Beijing Institute of Petrochemical Technology, Beijing 102617, China)

  • Changkuan Zou

    (College of Information Engineering, Beijing Institute of Petrochemical Technology, Beijing 102617, China)

  • Yuqing Zhang

    (College of Information Engineering, Beijing Institute of Petrochemical Technology, Beijing 102617, China)

  • Lei Du

    (College of Information Engineering, Beijing Institute of Petrochemical Technology, Beijing 102617, China)

  • Song Wu

    (College of Information Engineering, Beijing Institute of Petrochemical Technology, Beijing 102617, China)

Abstract

Traffic prediction is an important part of the Intelligent Transportation System (ITS) and has broad application prospects. However, traffic data are affected not only by time, but also by the traffic status of other nearby roads. They have complex temporal and spatial correlations. Developing a means for extracting specific features from them and effectively predicting traffic status such as road speed remains a huge challenge. Therefore, in order to reduce the speed prediction error and improve the prediction accuracy, this paper proposes a dual-GRU traffic speed prediction model based on neighborhood aggregation and the attention mechanism: NA-DGRU (Neighborhood aggregation and Attention mechanism–Dual GRU). NA-DGRU uses the neighborhood aggregation method to extract spatial features from the neighborhood space of the road, and it extracts the correlation between speed and time from the original features and neighborhood aggregation features through two GRUs, respectively. Finally, the attention model is introduced to collect and summarize the information of the road and its neighborhood in the global time to perform traffic prediction. In this paper, the prediction performance of NA-DGRU is tested on two real-world datasets, SZ-taxi and Los-loop. In the 15-, 30-, 45- and 60-min speed prediction results of NA-DGRU on the SZ-taxi dataset, the RMSE values were 4.0587, 4.0683, 4.0777 and 4.0851, respectively, and the MAE values were 2.7387, 2.728, 2.7393 and 2.7487; on the Los-loop dataset, the RMSE values for the speed prediction results were 5.1348, 6.1358, 6.7604 and 7.2776, respectively, and the MAE values were 3.0281, 3.6692, 4.0567 and 4.4256, respectively. On the SZ-taxi dataset, compared with other baseline methods, NA-DGRU demonstrated a maximum reduction in RMSE of 6.49% and a maximum reduction in MAE of 6.17%; on the Los-loop dataset, the maximum reduction in RMSE was 31.01%, and the maximum reduction in MAE reached 24.89%.

Suggested Citation

  • Xiaoping Tian & Changkuan Zou & Yuqing Zhang & Lei Du & Song Wu, 2023. "NA-DGRU: A Dual-GRU Traffic Speed Prediction Model Based on Neighborhood Aggregation and Attention Mechanism," Sustainability, MDPI, vol. 15(4), pages 1-20, February.
    • Handle: RePEc:gam:jsusta:v:15:y:2023:i:4:p:2927-:d:1059431
    as

    Download full text from publisher

    File URL: https://www.mdpi.com/2071-1050/15/4/2927/pdf
    Download Restriction: no
    File URL: https://www.mdpi.com/2071-1050/15/4/2927/
    Download Restriction: no
    ---><---

    References listed on IDEAS

    as
    1. Wang, Jun & Wang, Wenjun & Liu, Xueli & Yu, Wei & Li, Xiaoming & Sun, Peiliang, 2022. "Traffic prediction based on auto spatiotemporal Multi-graph Adversarial Neural Network," Physica A: Statistical Mechanics and its Applications, Elsevier, vol. 590(C).
    2. Isaac Oyeyemi Olayode & Lagouge Kwanda Tartibu & Modestus O. Okwu & Alessandro Severino, 2021. "Comparative Traffic Flow Prediction of a Heuristic ANN Model and a Hybrid ANN-PSO Model in the Traffic Flow Modelling of Vehicles at a Four-Way Signalized Road Intersection," Sustainability, MDPI, vol. 13(19), pages 1-28, September.
    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. Isaac Oyeyemi Olayode & Alessandro Gaetano Severino & Tiziana Campisi & Lagouge Kwanda Tartibu, 2022. "Comprehensive Literature Review on the Impacts of COVID-19 Pandemic on Public Road Transportation System: Challenges and Solutions," Sustainability, MDPI, vol. 14(15), pages 1-23, August.
    2. Zhang, Ke & Lin, Xi & Li, Meng, 2023. "Graph attention reinforcement learning with flexible matching policies for multi-depot vehicle routing problems," Physica A: Statistical Mechanics and its Applications, Elsevier, vol. 611(C).
    3. Enci Liu & Jie Li & Anni Zheng & Haoran Liu & Tao Jiang, 2022. "Research on the Prediction Model of the Used Car Price in View of the PSO-GRA-BP Neural Network," Sustainability, MDPI, vol. 14(15), pages 1-19, July.
    4. Tian, Jing & Song, Xianmin & Tao, Pengfei & Liang, Jiahui, 2022. "Pattern-adaptive generative adversarial network with sparse data for traffic state estimation," Physica A: Statistical Mechanics and its Applications, Elsevier, vol. 608(P1).
    5. Zibin Wei & Tao Peng & Sijia Wei, 2022. "A Robust Adaptive Traffic Signal Control Algorithm Using Q-Learning under Mixed Traffic Flow," Sustainability, MDPI, vol. 14(10), pages 1-16, May.
    6. Wang, Yaguan & Qin, Yong & Guo, Jianyuan & Cao, Zhiwei & Jia, Limin, 2022. "Multi-point short-term prediction of station passenger flow based on temporal multi-graph convolutional network," Physica A: Statistical Mechanics and its Applications, Elsevier, vol. 604(C).
    7. Liu, Shuai & Feng, Xiaoyuan & Ren, Yilong & Jiang, Han & Yu, Haiyang, 2023. "DCENet: A dynamic correlation evolve network for short-term traffic prediction," Physica A: Statistical Mechanics and its Applications, Elsevier, vol. 614(C).
    8. Ma, Changxi & Zhao, Mingxi, 2023. "Spatio-temporal multi-graph convolutional network based on wavelet analysis for vehicle speed prediction," Physica A: Statistical Mechanics and its Applications, Elsevier, vol. 630(C).

    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:15:y:2023:i:4:p:2927-:d:1059431. 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.