期刊
TRANSPORTMETRICA B-TRANSPORT DYNAMICS
卷 9, 期 1, 页码 153-171出版社
TAYLOR & FRANCIS LTD
DOI: 10.1080/21680566.2020.1822765
关键词
Traffic prediction; deep learning; graph attention network; temporal convolutional network
资金
- National Key Research and Development Program of China [2018YFB1601600]
Traffic speed forecasting is crucial in intelligent transportation systems, and improving prediction accuracy and achieving timely performance require capturing spatio-temporal dependencies and creating parallel model architecture. The proposed GATCN framework utilizes graph attention network and temporal convolution operation to predict traffic speed, outperforming other state-of-the-art baselines in experiments.
Traffic speed forecasting plays an increasingly essential role in successful intelligent transportation systems. However, this still remains a challenging task when the accuracy requirement is demanding. To improve the prediction accuracy and achieve a timely performance, the capture of the intrinsically spatio-temporal dependencies and the creation of a parallel model architecture are required. Accordingly, we propose a novel end-to-end deep learning framework named Graph Attention Temporal Convolutional Network (GATCN). The proposed model employs the graph attention network to mine the complex spatial correlations within the traffic network and temporal convolution operation to capture temporal dependencies. In addition, the multi-head self-attention mechanism is incorporated into the model to extract the spatio-temporal coupling effects. Experiments show that the proposed model consistently outperforms other state-of-the-art baselines for various prediction intervals on two real-world datasets. Moreover, we reveal that the proposed model can effectively distinguish the sophisticated traffic patterns of ramps on expressways by analyzing the graph attention heatmap.
作者
我是这篇论文的作者
点击您的名字以认领此论文并将其添加到您的个人资料中。
推荐
暂无数据