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.

Spatial-temporal traffic data imputation based on dynamic multi-level generative adversarial networks for urban governance

AGNP: Network-wide Short-Term Probabilistic Traffic Speed Prediction and Imputation

A Bipartite Graph Based Method for Traffic Continuous Data Imputation

Bidirectional Spatial-Temporal Traffic Data Imputation via Graph Attention Recurrent Neural Network

A Missing Traffic Data Imputation Method Based on a Diffusion Convolutional Neural Network–Generative Adversarial Network

Traffic Volume Imputation Using Attention-based Spatiotemporal Generative Adversarial Imputation Network

Modified GAN Model for Traffic Missing Data Imputation

Missing Data Repairs for Traffic Flow With Self-Attention Generative Adversarial Imputation Net

A novel reinforced dynamic graph convolutional network model with data imputation for network-wide traffic flow prediction

Robust spatial temporal imputation based on spatio-temporal generative adversarial nets

ST-LBAGAN: Spatio-temporal learnable bidirectional attention generative adversarial networks for missing traffic data imputation

Turning traffic volume imputation for persistent missing patterns with GNNs

A Latent-Factor-Model-Based Approach for Traffic Data Imputation with Road Network Information

An Experimental Evaluation of Imputation Models for Spatial-Temporal Traffic Data

A hybrid model for missing traffic flow data imputation based on clustering and attention mechanism optimizing LSTM and AdaBoost

A Kriging based spatiotemporal approach for traffic volume data imputation

A Customized Deep Learning Approach to Integrate Network-Scale Online Traffic Data Imputation and Prediction

A Hybrid Data-Driven Framework for Spatiotemporal Traffic Flow Data Imputation

Real-Time Traffic Speed Estimation for Smart Cities with Spatial Temporal Data: A Gated Graph Attention Network Approach

Graph Markov Network for Traffic Forecasting with Missing Data

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