Abstract:Accurate traffic flow prediction in road networks is essential for intelligent transportation systems (ITS). Since traffic data are collected from the road network with spatial topological and time series sequences, the traffic flow prediction is regarded as a spatial–temporal prediction task. With the powerful ability to model the non-Euclidean data, the graph convolutional network (GCN)-based models have become the mainstream framework for traffic forecasting. However, existing GCN-based models either use the manually predefined graph structure to capture the spatial features, ignoring the heterogeneity of road networks, or simply perform 1-D convolution with fixed kernel to capture the temporal dependencies of traffic data, resulting in insufficient long-term temporal feature extraction. To solve those issues, a spatial–temporal correlation constrained dynamic graph convolutional network (STC-DGCN) is proposed for traffic flow forecasting. In STC-DGCN, a spatial–temporal embedding encoder module (STEM) is first constructed to encode the dynamic spatial relationships for road networks at different time steps. Then, a temporal feature encoder module with heterogeneous time series correlation modeling (TFE-HCM) and a spatial feature encoder module with dynamic multi-graph modeling (SFE-DCM) are designed to generate dynamic graph structures for effectively capturing the dynamic spatial and temporal correlations. Finally, a spatial–temporal feature fusion module based on a gating fusion mechanism (STM-GM) is proposed to effectively learn and leverage the inherent spatial–temporal relationships for traffic flow forecasting. Experimental results from three real-world traffic flow datasets demonstrate the superior performance of the proposed STC-DGCN compared with state-of-the-art traffic flow forecasting models.

Spatial-temporal graph neural network based on gated convolution and topological attention for traffic flow prediction

A Spatial-Temporal Gated Hypergraph Convolution Network for Traffic Prediction

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

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

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

Spatio-temporal causal graph attention network for traffic flow prediction in intelligent transportation systems

Adaptive Spatio-temporal Graph Neural Network for traffic forecasting

Spatial–Temporal-Correlation-Constrained Dynamic Graph Convolutional Network for Traffic Flow Forecasting

Spatial–Temporal Complex Graph Convolution Network for Traffic Flow Prediction

Spatial-temporal graph convolutional networks for traffic flow prediction considering multiple traffic parameters

Graph Attention Convolutional Network: Spatiotemporal Modeling for Urban Traffic Prediction

An Attention and Wavelet Based Spatial-Temporal Graph Neural Network for Traffic Flow and Speed Prediction

Dynamic spatial aware graph transformer for spatiotemporal traffic flow forecasting

Dual Dynamic Spatial-Temporal Graph Convolution Network for Traffic Prediction

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

Spatio-Temporal Graph Convolutional Networks: A Deep Learning Framework for Traffic Forecasting

Spatial-Temporal Interactive Graph Neural Network for Traffic Forecasting.

Spatial-Temporal Adaptive Graph Convolution with Attention Network for Traffic Forecasting

Make More Connections: Urban Traffic Flow Forecasting with Spatiotemporal Adaptive Gated Graph Convolution Network

Predicting Traffic Flow Using Dynamic Spatial-Temporal Graph Convolution Networks

Spatio-Temporal Traffic Flow Prediction Based on Coordinated Attention