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.


    Access

    Download

    Check availability in my library

    Order at Subito €


    Export, share and cite



    Title :

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


    Additional title:

    Transportation Research Record: Journal of the Transportation Research Board


    Contributors:


    Publication date :

    2023-09-09




    Type of media :

    Article (Journal)


    Type of material :

    Electronic Resource


    Language :

    English



    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

    Free access


    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