Accurate traffic state prediction plays an important role in traffic guidance, travel planning, etc. Due to the existence of complex spatio-temporal relationships, there are some challenges in forecasting. Firstly, in terms of spatial correlation, some models only consider the road network structure information, and ignore the relative location relationships between nodes. Secondly, some models ignore the different impacts of nodes in the global road network on traffic. To solve these problems, we propose a new traffic state-forecasting model, namely, spatio-temporal attention-gated recurrent neural network (ST-AGRNN). In the proposed model, structure-based and location-based localized spatial features are obtained simultaneously by Graph Convolutional Networks (GCNs) and DeepWalk. The localized temporal features are obtained by gated recurrent unit (GRU). The attention-based approach is used to obtain global spatio-temporal features. Experimental validation is performed with two real-world public datasets, and the results show that the ST-AGRNN model outperforms the state-of-the-art methods.


    Access

    Download


    Export, share and cite



    Title :

    ST-AGRNN: A Spatio-Temporal Attention-Gated Recurrent Neural Network for Traffic State Forecasting


    Contributors:
    Jian Yang (author) / Jinhong Li (author) / Lu Wei (author) / Lei Gao (author) / Fuqi Mao (author)


    Publication date :

    2022




    Type of media :

    Article (Journal)


    Type of material :

    Electronic Resource


    Language :

    Unknown




    Spatio-Temporal Graph Attention Convolution Network for Traffic Flow Forecasting

    Liu, Kun / Zhu, Yifan / Wang, Xiao et al. | Transportation Research Record | 2024


    Attention-based Gated Recurrent Unit for Links Traffic Speed Forecasting

    Khodabandelou, Ghazaleh / Katranji, Mehdi / Kraiem, Sami et al. | IEEE | 2019


    Traffic Flow Forecasting of Graph Convolutional Network Based on Spatio-Temporal Attention Mechanism

    Zhang, Hong / Chen, Linlong / Cao, Jie et al. | Springer Verlag | 2023


    GSTGAT: Gated spatiotemporal graph attention network for traffic demand forecasting

    Shuilin Yao / Huizhen Zhang / Chenxi Wang et al. | DOAJ | 2024

    Free access

    GSTGAT: Gated spatiotemporal graph attention network for traffic demand forecasting

    Yao, Shuilin / Zhang, Huizhen / Wang, Chenxi et al. | Wiley | 2024

    Free access