STGSA: A Novel Spatial-Temporal Graph Synchronous Aggregation Model for Traffic Prediction

• Published in: IEEE/CAA journal of automatica sinica Vol. 10; no. 1; pp. 226 - 238
• Main Authors: Wei, Zebing, Zhao, Hongxia, Li, Zhishuai, Bu, Xiaojie, Chen, Yuanyuan, Zhang, Xiqiao, Lv, Yisheng, Wang, Fei-Yue
• Format: Journal Article
• Language: English
• Published: Piscataway Chinese Association of Automation (CAA) 01.01.2023; The Institute of Electrical and Electronics Engineers, Inc. (IEEE); State Key Laboratory for Management and Control of Complex Systems,Institute of Automation,Chinese Academy of Sciences,Beijing 100190,China; School of Artificial Intelligence,University of Chinese Academy of Sciences,Beijing 100049,China%State Key Laboratory for Management and Control of Complex Systems,Institute of Automation,Chinese Academy of Sciences,Beijing 100190,China%College of Science,China University of Petroleum,Qingdao 266580,China%Department of Transportation Engineering,School of Transportation Science and Technology,Harbin Institute of Technology,Harbin 150090,China%State Key Laboratory for Management and Control of Complex Systems,Institute of Automation,Chinese Academy of Sciences,Beijing 100190,China; Shandong Key Laboratory of Smart Transportation(Preparation),Shandong Jiao Tong University,Shanghai 200240,China
• Subjects: Agglomeration; Benchmark testing; Convolution; Correlation; Deep learning; Feature extraction; graph neural network (GNN); Intelligent transportation systems; Modules; multi-stream; Nodes; Predictive models; Roads; spatial-temporal feature extraction; Stacking; Subspaces; temporal graph; Traffic information; Traffic models; traffic prediction; Deep learning; temporal graph; spatial-temporal feature extraction; multi-stream; graph neural network (GNN); traffic prediction
• ISSN: 2329-9266; 2329-9274
• DOI: 10.1109/JAS.2023.123033

The success of intelligent transportation systems relies heavily on accurate traffic prediction, in which how to model the underlying spatial-temporal information from traffic data has come under the spotlight. Most existing frameworks typically utilize separate modules for spatial and temporal correlations modeling. However, this stepwise pattern may limit the effectiveness and efficiency in spatial-temporal feature extraction and cause the overlook of important information in some steps. Furthermore, it is lacking sufficient guidance from prior information while modeling based on a given spatial adjacency graph (e.g., deriving from the geodesic distance or approximate connectivity), and may not reflect the actual interaction between nodes. To overcome those limitations, our paper proposes a spatial-temporal graph synchronous aggregation (STGSA) model to extract the localized and long-term spatial-temporal dependencies simultaneously. Specifically, a tailored graph aggregation method in the vertex domain is designed to extract spatial and temporal features in one graph convolution process. In each STGSA block, we devise a directed temporal correlation graph to represent the localized and long-term dependencies between nodes, and the potential temporal dependence is further fine-tuned by an adaptive weighting operation. Meanwhile, we construct an elaborated spatial adjacency matrix to represent the road sensor graph by considering both physical distance and node similarity in a data-driven manner. Then, inspired by the multi-head attention mechanism which can jointly emphasize information from different representation subspaces, we construct a multi-stream module based on the STGSA blocks to capture global information. It projects the embedding input repeatedly with multiple different channels. Finally, the predicted values are generated by stacking several multi-stream modules. Extensive experiments are constructed on six real-world datasets, and numerical results show that the proposed STGSA model significantly outperforms the benchmarks.