Abstract:Traffic flow prediction, which provides dynamic data support for intelligent transportation systems, has always been a topic of great interest. Accurate prediction of traffic flow is vital for optimizing traffic management and planning and improving traffic efficiency. Existing traffic flow prediction models usually consider only a single influencing factor when modeling spatial-temporal correlation and fail to comprehensively consider the multiscale factors corresponding to each module, resulting in incomplete spatial-temporal modeling and thus affecting prediction accuracy. For this reason, this paper proposes a spatial-temporal multifactor fusion graph neural network (STMFGNN), which aims to characterize spatial-temporal features more comprehensively. First, in the spatial feature extraction module, we parallelly utilize dynamic similarity graphs and static adjacency graphs to perform graph convolution and introduce the gated fusion module to self-learn the dynamic influence weight so that our model can integrate the information of global hidden knowledge and local prior knowledge and capture the multiscale spatial dependencies between nodes. Second, the model combines gated tanh unit convolution with a multireceptive field and gated recurrent units in the temporal feature extraction module. Similarly, it utilizes the gated fusion module to adaptively and dynamically adjust the importance weights of the two components. This enables the acquisition of multiscale temporal dependency information at both short-term and long-term levels. In addition, we employ an improved generative adversarial imputation network for incomplete traffic flow data. The experimental results on four real-world datasets show that our proposed method consistently outperforms other baseline models, achieving state-of-the-art performance. The key source code and data are available at https://github.com/Keaiii3/STMFGNN.

Automated Spatio-Temporal Synchronous Modeling with Multiple Graphs for Traffic Prediction

A Spatial-Temporal Gated Hypergraph Convolution Network for Traffic Prediction

Adaptive Spatio-temporal Graph Neural Network for traffic forecasting

Spatio-Temporal Graph Neural Network for Traffic Prediction Based on Adaptive Neighborhood Selection

Spatial-temporal multi-factor fusion graph neural network for traffic prediction

A Spatio-Temporal Graph Neural Network Approach for Traffic Flow Prediction

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

Dynamic multi-scale spatial-temporal graph convolutional network for traffic flow prediction

Make More Connections: Urban Traffic Flow Forecasting with Spatiotemporal Adaptive Gated Graph Convolution Network

Adaptive spatial-temporal graph attention networks for traffic flow forecasting

Dynamic multiple-graph spatial-temporal synchronous aggregation framework for traffic prediction in intelligent transportation systems

Spatial-temporal graph neural network based on gated convolution and topological attention for traffic flow prediction

Spatial–Temporal Dynamic Graph Convolutional Network With Interactive Learning for Traffic Forecasting

Spatial–Temporal Complex Graph Convolution Network for Traffic Flow Prediction

Traffic Flow Prediction via Spatial Temporal Graph Neural Network

MST-GNN: graph neural network with multi-granularity in space and time for traffic prediction

Dynamic Multi-Granularity Spatial-Temporal Graph Attention Network for Traffic Forecasting

Spatial-temporal correlated graph neural networks based on neighborhood feature selection for traffic data prediction

Spatio-Temporal Graph Convolutional Networks: A Deep Learning Framework for Traffic Forecasting

Spatial-Temporal Interactive Dynamic Graph Convolution Network for Traffic Forecasting

Spatial-Temporal Interactive Graph Neural Network for Traffic Forecasting.