Traffic flow forecasting is an essential task of an intelligent transportation system (ITS), closely related to intelligent transportation management and resource scheduling. Dynamic spatial-temporal dependencies in traffic data make traffic flow forecasting to be a challenging task. Most existing research cannot model dynamic spatial and temporal correlations to achieve well-forecasting performance. The multi-head self-attention mechanism is a valuable method to capture dynamic spatial-temporal correlations, and combining it with graph convolutional networks is a promising solution. Therefore, we propose a multi-head self-attention spatiotemporal graph convolutional network (MSASGCN) model. It can effectively capture local correlations and potential global correlations of spatial structures, can handle dynamic evolution of the road network, and, in the time dimension, can effectively capture dynamic temporal correlations. Experiments on two real datasets verify the stability of our proposed model, obtaining a better prediction performance than the baseline algorithms. The correlation metrics get significantly reduced compared with traditional time series prediction methods and deep learning methods without using graph neural networks, according to MAE and RMSE results. Compared with advanced traffic flow forecasting methods, our model also has a performance improvement and a more stable prediction performance. We also discuss some problems and challenges in traffic forecasting.


    Access

    Download


    Export, share and cite



    Title :

    MSASGCN :  Multi-Head Self-Attention Spatiotemporal Graph Convolutional Network for Traffic Flow Forecasting


    Contributors:
    Yang Cao (author) / Detian Liu (author) / Qizheng Yin (author) / Fei Xue (author) / Hengliang Tang (author)


    Publication date :

    2022




    Type of media :

    Article (Journal)


    Type of material :

    Electronic Resource


    Language :

    Unknown




    Attention-Based Spatiotemporal Adaptive Graph Diffusion Convolutional Network For Traffic Flow Prediction

    He, Qiansong / Xia, Dawen / Li, Jianjun et al. | Transportation Research Record | 2025


    Graph Attention Convolutional Network: Spatiotemporal Modeling for Urban Traffic Prediction

    Song, Qingyu / Ming, RuiBo / Hu, Jianming et al. | IEEE | 2020


    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

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

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