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

STLGRU:用于交通流预测的时空轻量级图GRU

STLGRU: Spatio-Temporal Lightweight Graph GRU for Traffic Flow Prediction

Sungkyunkwan University(成均馆大学);University of California, Riverside(加州大学河滨分校);Cornell University(康奈尔大学)

Kishor Kumar Bhaumik, Fahim Faisal Niloy, Saif Mahmud, Simon Woo

AI总结:

为解决交通流预测中模型内存消耗大且难以同时捕捉时空依赖的问题,提出轻量级模型STLGRU,通过记忆增强注意力与门控机制同步学习时空关系,在减少参数与内存占用的同时实现了最先进的预测性能和计算效率。

摘要:

可靠的交通流预测需要对交通数据进行高效建模。事实上,动态交通网络中会出现不同的相关性与影响,使得建模成为一项复杂的任务。现有文献已提出许多不同的方法来捕捉交通网络复杂的潜在时空关系。然而,鉴于交通数据的异构性,持续捕捉空间与时间依赖性仍是一个重大挑战。同时,随着越来越多复杂方法的提出,模型正变得日益消耗内存,从而不适合低功耗设备。为此,我们提出了时空轻量级图GRU(Spatio-Temporal Lightweight Graph GRU),即STLGRU,一种用于准确预测交通流的新型交通预测模型。具体而言,我们提出的STLGRU能够利用记忆增强注意力与门控机制,以持续同步的方式有效捕捉交通网络的动态局部和全局时空关系。此外,我们证明了不采用独立的时空组件,我们的记忆模块与门控单元能够以更低的内存占用和更少的参数成功学习时空依赖性。在三个真实世界公共交通数据集上的大量实验结果表明,我们的方法不仅能达到最先进的性能,还展现出极具竞争力的计算效率。我们的代码可在 https://github.com/Kishor-Bhaumik/STLGRU 获取。

英文摘要:

Reliable forecasting of traffic flow requires efficient modeling of traffic data. Indeed, different correlations and influences arise in a dynamic traffic network, making modeling a complicated task. Existing literature has proposed many different methods to capture traffic networks' complex underlying spatial-temporal relations. However, given the heterogeneity of traffic data, consistently capturing both spatial and temporal dependencies presents a significant challenge. Also, as more and more sophisticated methods are being proposed, models are increasingly becoming memory-heavy and, thus, unsuitable for low-powered devices. To this end, we propose Spatio-Temporal Lightweight Graph GRU, namely STLGRU, a novel traffic forecasting model for predicting traffic flow accurately. Specifically, our proposed STLGRU can effectively capture dynamic local and global spatial-temporal relations of traffic networks using memory-augmented attention and gating mechanisms in a continuously synchronized manner. Moreover, instead of employing separate temporal and spatial components, we show that our memory module and gated unit can successfully learn the spatial-temporal dependencies with reduced memory usage and fewer parameters. Extensive experimental results on three real-world public traffic datasets demonstrate that our method can not only achieve state-of-the-art performance but also exhibit competitive computational efficiency. Our code is available at https://github.com/Kishor-Bhaumik/STLGRU

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

↑