Abstract Due to its importance in Intelligent Transport Systems (ITS), traffic flow prediction has been the focus of many studies in the last few decades. Existing traffic flow prediction models mainly extract static spatial-temporal correlations, although these correlations are known to be dynamic in traffic networks. Attention-based models have emerged in recent years, mostly in the field of natural language processing, and have resulted in major progresses in terms of both accuracy and interpretability. This inspires us to introduce the application of attentions for traffic flow prediction. In this study, a deep learning based traffic flow predictor with spatial and temporal attentions (STANN) is proposed. The spatial and temporal attentions are used to exploit the spatial dependencies between road segments and temporal dependencies between time steps respectively. Experiment results with a real-world traffic dataset demonstrate the superior performance of the proposed model. The results also show that the utilization of multiple data resolutions could help improve prediction accuracy. Furthermore, the proposed model is demonstrated to have potential for improving the understanding of spatial-temporal correlations in a traffic network.


    Access

    Check access

    Check availability in my library

    Order at Subito €


    Export, share and cite



    Title :

    An effective spatial-temporal attention based neural network for traffic flow prediction


    Contributors:
    Do, Loan N.N. (author) / Vu, Hai L. (author) / Vo, Bao Q. (author) / Liu, Zhiyuan (author) / Phung, Dinh (author)


    Publication date :

    2019-09-10


    Size :

    17 pages




    Type of media :

    Article (Journal)


    Type of material :

    Electronic Resource


    Language :

    English




    A Short-term Traffic Flow Forecasting Model Based on Spatial-temporal Attention Neural Network

    Dong, Honghui / Zhu, Pengcheng / Gao, Jiayang et al. | IEEE | 2022



    Spatial-temporal Traffic Congestion Prediction Based on Attention Mechanism

    Pu, Shilin / Chu, Liang / Zhang, Yuanjian et al. | IEEE | 2021


    Spatial-Temporal Multiscale Fusion Graph Neural Network for Traffic Flow Prediction

    Hou, Hongxin / Ning, Nianwen / Shi, Huaguang et al. | IEEE | 2022


    STATNet: Spatial-temporal attention in the traffic prediction

    Moghadas, Seyed Mohamad / Gheibi, Amin / Alahi, Alexander | TIBKAT | 2022