融合图注意力与双向卷积,实现多交通模式时空联合预测
Multimodal joint prediction of traffic spatial-temporal data with graph sparse attention mechanism and bidirectional temporal convolutional network
- 用稀疏注意力捕捉全局空间特征,自注意力提取局部特征
- 双向卷积增强时序关联性,共享-独有模块区分模态间差异
- 可灵活扩展,适合城市多交通模式协同预测场景
交通流量预测在城市交通管理中至关重要。尽管已有大量针对单一交通模式的研究,但跨模式联合预测仍较有限,且现有方法在时空特征提取上灵活性不足。为此,本文提出一种基于图稀疏注意力机制与双向时间卷积网络的多模态交通时空联合预测方法(GSABT)。首先,通过乘以自注意力权重的多模态图捕获空间局部特征,并利用Top-U稀疏注意力机制获取空间全局特征;其次,采用双向时间卷积网络增强输出与输入数据间的时序相关性,并通过共享-独有模块提取模态间与模态内的时间特征;最后,设计可灵活扩展的多模态联合预测框架。在三个真实数据集上的大量实验表明,该模型持续达到最先进性能。
原文摘要 · Abstract (English)
Traffic flow prediction plays a crucial role in the management and operation of urban transportation systems. While extensive research has been conducted on predictions for individual transportation modes, there is relatively limited research on joint prediction across different transportation modes. Furthermore, existing multimodal traffic joint modeling methods often lack flexibility in spatial-temporal feature extraction. To address these issues, we propose a method called Graph Sparse Attention Mechanism with Bidirectional Temporal Convolutional Network (GSABT) for multimodal traffic spatial-temporal joint prediction. First, we use a multimodal graph multiplied by self-attention weights to capture spatial local features, and then employ the Top-U sparse attention mechanism to obtain spatial global features. Second, we utilize a bidirectional temporal convolutional network to enhance the temporal feature correlation between the output and input data, and extract inter-modal and intra-modal temporal features through the share-unique module. Finally, we have designed a multimodal joint prediction framework that can be flexibly extended to both spatial and temporal dimensions. Extensive experiments conducted on three real datasets indicate that the proposed model consistently achieves state-of-the-art predictive performance.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。