arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2310.08138cs.LGcs.AI

多尺度时空循环网络用于交通流预测

Multi-Scale Spatial-Temporal Recurrent Networks for Traffic Flow Prediction

Soochow University(苏州大学);University of North Carolina at Greensboro(北卡罗来纳大学格林斯伯勒分校);The Hong Kong Polytechnic University(香港理工大学)

Haiyang Liu, Chunjiang Zhu, Detian Zhang, Qing Li

AI总结:

本文提出多尺度时空循环网络MSSTRN,结合单步与多步门控循环单元及时空同步注意力机制,在四个真实数据集上以显著优势超越二十种基线方法,实现高精度交通流预测。

摘要:

交通流预测是智能交通系统中最基本的任务之一。复杂且动态的时空依赖性使得交通流预测颇具挑战性。尽管现有的时空图神经网络表现突出,但它们常常面临以下挑战:(1)忽略固定图限制了模型的预测性能,(2)未能同时充分捕获复杂的时空依赖性,以及(3)缺乏对不同时间长度下时空信息的关注。在本文中,我们提出了一种用于交通流预测的多尺度时空循环网络,即MSSTRN,它由两种不同的循环神经网络组成:单步门控循环单元和多步门控循环单元,以在不同时间窗口下充分捕获交通数据中复杂的时空信息。此外,我们提出了一种时空同步注意力机制,该机制将自适应位置图卷积集成到自注意力机制中,以实现时空依赖性的同步捕获。我们在四个真实交通数据集上进行了大量实验,结果表明,与所有二十种基线方法相比,我们的模型以显著的差距实现了最佳预测精度。

英文摘要:

Traffic flow prediction is one of the most fundamental tasks of intelligent transportation systems. The complex and dynamic spatial-temporal dependencies make the traffic flow prediction quite challenging. Although existing spatial-temporal graph neural networks hold prominent, they often encounter challenges such as (1) ignoring the fixed graph that limits the predictive performance of the model, (2) insufficiently capturing complex spatial-temporal dependencies simultaneously, and (3) lacking attention to spatial-temporal information at different time lengths. In this paper, we propose a Multi-Scale Spatial-Temporal Recurrent Network for traffic flow prediction, namely MSSTRN, which consists of two different recurrent neural networks: the single-step gate recurrent unit and the multi-step gate recurrent unit to fully capture the complex spatial-temporal information in the traffic data under different time windows. Moreover, we propose a spatial-temporal synchronous attention mechanism that integrates adaptive position graph convolutions into the self-attention mechanism to achieve synchronous capture of spatial-temporal dependencies. We conducted extensive experiments on four real traffic datasets and demonstrated that our model achieves the best prediction accuracy with non-trivial margins compared to all the twenty baseline methods.

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

↑