Abstract:Real-time and precise traffic flow prediction is vital for the efficiency of intelligent transportation systems. Traditional methods often employ graph neural networks (GNNs) with predefined graphs to describe spatial correlations among traffic nodes in urban road networks. However, these pre-defined graphs are limited by existing knowledge and graph generation methodologies, offering an incomplete picture of spatial correlations. While time-varying graphs based on data-driven learning have attempted to address these limitations, they still struggle with adequately capturing the inherent spatial correlations in traffic data. Moreover, most current methods for capturing dynamic temporal correlations rely on a unified calculation scheme using a temporal multi-head self-attention mechanism, which at some level might leads to inaccuracies. In order to overcome these challenges, we have proposed a novel hybrid time-varying graph neural network (HTVGNN) for traffic flow prediction. Firstly, a novel enhanced temporal perception multi-head self-attention mechanism based on time-varying mask enhancement was reported to more accurately model the dynamic temporal dependencies among distinct traffic nodes in the traffic network. Secondly, we have proposed a novel graph learning strategy to concurrently learn both static and dynamic spatial associations between different traffic nodes in road networks. Meanwhile, in order to enhance the learning ability of time-varying graphs, a coupled graph learning mechanism was designed to couple the graphs learned at each time step. Finally, the effectiveness of the proposed method HTVGNN was demonstrated with four real data sets. Simulation results revealed that HTVGNN achieves superior prediction accuracy compared to the state of the art spatio-temporal graph neural network models. Additionally, the ablation experiment verifies that the coupled graph learning mechanism can effectively improve the long-term prediction performance of HTVGNN.

AARGNN: an Attentive Attributed Recurrent Graph Neural Network for Traffic Flow Prediction Considering Multiple Dynamic Factors

A Spatial-Temporal Gated Hypergraph Convolution Network for Traffic Prediction

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

Traffic Flow Prediction via Spatial Temporal Graph Neural Network

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

A novel hybrid time-varying graph neural network for traffic flow forecasting

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

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

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

A Spatio-Temporal Graph Neural Network Approach for Traffic Flow Prediction

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

Road traffic flow prediction based on dynamic spatiotemporal graph attention network

Attention-based Spatial-Temporal Graph Convolutional Recurrent Networks for Traffic Forecasting

IGAGCN: Information geometry and attention-based spatiotemporal graph convolutional networks for traffic flow prediction

Dynamic multi-scale spatial-temporal graph convolutional network for traffic flow prediction

Attention-Based Multiple Graph Convolutional Recurrent Network for Traffic Forecasting

Attention-based spatial–temporal adaptive dual-graph convolutional network for traffic flow forecasting

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

The Prediction of Multistep Traffic Flow Based on AST-GCN-LSTM

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

Channel attention-based spatial-temporal graph neural networks for traffic prediction