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

Bin Wang,Fanghong Gao,Le Tong,Qian Zhang,Sulei Zhu
DOI: https://doi.org/10.1108/dta-09-2022-0378
IF: 1.713
2023-05-05
Data Technologies and Applications
Abstract:Purpose Traffic flow prediction has always been a top priority of intelligent transportation systems. There are many mature methods for short-term traffic flow prediction. However, the existing methods are often insufficient in capturing long-term spatial-temporal dependencies. To predict long-term dependencies more accurately, in this paper, a new and more effective traffic flow prediction model is proposed. Design/methodology/approach This paper proposes a new and more effective traffic flow prediction model, named channel attention-based spatial-temporal graph neural networks. A graph convolutional network is used to extract local spatial-temporal correlations, a channel attention mechanism is used to enhance the influence of nearby spatial-temporal dependencies on decision-making and a transformer mechanism is used to capture long-term dependencies. Findings The proposed model is applied to two common highway datasets: METR-LA collected in Los Angeles and PEMS-BAY collected in the California Bay Area. This model outperforms the other five in terms of performance on three performance metrics a popular model. Originality/value (1) Based on the spatial-temporal synchronization graph convolution module, a spatial-temporal channel attention module is designed to increase the influence of proximity dependence on decision-making by enhancing or suppressing different channels. (2) To better capture long-term dependencies, the transformer module is introduced.
computer science, information systems,information science & library science
What problem does this paper attempt to address?