Spatial–Temporal Complex Graph Convolution Network for Traffic Flow Prediction

Abstract

Traffic flow prediction remains an ongoing hot topic in the field of Intelligent Transportation System. The state-of-the-art traffic flow prediction models can effectively extract both spatial and temporal features of traffic flow data, but ignore the correlation and external interference between traffic nodes. To this end, this paper proposes a novel method based on Spatial–Temporal Complex Graph Convolution Network (ST-CGCN) for traffic flow prediction. Specifically, we first constructs the distance matrix, the data correlation matrix, and the comfort measurement matrix according to the geographical locations, the historical data record, and the external interference between traffic nodes. Then, these three matrices are fused into a complex correlation matrix by introducing self-learning dynamic weights to improve the joint modeling ability of spatial–temporal features and external factors. Next, a spatial feature extraction module and a temporal feature extraction module are designed to characterize dynamic spatial–temporal features. The spatial feature extraction module consists of a graph convolution operator with a proposed complex correlation matrix and a residual unit. The temporal feature extraction module consists of a 3D convolution operator and a Long Short-Term Memory (LSTM). Experiments constructed on five real-world datasets demonstrate that the new proposed ST-CGCN is more effective than several existing deep learning based traffic flow prediction models. The key source code and data are available at https://github.com/Bounger2/ST-CGCN.