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

基于多段融合张量图卷积网络的增强交通流预测

Enhanced Traffic Flow Prediction with Multi-Segment Fusion Tensor Graph Convolutional Networks

Chongqing University of Posts and Telecommunications(重庆邮电大学);Southwest University(西南大学)

Wei Zhang, Peng Tang

AI总结:

本文提出多段融合张量图卷积网络MS-FTGCN,通过张量M-product统一时空建模,并结合小时、日、周组件与注意力融合,在两个数据集上实现了优于现有模型的交通流预测性能。

摘要:

准确的交通流预测可以辅助交通管理、路线规划和拥堵缓解,对于提升智能交通系统(ITS)的效率和可靠性具有重要意义。然而,现有交通流预测模型在捕捉交通网络内复杂的时空依赖关系方面存在局限性。为解决这一问题,本研究提出了一种用于交通流预测的多段融合张量图卷积网络(MS-FTGCN),其思路包括以下三个方面:a) 构建基于张量M-product的统一时空图卷积框架,以同时捕捉时空模式;b) 引入小时、日和周三个组件,分别对交通流的多时间特性进行建模;c) 通过注意力机制融合三个组件的输出,以获得最终的交通流预测结果。在两个交通流数据集上进行的实验结果表明,所提出的MS-FTGCN优于最先进的模型。

英文摘要:

Accurate traffic Flow Prediction can assist in traffic management, route planning, and congestion mitigation, which holds significant importance in enhancing the efficiency and reliability of intelligent transportation systems (ITS). However, existing traffic flow prediction models suffer from limitations in capturing the complex spatial-temporal dependencies within traffic networks. In order to address this issue, this study proposes a multi-segment fusion tensor graph convolutional network (MS-FTGCN) for traffic flow prediction with the following three-fold ideas: a) building a unified spatial-temporal graph convolutional framework based on Tensor M-product, which capture the spatial-temporal patterns simultaneously; b) incorporating hourly, daily, and weekly components to model multi temporal properties of traffic flows, respectively; c) fusing the outputs of the three components by attention mechanism to obtain the final traffic flow prediction results. The results of experiments conducted on two traffic flow datasets demonstrate that the proposed MS-FTGCN outperforms the state-of-the-art models.

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

↑