Traffic prediction is critical to intelligent transportation and smart cities. The prediction performance of many existing traffic prediction models is limited by the fixed original graph structure and inappropriate spatio-temporal dependency extraction. For this situation, this paper proposes a spatio-temporal graph neural network based on adaptive neighborhood selection (STGNN-ANS). To obtain more flexible graph structures, STGNN-ANS designs a neighbor selection mechanism to generate a new graph structure by filtering inappropriate neighbors. To further capture the spatio-temporal dependence of traffic data, a spatio-temporal serial module of STGNN-ANS adopts the bidirectional learning manner of bidirectional long short-term memory (BiLSTM) and the graph convolution network (GCN) enhanced by self-attention mechanism to reach excellent prediction accuracy in both short-range and long-range scenarios. In this paper, a new baseline comprehensive comparison metric (BCCM) is invented to cope with the complexity in the comparative analysis of large numbers of experimental results. Many experiments have been performed on four real-world traffic datasets, and the results show that the comprehensive prediction performance of STGNN-ANS is better than previous models.


    Zugriff

    Download

    Verfügbarkeit in meiner Bibliothek prüfen

    Bestellung bei Subito €


    Exportieren, teilen und zitieren



    Titel :

    Spatio-Temporal Graph Neural Network for Traffic Prediction Based on Adaptive Neighborhood Selection


    Weitere Titelangaben:

    Transportation Research Record: Journal of the Transportation Research Board


    Beteiligte:
    Sun, HuanZhong (Autor:in) / Tang, XiangHong (Autor:in) / Lu, JianGuang (Autor:in) / Liu, FangJie (Autor:in)


    Erscheinungsdatum :

    09.09.2023




    Medientyp :

    Aufsatz (Zeitschrift)


    Format :

    Elektronische Ressource


    Sprache :

    Englisch



    Traffic Flow Prediction Based on Spatio-Temporal Aggregated Graph Neural Networks

    Wu, Shuangshuang / Hu, Yao | Transportation Research Record | 2025


    Bi-GRCN: A Spatio-Temporal Traffic Flow Prediction Model Based on Graph Neural Network

    Wenhao Jiang / Yunpeng Xiao / Yanbing Liu et al. | DOAJ | 2022

    Freier Zugriff


    Spatio-temporal graph attention networks for traffic prediction

    Ma, Chuang / Yan, Li / Xu, Guangxia | Taylor & Francis Verlag | 2024


    Traffic Congestion Prediction Based on Spatio-Temporal Graph Structure Learning

    Bannur, Charvi / Bhat, Chaitra / Goutham, Gagan et al. | IEEE | 2023