Abstract:Timely and accurate traffic flow prediction is crucial for stabilizing road conditions, reducing environmental pollution, and mitigating economic losses. While current graph convolution methods have achieved certain results, they do not fully leverage the true advantages of graph convolution. There is still room for improvement in simultaneously addressing multi-graph convolution, optimizing graphs, and simulating road conditions. Based on this, this paper proposes MSA-GCN: Multistage Spatio-Temporal Aggregation Graph Convolutional Networks for Traffic Flow Prediction. This method overcomes the aforementioned issues by dividing the process into different stages and achieves promising prediction results. In the first stage, we construct a latent similarity adjacency matrix and address the randomness interference features in similarity features through two optimizations using the proposed ConvGRU Attention Layer (CGAL module) and the Causal Similarity Capture Module (CSC module), which includes Granger causality tests. In the second stage, we mine the potential correlation between roads using the Correlation Completion Module (CC module) to create a global correlation adjacency matrix as a complement for potential correlations. In the third stage, we utilize the proposed Auto-LRU autoencoder to pre-train various weather features, encoding them into the model's prediction process to enhance its ability to simulate the real world and improve interpretability. Finally, in the fourth stage, we fuse these features and use a Bidirectional Gated Recurrent Unit (BiGRU) to model time dependencies, outputting the prediction results through a linear layer. Our model demonstrates a performance improvement of 29.33%, 27.03%, and 23.07% on three real-world datasets (PEMSD8, LOSLOOP, and SZAREA) compared to advanced baseline methods, and various ablation experiments validate the effectiveness of each stage and module.

Extended Multi-Component Gated Recurrent Graph Convolutional Network for Traffic Flow Prediction

A Spatial-Temporal Gated Hypergraph Convolution Network for Traffic Prediction

Attention Based Graph Covolution Networks for Intelligent Traffic Flow Analysis

Augmented Multi-Component Recurrent Graph Convolutional Network for Traffic Flow Forecasting

Attention Mechanism Based on Temporal Graph Convolutional Neural Network for Traffic Flow Prediction

A deep learning traffic flow prediction framework based on multi-channel graph convolution

Temporal Multi-Graph Convolutional Network for Traffic Flow Prediction

Principal graph embedding convolutional recurrent network for traffic flow prediction

Attention-Based Gated Recurrent Graph Convolutional Network for Short-Term Traffic Flow Forecasting

Traffic flow prediction based on graph convolutional networks with a parallel attention network and stacked gate recurrent units

Tuning the rigging before sailing off into the stormy sea of stem cell transplants for patients with myelodysplastic syndromes.

Forecasting traffic flows in irregular regions with multi-graph convolutional network and gated recurrent unit

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

Bi-GRCN: A Spatio-Temporal Traffic Flow Prediction Model Based on Graph Neural Network

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

MSA-GCN: Multistage Spatio-Temporal Aggregation Graph Convolutional Networks for Traffic Flow Prediction

Traffic flow prediction based on temporal multigraph convolutional neural network

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

MD-GCN: A Multi-Scale Temporal Dual Graph Convolution Network for Traffic Flow Prediction

GCN-GAN: Integrating Graph Convolutional Network and Generative Adversarial Network for Traffic Flow Prediction

STGMN: A gated multi-graph convolutional network framework for traffic flow prediction