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

用于 COVID-19 期间城市轨道交通系统短期客流预测的 ST-former

ST-former for short-term passenger flow prediction during COVID-19 in urban rail transit system

Shuxin Zhang, Jinlei Zhang, Lixing Yang, Chengcheng Wang, Ziyou Gao

AI总结:

针对疫情期间城市轨道交通客流预测,本文提出基于编码器—解码器的 ST-former,利用 CPSA、AMGCN 及多源数据融合建模复杂时空依赖与疫情影响,实验证明其优于十一种基线方法。

摘要:

准确的城市轨道交通客流预测对于提升智能交通系统的性能至关重要,尤其是在疫情期间。如何对客流复杂的时空依赖关系进行动态建模,是在疫情期间实现准确客流预测的主要问题。为解决这一问题,本文在编码器—解码器框架下,专门针对 COVID-19 提出了一种全新的、基于 transformer 的架构,称为 STformer。具体而言,我们开发了一种名为因果卷积 ProbSparse 自注意力(Causal-Convolution ProbSparse Self-Attention,CPSA)的改进自注意力机制,以较低的计算成本对客流的多重时间依赖关系进行建模。为捕捉复杂且动态的空间依赖关系,我们通过以自适应方式利用多个图,提出了一种新颖的自适应多图卷积网络(Adaptive Multi-Graph Convolution Network,AMGCN)。此外,多源数据融合模块融合客流数据、COVID-19 确诊病例数据以及相关社交媒体数据,以研究 COVID-19 对客流的影响。在真实客流数据集上的实验表明,ST-former 优于其他十一种最先进方法。本文还开展了多项消融研究,以验证模型结构的有效性和可靠性。研究结果可为 URT 系统的运营提供关键洞见。

英文摘要:

Accurate passenger flow prediction of urban rail transit is essential for improving the performance of intelligent transportation systems, especially during the epidemic. How to dynamically model the complex spatiotemporal dependencies of passenger flow is the main issue in achieving accurate passenger flow prediction during the epidemic. To solve this issue, this paper proposes a brand-new transformer-based architecture called STformer under the encoder-decoder framework specifically for COVID-19. Concretely, we develop a modified self-attention mechanism named Causal-Convolution ProbSparse Self-Attention (CPSA) to model the multiple temporal dependencies of passenger flow with low computational costs. To capture the complex and dynamic spatial dependencies, we introduce a novel Adaptive Multi-Graph Convolution Network (AMGCN) by leveraging multiple graphs in a self-adaptive manner. Additionally, the Multi-source Data Fusion block fuses the passenger flow data, COVID-19 confirmed case data, and the relevant social media data to study the impact of COVID-19 to passenger flow. Experiments on real-world passenger flow datasets demonstrate the superiority of ST-former over the other eleven state-of-the-art methods. Several ablation studies are carried out to verify the effectiveness and reliability of our model structure. Results can provide critical insights for the operation of URT systems.

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

↑