提出新模型捕捉交通流生成与演化,参数少性能强。
Rethinking Traffic Flow Forecasting: From Transition to Generatation
- 分路建模生成与转移过程,分别用相似性模块和注意力机制。
- 在三个数据集上长短期预测均超越基线,参数量仅为18%。
- 适合关注效率与精度平衡的交通预测研究者。
交通流预测在智能交通系统中至关重要,广泛用于交通管理和城市规划。现有方法主要关注流量转移建模,忽视了节点级流量生成过程,表现为:(i) 基于马尔可夫假设,忽略节点流量生成的多周期性;(ii) 使用相同结构编码生成与转移,未区分两者差异。为此,本文提出有效多分支相似性变压器EMBSFormer。数据分析表明,影响交通流的因素包括节点级流量生成和图级流量转移,分别反映多周期性与节点交互模式。为捕捉生成模式,提出支持多分支编码的相似性分析模块,动态扩展显著周期;为建模转移,采用时空自注意力保持全局节点交互,并分别使用GNN和时间卷积建模局部交互。在三个真实世界数据集上评估长期与短期预测任务,实验结果表明EMBSFormer在两项任务上均优于基线。此外,相较于基于流量转移建模的模型(如GMAN,513k参数),其变体仅需93K参数(18%),达到相当性能。
原文摘要 · Abstract (English)
Traffic flow prediction plays an important role in Intelligent Transportation Systems in traffic management and urban planning. There have been extensive successful works in this area. However, these approaches focus only on modelling the flow transition and ignore the flow generation process, which manifests itself in two ways: (i) The models are based on Markovian assumptions, ignoring the multi-periodicity of the flow generation in nodes. (ii) The same structure is designed to encode both the transition and generation processes, ignoring the differences between them. To address these problems, we propose an Effective Multi-Branch Similarity Transformer for Traffic Flow Prediction, namely EMBSFormer. Through data analysis, we find that the factors affecting traffic flow include node-level traffic generation and graph-level traffic transition, which describe the multi-periodicity and interaction pattern of nodes, respectively. Specifically, to capture traffic generation patterns, we propose a similarity analysis module that supports multi-branch encoding to dynamically expand significant cycles. For traffic transition, we employ a temporal and spatial self-attention mechanism to maintain global node interactions, and use GNN and time conv to model local node interactions, respectively. Model performance is evaluated on three real-world datasets on both long-term and short-term prediction tasks. Experimental results show that EMBSFormer outperforms baselines on both tasks. Moreover, compared to models based on flow transition modelling (e.g. GMAN, 513k), the variant of EMBSFormer(93K) only uses 18\% of the parameters, achieving the same performance.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。