Short-term traffic speed forecasting based on graph attention temporal convolutional networks
Short-term traffic speed forecasting based on graph attention temporal convolutional networks
复制标题
DOI:
10.1016/j.neucom.2020.06.001
复制
发表时间:
2020-10-14
期刊:
影响因子:
6
通讯作者:
Yuan, Wei
中科院分区:
文献类型:
--
作者:
Guo, Ge;Yuan, Wei
Accurate and timely traffic forecasting is significant for intelligent transportation management. However, existing approaches model the temporal and spatial features of traffic flow inadequately. To address these limitations, a novel deep learning traffic forecasting framework based on graph attention network (GAT) and temporal convolutional network (TCN) is presented in this paper, termed as graph attention temporal convolutional networks (GATCN). More specifically, GATCN deal with the spatial features by GAT, and the temporal features by TCN. The layer fused by GAT and TCN enables the proposed model to learn the spatio-temporal characteristics that lie in traffic flow, while considering exogenous factors. In addition, nodes in the graph can capture the information of their neighborhoods by stacking multiple layers. Precision and robustness of the proposed method have been evaluated through testing on the real-world dataset. Results show that the proposed model outperforms other baselines. (C) 2020 Elsevier B.V. All rights reserved.