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 Message Channel Prediction Based on Graph Convolutional Network

A Spatial-Temporal Gated Hypergraph Convolution Network for Traffic Prediction

Attention Based Graph Covolution Networks for Intelligent Traffic Flow Analysis

A deep learning traffic flow prediction framework based on multi-channel graph convolution

T-GCN: A Temporal Graph ConvolutionalNetwork for Traffic Prediction

T-GCN: A Temporal Graph Convolutional Network for Traffic Prediction

Probabilistic spatio-temporal graph convolutional network for traffic forecasting

A Graph Convolutional Method for Traffic Flow Prediction in Highway Network

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

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

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

LST-GCN: Long Short-Term Memory Embedded Graph Convolution Network for Traffic Flow Forecasting

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

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

Traffic flow prediction based on temporal multigraph convolutional neural network

Dynamic Correlation Adjacency-Matrix-Based Graph Neural Networks for Traffic Flow Prediction

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

Temporal Multi-Graph Convolutional Network for Traffic Flow Prediction

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

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

DTM-GCN: A traffic flow prediction model based on dynamic graph convolutional network