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

Yajun Ge (), Jiannan Wang, Bo Zhang, Fan Peng, Jing Ma, Chenyu Yang, Yue Zhao and Ming Liu ()
Additional contact information
Yajun Ge: Shaanxi Transportation Holding Group Co., Ltd., Xi’an 710000, China
Jiannan Wang: Operation Management Branch of Shaanxi Transportation Holding Group Co., Ltd., Xi’an 710000, China
Bo Zhang: Operation Management Branch of Shaanxi Transportation Holding Group Co., Ltd., Xi’an 710000, China
Fan Peng: Operation Management Branch of Shaanxi Transportation Holding Group Co., Ltd., Xi’an 710000, China
Jing Ma: Shaanxi Expressway Testing & Measuring Co., Ltd., Xi’an 710000, China
Chenyu Yang: School of Economics, Renmin University of China, Beijing 100872, China
Yue Zhao: School of Civil Engineering and Architecture, Xi’an University of Technology, Xi’an 710048, China
Ming Liu: School of Materials Science and Engineering, Xi’an University of Technology, Xi’an 710048, China

Mathematics, 2024, vol. 12, issue 19, 1-18

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.

Keywords: traffic flow forecasting; spatial–temporal correlation; dynamic graph convolutional network; attention mechanism (search for similar items in EconPapers)
JEL-codes: C (search for similar items in EconPapers)
Date: 2024
References: View references in EconPapers View complete reference list from CitEc
Citations:

Downloads: (external link)
https://www.mdpi.com/2227-7390/12/19/3159/pdf (application/pdf)
https://www.mdpi.com/2227-7390/12/19/3159/ (text/html)

Related works:
This item may be available elsewhere in EconPapers: Search for items with the same title.

Export reference: BibTeX RIS (EndNote, ProCite, RefMan) HTML/Text

Persistent link: https://EconPapers.repec.org/RePEc:gam:jmathe:v:12:y:2024:i:19:p:3159-:d:1495112

Access Statistics for this article

Mathematics is currently edited by Ms. Emma He

More articles in Mathematics from MDPI
Bibliographic data for series maintained by MDPI Indexing Manager ().