A heterogeneous traffic spatio-temporal graph convolution model for traffic prediction

Abstract

Smart cities require advanced traffic management systems. Traffic forecasting is an essential task of the advanced transportation system. Traffic spatio-temporal data are often heterogeneous. Most existing traffic prediction models predominantly use separate components to extract the temporal and spatial features of traffic data. However, this overlooks the intrinsic connections between the spatio-temporal features of traffic data. To directly mine the spatio-temporal heterogeneity, this study constructs a global heterogeneous traffic spatio-temporal graph and proposes the Heterogeneous Traffic Spatio-Temporal Graph Convolution (HTSTGC). To reduce the complexity of the model, Simple Graph Convolution (SGC) is used to extract semi-structured meta-graph information. The receptive fields that capture temporal and spatial features can be flexibly adjusted separately through clever design, which can balance the performance and efficiency of the model. Finally, the feature fusion module applies Gated Graph Neural Network (GGNN) to fuse temporal and spatial features. The results on the PEMS datasets reveal that jointly modeling different types of relationships can improve the traffic prediction performance of the model. The proposed HTSTGC has better performance than the baseline methods in most cases. The research results can support urban traffic control, traffic pollution reduction, and sustainable urban development.