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

Traffic Flow Prediction with Attention Mechanism Based on TS-NAS

Author

Listed:
  • Cai Zhao

    (Institute of Public-Safety and Big Data, College of Data Science, Taiyuan University of Technology, Taiyuan 030024, China
    Center of Information Management and Development, Taiyuan University of Technology, Taiyuan 030024, China)

  • Ruijing Liu

    (Institute of Public-Safety and Big Data, College of Data Science, Taiyuan University of Technology, Taiyuan 030024, China)

  • Bei Su

    (Institute of Public-Safety and Big Data, College of Data Science, Taiyuan University of Technology, Taiyuan 030024, China)

  • Lei Zhao

    (Institute of Public-Safety and Big Data, College of Data Science, Taiyuan University of Technology, Taiyuan 030024, China)

  • Zhiyong Han

    (Institute of Public-Safety and Big Data, College of Data Science, Taiyuan University of Technology, Taiyuan 030024, China)

  • Wen Zheng

    (Institute of Public-Safety and Big Data, College of Data Science, Taiyuan University of Technology, Taiyuan 030024, China
    Center for Big Data Research in Health, Changzhi Medical College, Changzhi 046000, China)

Abstract

The prediction of traffic flow is of great significance in the traffic field. However, because of the high uncertainty and complexity of traffic data, it is challenging that doing traffic flow prediction. Most of the existing methods have achieved good results in traffic flow prediction, but are not accurate enough to capture the dynamic temporal and spatial relationship of data by using the structural information of traffic flow. In this study, we propose a traffic flow prediction method with temporal attention mechanism and spatial attention mechanism based on neural architecture search (TS-NAS). Firstly, based on temporal and spatial attention mechanisms, we design a new attention mechanism. Secondly, we define a novel model to learn temporal flow and space flow in traffic network. Finally, the proposed method uses different modules about time, space and convolution and neural architecture search to be used for optimizing the model. We use two datasets to test the method. Experimental results show that the performance of the method is better than that of the existing method.

Suggested Citation

  • Cai Zhao & Ruijing Liu & Bei Su & Lei Zhao & Zhiyong Han & Wen Zheng, 2022. "Traffic Flow Prediction with Attention Mechanism Based on TS-NAS," Sustainability, MDPI, vol. 14(19), pages 1-12, September.
    • Handle: RePEc:gam:jsusta:v:14:y:2022:i:19:p:12232-:d:926309
    as

    Download full text from publisher

    File URL: https://www.mdpi.com/2071-1050/14/19/12232/pdf
    Download Restriction: no
    File URL: https://www.mdpi.com/2071-1050/14/19/12232/
    Download Restriction: no
    ---><---

    Citations

    Citations are extracted by the CitEc Project, subscribe to its RSS feed for this item.
    as


    Cited by:

    1. Tianhe Lan & Xiaojing Zhang & Dayi Qu & Yufeng Yang & Yicheng Chen, 2023. "Short-Term Traffic Flow Prediction Based on the Optimization Study of Initial Weights of the Attention Mechanism," Sustainability, MDPI, vol. 15(2), pages 1-16, January.

    More about this item

    Keywords

    ;
    ;
    ;
    ;

    Statistics

    Access and download statistics

    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:14:y:2022:i:19:p:12232-:d:926309. 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.

    We have no bibliographic references for this item. You can help adding them by using 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.