STG4Traffic: A Survey and Benchmark of Spatial-Temporal Graph Neural Networks for Traffic Prediction

by   Xunlian Luo, et al.

Traffic prediction has been an active research topic in the domain of spatial-temporal data mining. Accurate real-time traffic prediction is essential to improve the safety, stability, and versatility of smart city systems, i.e., traffic control and optimal routing. The complex and highly dynamic spatial-temporal dependencies make effective predictions still face many challenges. Recent studies have shown that spatial-temporal graph neural networks exhibit great potential applied to traffic prediction, which combines sequential models with graph convolutional networks to jointly model temporal and spatial correlations. However, a survey study of graph learning, spatial-temporal graph models for traffic, as well as a fair comparison of baseline models are pending and unavoidable issues. In this paper, we first provide a systematic review of graph learning strategies and commonly used graph convolution algorithms. Then we conduct a comprehensive analysis of the strengths and weaknesses of recently proposed spatial-temporal graph network models. Furthermore, we build a study called STG4Traffic using the deep learning framework PyTorch to establish a standardized and scalable benchmark on two types of traffic datasets. We can evaluate their performance by personalizing the model settings with uniform metrics. Finally, we point out some problems in the current study and discuss future directions. Source codes are available at


page 1

page 2

page 3

page 4


Adaptive Hybrid Spatial-Temporal Graph Neural Network for Cellular Traffic Prediction

Cellular traffic prediction is an indispensable part for intelligent tel...

Minimalist Traffic Prediction: Linear Layer Is All You Need

Traffic prediction is essential for the progression of Intelligent Trans...

Traffic Flow Forecasting with Spatial-Temporal Graph Diffusion Network

Accurate forecasting of citywide traffic flow has been playing critical ...

Spatial-Temporal Sequential Hypergraph Network for Crime Prediction

Crime prediction is crucial for public safety and resource optimization,...

Short-term Traffic Prediction with Deep Neural Networks: A Survey

In modern transportation systems, an enormous amount of traffic data is ...

Predicting COVID-19 pandemic by spatio-temporal graph neural networks: A New Zealand's study

Modeling and simulations of pandemic dynamics play an essential role in ...

Analysis of different temporal graph neural network configurations on dynamic graphs

In recent years, there has been an increasing interest in the use of gra...

Please sign up or login with your details

Forgot password? Click here to reset