A two-tower spatial-temporal graph neural network for traffic speed prediction

Yansong SHEN, Lin LI, Qing XIE, Xin LI, Guandong XU

Research output: Chapter in Book/Report/Conference proceedingChapters

2 Citations (Scopus)

Abstract

Recently, the remarkable effect of applying Dynamic Graph Neural Networks (DGNNs) to traffic speed prediction has received wide attention. Existing DGNN-based researches usually use a pre-defined or an adaptive matrix to capture the spatial correlations in traffic data. However, these static matrices are not enough to match the dynamic characteristics of spatial correlations. We argue that the global changes and local fluctuations of spatial correlations are dynamic with different frequencies. To this end, in this paper, we propose a Two-Tower DGNN (T² -GNN) framework which divides the traffic data into a seasonal static component and an acyclic dynamic component, thus enhancing traffic speed prediction. The two components generated by an auto-decomposing block reflect global changes and local fluctuations of spatial correlations, respectively. Moreover, we use two parallel dynamic graph generation layers to construct a seasonal graph and an acyclic graph at each time step. In this way, the high-level representations of these two kinds of dynamic changes are learned through two dynamic graph convolution layers. Besides, the impact of fixed road network structure is modeled on the pre-defined graph and added to the spatial correlations. And we capture temporal correlations in temporal block before modeling spatial correlations. Finally, skip connections are used to converge the spatial-temporal correlations for final prediction. Experimental results on an urban dataset and two highway datasets show our proposed framework achieves the state-of-the-art prediction performances in terms of Mean Average Error (MAE) and Root Mean Squared Error (RMSE). Copyright © 2022 The Author(s), under exclusive license to Springer Nature Switzerland AG.

Original languageEnglish
Title of host publicationAdvances in knowledge discovery and data mining: 26th Pacific-Asia Conference, PAKDD 2022, proceedings, part I
EditorsJoão GAMA, Tianrui LI, Yang YU, Enhong CHEN, Yu Zheng, Fei TENG
PublisherSpringer
Pages406-418
ISBN (Print)9783031059322
DOIs
Publication statusPublished - 2022

Citation

Shen, Y., Li, L., Xie, Q., Li, X., & Xu, G. (2022). A two-tower spatial-temporal graph neural network for traffic speed prediction. In J. Gama, T. Li, Y. Yu, E. Chen, Y. Zheng, & F. Teng (Eds.), Advances in knowledge discovery and data mining: 26th Pacific-Asia Conference, PAKDD 2022, proceedings, part I (pp. 406-418). Springer. https://doi.org/10.1007/978-3-031-05933-9_32

Fingerprint

Dive into the research topics of 'A two-tower spatial-temporal graph neural network for traffic speed prediction'. Together they form a unique fingerprint.