arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2202.08982cs.LG
收藏

PGCN:用于时空交通预测的渐进式图卷积网络

PGCN: Progressive Graph Convolutional Networks for Spatial-Temporal Traffic Forecasting

Korea Advanced Institute of Science and Technology(韩国科学技术院)

Yuyol Shin, Yoonjin Yoon

AI总结:

提出PGCN交通预测框架,通过在训练和测试阶段逐步适应在线输入数据构建渐进式邻接矩阵,结合膨胀因果卷积提取时空特征,在七个真实数据集上取得一致的最先进性能。

摘要:

交通网络中复杂的时空相关性使得交通预测问题具有挑战性。由于交通系统本身具有图结构,许多研究工作都投入到了图神经网络中。最近,为数据构建自适应图在依赖单一静态图结构的模型上展现出了可喜的结果。然而,图的适应过程是在训练阶段应用的,并不能反映测试阶段使用的数据。这种缺点在交通预测中可能会带来问题,因为交通数据经常出现时间序列中的意外变化和不规则性。在本研究中,我们提出了一种名为渐进式图卷积网络(PGCN)的新型交通预测框架。PGCN通过在训练和测试阶段逐步适应在线输入数据来构建一组图。具体而言,我们实现了该模型,通过学习图节点间的趋势相似性来构建渐进式邻接矩阵。然后,将模型与膨胀因果卷积和门控激活单元相结合以提取时间特征。借助残差和跳跃连接,PGCN执行交通预测。当应用于七个具有不同几何特性的真实世界交通数据集时,所提出的模型在所有数据集上均取得了一致的最先进性能。我们得出结论,PGCN逐步适应输入数据的能力使模型能够在不同的研究地点中泛化并保持鲁棒性。

英文摘要:

The complex spatial-temporal correlations in transportation networks make the traffic forecasting problem challenging. Since transportation system inherently possesses graph structures, many research efforts have been put with graph neural networks. Recently, constructing adaptive graphs to the data has shown promising results over the models relying on a single static graph structure. However, the graph adaptations are applied during the training phases and do not reflect the data used during the testing phases. Such shortcomings can be problematic especially in traffic forecasting since the traffic data often suffer from unexpected changes and irregularities in the time series. In this study, we propose a novel traffic forecasting framework called Progressive Graph Convolutional Network (PGCN). PGCN constructs a set of graphs by progressively adapting to online input data during the training and testing phases. Specifically, we implemented the model to construct progressive adjacency matrices by learning trend similarities among graph nodes. Then, the model is combined with the dilated causal convolution and gated activation unit to extract temporal features. With residual and skip connections, PGCN performs the traffic prediction. When applied to seven real-world traffic datasets of diverse geometric nature, the proposed model achieves state-of-the-art performance with consistency in all datasets. We conclude that the ability of PGCN to progressively adapt to input data enables the model to generalize in different study sites with robustness.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑