TY  - JOUR
PY  - 2024//
TI  - Spatio-temporal graph neural network for traffic prediction based on adaptive neighborhood selection
JO  - Transportation research record
A1  - Liu, FangJie
A1  - Lu, JianGuang
A1  - Tang, XiangHong
A1  - Sun, HuanZhong
SP  - 641
EP  - 655
VL  - 2678
IS  - 6
N2  - 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.<p />  <p>Language: en</p>
LA  - en
SN  - 0361-1981
UR  - http://dx.doi.org/10.1177/03611981231198851
ID  - ref1
ER  -