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.

Regularized Spatial–Temporal Graph Convolutional Networks for Metro Passenger Flow Prediction

A Spatial-Temporal Gated Hypergraph Convolution Network for Traffic Prediction

Hybrid Approach Combining Modified Gravity Model and Deep Learning for Short-Term Forecasting of Metro Transit Passenger Flows

Predicting Station-Level Short-Term Passenger Flow in a Citywide Metro Network Using Spatiotemporal Graph Convolutional Neural Networks

MGSTCN: A Multi-Graph Spatio-Temporal Convolutional Network for Metro Passenger Flow Prediction

Metro Station Functional Clustering and Dual-View Recurrent Graph Convolutional Network for Metro Passenger Flow Prediction

Multi-Spatio-Temporal Convolutional Neural Network for Short-Term Metro Passenger Flow Prediction

SPATIOTEMPORAL GRAPH CONVOLUTIONAL NEURAL NETWORKS FOR METRO FLOW PREDICTION

MTMGNN: Multi-time Multi-Graph Neural Network for Metro Passenger Flow Prediction

A dynamic multigraph and multidimensional attention neural network model for metro passenger flow prediction

Dynamic Relational Graph Convolutional Network for Metro Passenger Flow Forecasting

Metro Passenger Flow Prediction Via Dynamic Hypergraph Convolution Networks

Passenger Flow Prediction for Public Transportation Stations Based on Spatio-Temporal Graph Convolutional Network with Periodic Components

Multi-STGCnet: A Graph Convolution Based Spatial-Temporal Framework for Subway Passenger Flow Forecasting

A spatiotemporal graph convolution gated recurrent unit model for short-term passenger flow estimation

Parallel Multi-Graph Convolution Network For Metro Passenger Volume Prediction

Enhanced Self-node Weights Based Graph Convolutional Networks for Passenger Flow Prediction

Attention-based Graph Neural Network Enabled Method to Predict Short-term Metro Passenger Flow

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

A Comprehensive Spatio-Temporal Model for Subway Passenger Flow Prediction

Learning spatial-temporal dynamics and interactivity for short-term passenger flow prediction in urban rail transit