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




    A macro–micro spatio-temporal neural network for traffic prediction

    Feng, Siyuan / Wei, Shuqing / Zhang, Junbo et al. | Elsevier | 2023


    Spatio‐temporal adaptive graph convolutional networks for traffic flow forecasting

    Ma, Qiwei / Sun, Wei / Gao, Junbo et al. | Wiley | 2023

    Free access

    Spatio‐temporal adaptive graph convolutional networks for traffic flow forecasting

    Qiwei Ma / Wei Sun / Junbo Gao et al. | DOAJ | 2023

    Free access

    Automated Dilated Spatio-Temporal Synchronous Graph Modeling for Traffic Prediction

    Jin, Guangyin / Li, Fuxian / Zhang, Jinlei et al. | IEEE | 2023