Modeling Global Spatial–Temporal Graph Attention Network for Traffic Prediction

Abstract

Accurate and efficient traffic prediction is the key to the realization of intelligent transportation system (ITS), which helps to alleviate traffic congestion and reduce traffic accidents. Due to the complex dynamic spatial-temporal dependence between traffic networks, traffic prediction is extremely challenging. In previous studies, convolution neural network (CNN) and graph convolution network (GCN) were used to model spatial correlation. However, the non-Euclidean correlation of road network reduces the effect of convolution operator modeling. In addition, only considering the traffic interaction around the concerned points simplifies the influence of traffic network. In order to address the above problems, this article proposes an end-to-end global spatial-temporal graph attention network (GST-GAT), which uses the “global interaction + node query” to model the dynamic spatial-temporal correlation of traffic. In the encoder, the long short-term memory (LSTM) component flexibly transforms the traffic dynamic spatial-temporal graph into feedforward differentiable features. Global traffic interaction is proposed to summarize traffic network context changes and integrate all node features at each moment through a forward calculation. Then, each node computes the influence of traffic global interaction on a single node in parallel, and the spatial-temporal interaction information is adaptive fused by gating fusion mechanism. Finally, the end-to-end network structure is used to train the rich mixed feature coding to generate the traffic prediction status of each node. Experiments on public transportation data sets show that GST-GAT performs better than previous work in terms of accuracy and inference speed.