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

用于城市轨道交通系统节假日短期客流预测的时空注意力融合网络

Spatial-Temporal Attention Fusion Network for short-term passenger flow prediction on holidays in urban rail transit systems

Shuxin Zhang, Jinlei Zhang, Lixing Yang, Jiateng Yin, Ziyou Gao

AI总结:

针对城市轨道交通节假日客流突发且不规则、预测困难的问题,提出融合多图注意力、卷积注意力及社交媒体数据的 STAFN,并在两个大规模 AFC 数据集上验证其优于传统基准。

摘要:

城市轨道交通系统的短期客流预测对交通运营与管理具有重要意义。新兴的基于深度学习的模型为提升预测精度提供了有效方法。然而,现有模型大多主要预测普通工作日或周末的客流,仅有少数研究关注节假日客流预测;由于其突发性和不规则性,这对交通管理而言是一项极具挑战性的任务。为此,我们提出了一种名为 Spatial Temporal Attention Fusion Network(时空注意力融合网络,STAFN)的深度学习模型,用于节假日短期客流预测,该模型由新颖的 Multi-Graph Attention Network(多图注意力网络)、Conv-Attention Block(卷积注意力块)和 Feature Fusion Block(特征融合块)组成。多图注意力网络用于动态提取客流复杂的空间依赖关系,卷积注意力块用于从全局和局部视角提取客流的时间依赖关系。此外,除历史客流数据外,已被证明能够有效反映事件情形下客流演变趋势的社交媒体数据,也被融合到 STAFN 的特征融合块中。STAFN 在中国两个大规模城市轨道交通 AFC 数据集的元旦假期场景上进行了测试,并将模型的预测性能与若干传统预测模型进行了比较。结果表明,该模型在各基准方法中具有更好的鲁棒性和优势,能够为节假日短期客流预测的实际应用提供强有力的支持。

英文摘要:

The short term passenger flow prediction of the urban rail transit system is of great significance for traffic operation and management. The emerging deep learning-based models provide effective methods to improve prediction accuracy. However, most of the existing models mainly predict the passenger flow on general weekdays or weekends. There are only few studies focusing on predicting the passenger flow on holidays, which is a significantly challenging task for traffic management because of its suddenness and irregularity. To this end, we propose a deep learning-based model named Spatial Temporal Attention Fusion Network comprising a novel Multi-Graph Attention Network, a Conv-Attention Block, and Feature Fusion Block for short-term passenger flow prediction on holidays. The multi-graph attention network is applied to extract the complex spatial dependencies of passenger flow dynamically and the conv-attention block is applied to extract the temporal dependencies of passenger flow from global and local perspectives. Moreover, in addition to the historical passenger flow data, the social media data, which has been proven that they can effectively reflect the evolution trend of passenger flow under events, are also fused into the feature fusion block of STAFN. The STAFN is tested on two large-scale urban rail transit AFC datasets from China on the New Year holiday, and the prediction performance of the model are compared with that of several conventional prediction models. Results demonstrate its better robustness and advantages among benchmark methods, which can provide overwhelming support for practical applications of short term passenger flow prediction on holidays.

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

↑