arXiv:2412.01419cs.LGcs.AI2024-12被引 2

通过异步客流追踪提升地铁短时起讫点流量预测精度

CSP-AIT-Net: A contrastive learning-enhanced spatiotemporal graph attention framework for short-term metro OD flow prediction with asynchronous inflow tracking

  • 先预测出站流,再用时空图注意力分解起讫点流量
  • 引入遮蔽机制与客流守恒约束,提升计算效率与建模准确性
  • 结合对比学习挖掘站点土地利用语义,适合城市交通优化研究者

精准的起讫点(OD)客流预测对提升地铁系统效率、优化调度及改善乘客体验至关重要。然而,现有模型常无法有效捕捉OD客流的异步出发特征,且低估进站与出站数据的价值,限制了预测精度。为此,我们提出CSP-AIT-Net,一种融合异步进站追踪与高级站点语义表示的新型时空图注意力框架。该框架重构了OD流量预测范式:先预测出流量,再通过时空图注意力机制分解为各OD组合。为提升计算效率,引入遮蔽机制,并构建融合进站与OD流量、满足守恒约束的异步客流图。此外,采用对比学习提取地铁站点高维土地利用语义,丰富对乘客出行模式的上下文理解。在上海地铁系统的验证表明,该方法在短时OD流量预测上优于当前先进方法,有助于提升地铁运营效率、调度精度与整体系统安全性。

原文摘要 · Abstract (English)

Accurate origin-destination (OD) passenger flow prediction is crucial for enhancing metro system efficiency, optimizing scheduling, and improving passenger experiences. However, current models often fail to effectively capture the asynchronous departure characteristics of OD flows and underutilize the inflow and outflow data, which limits their prediction accuracy. To address these issues, we propose CSP-AIT-Net, a novel spatiotemporal graph attention framework designed to enhance OD flow prediction by incorporating asynchronous inflow tracking and advanced station semantics representation. Our framework restructures the OD flow prediction paradigm by first predicting outflows and then decomposing OD flows using a spatiotemporal graph attention mechanism. To enhance computational efficiency, we introduce a masking mechanism and propose asynchronous passenger flow graphs that integrate inflow and OD flow with conservation constraints. Furthermore, we employ contrastive learning to extract high-dimensional land use semantics of metro stations, enriching the contextual understanding of passenger mobility patterns. Validation of the Shanghai metro system demonstrates improvement in short-term OD flow prediction accuracy over state-of-the-art methods. This work contributes to enhancing metro operational efficiency, scheduling precision, and overall system safety.

地铁流量预测时空图神经网络对比学习异步建模

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。