arrow
Return

A traffic flow forecasting method based on hybrid spatial-temporal gated convolution

delete2024-09-02
delete0
PRE
AI
张影 (Ying Zhang)
S
Songhao Yang
H
Hongchao Wang
Y
Yongqiang Cheng *
J
Jinyu Wang
L
Liping Cao
DOI:10.1007/s13042-024-02364-4delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
Influenced by the urban road network, traffic flow has complex temporal and spatial correlation characteristics. Traffic flow forecasting is an important problem in the intelligent transportation system, which is related to the safety and stability of the transportation system. At present, many researchers ignore the research need for traffic flow forecasting beyond one hour. To address the issue of long-term traffic flow prediction, this paper proposes a traffic flow prediction model (HSTGCNN) based on a hybrid spatial-temporal gated convolution. Spatial-temporal attention mechanism and Gated convolution are the main components of HSTGCNN. The spatial-temporal attention mechanism can effectively obtain the spatial-temporal features of traffic flow, and gated convolution plays an important role in extracting longer-term features. The usage of dilated causal convolution effectively improves the long-term prediction ability of the model. HSTGCNN predicts the traffic conditions of 1 h, 1.5 h, and 2 h on two general traffic flow datasets. Experimental results show that the prediction accuracy of HSTGCNN is generally better than that of Temporal Graph Convolutional Network (T-GCN), Graph WaveNet, and other baselines.
Keywords:
Traffic flow forecasting
Spatial-temporal fusion
Dilated causal convolution
Attention mechanism
Gated convolution

Journal

International Journal of Machine Learning and Cybernetics cover
International Journal of Machine Learning and Cybernetics
IF:
2.7
Papers:
3.1K
Citations:
5.6K

Organization

N
north china electric power university
Scholars:
2.5W
Papers: 1.7W
Citations: 16