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.

Traffic Flow Prediction Via Spatial-Temporal Geometric Graph Convolutional Network

A Spatial-Temporal Gated Hypergraph Convolution Network for Traffic Prediction

Spatial–Temporal Complex Graph Convolution Network for Traffic Flow Prediction

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

Predicting Traffic Flow Using Dynamic Spatial-Temporal Graph Convolution Networks

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

Traffic Flow Prediction via Spatial Temporal Graph Neural Network

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

Temporal Multi-Graph Convolutional Network for Traffic Flow Prediction

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

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

Dual Dynamic Spatial-Temporal Graph Convolution Network for Traffic Prediction

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

Spatio-Temporal Graph-TCN Neural Network for Traffic Flow Prediction

Spatio-Temporal Traffic Flow Prediction Based on Coordinated Attention

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

Traffic flow prediction based on temporal multigraph convolutional neural network

FD-TGCN: Fast and dynamic temporal graph convolution network for traffic flow prediction

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

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

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