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.

Short-Term Traffic Condition Prediction Based on Multi-source Data Fusion

Short-Term Forecasting of Traffic Flow and Speed: A Deep Learning Approach

Short-Term Traffic Speed Prediction of Urban Road with Multi-Source Data

Fuzzy Hybrid Framework with Dynamic Weights for Short‐term Traffic Flow Prediction by Mining Spatio‐temporal Correlations

A Short-term Traffic Flow Forecasting Method Based on Combination of Fuzzy C-mean Clustering and Neural Network

Daily Long-Term Traffic Flow Forecasting Based on a Deep Neural Network

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

A Multi-Stream Feature Fusion Approach for Traffic Prediction

A Multifeature Fusion Short-Term Traffic Flow Prediction Model Based on Deep Learnings

Spatial-Temporal Traffic Flow Prediction with Fusion Graph Convolution Network and Enhanced Gated Recurrent Units

Short-Term Traffic Flow Prediction Based on Sparse Regression and Spatio-Temporal Data Fusion

An Effective Dynamic Spatio-temporal Framework with Multi-Source Information for Traffic Prediction

Hybrid Multi-Sensor Data for Traffic Condition Forecasting.

Traffic Flow Prediction Based on Depthwise Separable Convolution Fusion Network

Combined Neural Network Approach for Short-Term Urban Freeway Traffic Flow Prediction

Multi-Scale Non-Local Spatio-Temporal Information Fusion Networks for Multi-Step Traffic Flow Forecasting

A multi-feature spatial–temporal fusion network for traffic flow prediction

A Deep Learning Approach to the Prediction of Short-term Traffic Accident Risk

Global Spatio-Temporal Fusion-based Traffic Prediction Algorithm with Anomaly Aware

Short-term Traffic Flow Prediction Based on Multi-Auxiliary Information

Condition-Guided Urban Traffic Co-Prediction with Multiple Sparse Surveillance Data