Abstract:It is crucial for both traffic management organisations and individual commuters to be able to forecast traffic flows accurately. Graph neural networks made great strides in this field owing to their exceptional capacity to capture spatial correlations. However, existing approaches predominantly focus on local geographic correlations, ignoring cross-region interdependencies in a global context, which is insufficient to extract comprehensive semantic relationships, thereby limiting prediction accuracy. Additionally, most GCN-based models rely on pre-defined graphs and unchanging adjacency matrices to reflect the spatial relationships among node features, neglecting the dynamics of spatio-temporal features and leading to challenges in capturing the complexity and dynamic spatial dependencies in traffic data. To tackle these issues, this paper puts forward a fresh approach: a new self-supervised dynamic spatio-temporal graph convolutional network (SDSC) for traffic flow forecasting. The proposed SDSC model is a hierarchically structured graph–neural architecture that is intended to augment the representation of dynamic traffic patterns through a self-supervised learning paradigm. Specifically, a dynamic graph is created using a combination of temporal, spatial, and traffic data; then, a regional graph is constructed based on geographic correlation using clustering to capture cross-regional interdependencies. In the feature learning module, spatio-temporal correlations in traffic data are subjected to recursive extraction using dynamic graph convolution facilitated by Recurrent Neural Networks (RNNs). Furthermore, self-supervised learning is embedded within the network training process as an auxiliary task, with the objective of enhancing the prediction task by optimising the mutual information of the learned features across the two graph networks. The superior performance of the proposed SDSC model in comparison with SOTA approaches was confirmed by comprehensive experiments conducted on real road datasets, PeMSD4 and PeMSD8. These findings validate the efficacy of dynamic graph modelling and self-supervision tasks in improving the precision of traffic flow prediction.

Road traffic state prediction based on a graph embedding recurrent neural network under the SCATS.

Principal graph embedding convolutional recurrent network for traffic flow prediction

Road Traffic Network State Prediction Based on a Generative Adversarial Network

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

Gated Fusion Adaptive Graph Neural Network for Urban Road Traffic Flow Prediction

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

Traffic Flow Prediction via Spatial Temporal Graph Neural Network

Adaptive spatial-temporal graph attention networks for traffic flow forecasting

Uncertainty-Aware Probabilistic Graph Neural Networks for Road-Level Traffic Accident Prediction

Spatio-Temporal Graph Neural Network for Traffic Prediction Based on Adaptive Neighborhood Selection

A Spatiotemporal Graph Neural Network with Graph Adaptive and Attention Mechanisms for Traffic Flow Prediction

Network-level Short-term Traffic State Prediction Incorporating Critical Nodes: A Knowledge-based Deep Fusion Approach

Dynamic Graph Convolutional Recurrent Network for Traffic Prediction: Benchmark and Solution

Uncertainty-aware probabilistic graph neural networks for road-level traffic crash prediction

Global-Aware Enhanced Spatial-Temporal Graph Recurrent Networks: A New Framework For Traffic Flow Prediction

Optimized Graph Convolution Recurrent Neural Network for Traffic Prediction

Adaptive Hybrid Spatial-Temporal Graph Neural Network for Cellular Traffic Prediction

Hierarchical Dynamic Spatio-Temporal Graph Convolutional Networks with Self-Supervised Learning for Traffic Flow Forecasting

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

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

Multi-graph Spatio-temporal Graph Convolutional Network for Traffic Flow Prediction