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.

Enhanced Traffic Flow Prediction with Multi-Segment Fusion Tensor Graph Convolutional Networks

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

Temporal Multi-Graph Convolutional Network for Traffic Flow Prediction

A Multi-Stream Feature Fusion Approach for Traffic Prediction

Multi-Source Information Fusion Graph Convolution Network for traffic flow prediction

MFDGCN: Multi-Stage Spatio-Temporal Fusion Diffusion Graph Convolutional Network for Traffic Prediction

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

MFSTGN: a Multi-Scale Spatial-Temporal Fusion Graph Network for Traffic Prediction

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

STFGCN: Spatial-temporal Fusion Graph Convolutional Network for Traffic Prediction

Deep Spatial-Temporal Information Fusion Dynamic Graph Convolutional Network for Traffic Flow Prediction

Dynamic Graph Convolutional Network with Attention Fusion for Traffic Flow Prediction

LGTCN: A Spatial–Temporal Traffic Flow Prediction Model Based on Local–Global Feature Fusion Temporal Convolutional Network

Traffic flow prediction based on temporal multigraph convolutional neural network

FD-TGCN: Fast and dynamic temporal graph convolution network for traffic flow prediction

Expressway traffic flow prediction based on MF-TAN and STSA

Spatial–Temporal Traffic Prediction Model Based on Adaptive Graphs Fusion and Dual-Graph Collaborative Convolution

Attention Based Spatial-Temporal Graph Convolutional Networks for Traffic Flow Forecasting

TADGCN: A Time-Aware Dynamic Graph Convolution Network for Long-Term Traffic Flow Prediction

Residual Attention Enhanced Time-varying Multi-Factor Graph Convolutional Network for Traffic Flow Prediction

A traffic flow-forecasting model based on multi-head spatio–temporal attention and adaptive graph convolutional networks