论文标题
A3T-GCN:注意交通预测的时间图卷积网络
A3T-GCN: Attention Temporal Graph Convolutional Network for Traffic Forecasting
论文作者
论文摘要
准确的实时流量预测是针对实施智能运输系统的核心技术问题。但是,考虑到交通流量之间复杂的空间和时间依赖性,这仍然很具有挑战性。在空间维度中,由于道路网络的连通性,链接道路之间的交通流量密切相关。就时间因素而言,尽管相邻时间点之间存在着一种趋势,但遥远的过去点的重要性不一定小于最近点,因为交通流也受到外部因素的影响。在这项研究中,提出了一个注意力图卷积网络(A3T-GCN)的流量预测方法,以同时捕获全局的时间动力学和空间相关性。 A3T-GCN模型通过使用封闭式复发单元来了解时间序列的短期趋势,并通过图形卷积网络根据道路网络的拓扑来了解空间依赖性。此外,引入了注意机制,以调整不同时间点的重要性并组装全球时间信息以提高预测准确性。实际数据集中的实验结果证明了提出的A3T-GCN的有效性和鲁棒性。可以在https://github.com/lehaifeng/t-gcn/a3t上访问源代码。
Accurate real-time traffic forecasting is a core technological problem against the implementation of the intelligent transportation system. However, it remains challenging considering the complex spatial and temporal dependencies among traffic flows. In the spatial dimension, due to the connectivity of the road network, the traffic flows between linked roads are closely related. In terms of the temporal factor, although there exists a tendency among adjacent time points in general, the importance of distant past points is not necessarily smaller than that of recent past points since traffic flows are also affected by external factors. In this study, an attention temporal graph convolutional network (A3T-GCN) traffic forecasting method was proposed to simultaneously capture global temporal dynamics and spatial correlations. The A3T-GCN model learns the short-time trend in time series by using the gated recurrent units and learns the spatial dependence based on the topology of the road network through the graph convolutional network. Moreover, the attention mechanism was introduced to adjust the importance of different time points and assemble global temporal information to improve prediction accuracy. Experimental results in real-world datasets demonstrate the effectiveness and robustness of proposed A3T-GCN. The source code can be visited at https://github.com/lehaifeng/T-GCN/A3T.