arXiv:2606.09872cs.LGcs.AI2026-06

用分块方法解决不规则传感器网络的交通预测难题

PatchSTG: Scalable Spatiotemporal Graph Transformers for Traffic Forecasting on Irregular Sensor Networks

  • 将传感器按地理信息分块,构建层次化空间结构
  • 双注意力机制使计算复杂度从二次方降至近线性
  • 适合大规模不规则交通网络,兼顾精度与效率

交通预测是智能交通系统的核心,但在实际场景中因传感器分布不均和建模大规模时空依赖的高计算成本而面临挑战。真实交通网络中传感器分布不均,导致空间结构非均匀,限制了现有图模型和注意力模型的性能与可扩展性。为此,我们提出 PatchSTG,一种基于分块的时空图变压器,用于高效预测不规则传感器网络上的交通流量。核心思想是引入层级空间表示,根据地理信息将传感器划分为平衡且保持局部性的分块。在此结构上,双注意力编码器交替使用块内注意力捕捉局部交互和块间注意力建模全局依赖,使计算复杂度从二次方降至近线性。我们在罗德岛真实交通数据及多个大规模数据集上评估了 PatchSTG。实验结果表明,该模型在多个预测时序下均实现稳定且具有竞争力的性能,同时显著提升计算效率。消融实验证实了空间分块和双注意力机制在捕捉局部与长程交通动态方面的有效性。这些结果表明,基于分块的时空建模为不规则空间设置下的交通预测提供了一个可扩展且高效的框架。

原文摘要 · Abstract (English)

Traffic forecasting is a fundamental component of intelligent transportation systems, yet remains challenging in real-world settings due to irregular sensor distributions and the high computational cost of modeling large-scale spatiotemporal dependencies. In practical traffic networks, sensors are unevenly distributed across regions, leading to non-uniform spatial structures that limit the effectiveness and scalability of existing graph-based and attention-based models. To address these challenges, we propose PatchSTG, a patch-based spatiotemporal graph Transformer designed for efficient forecasting on irregular sensor networks. The key idea is to introduce a hierarchical spatial representation that partitions sensors into balanced, locality-preserving patches based on geographic information. On top of this structure, a dual attention encoder alternates between intra-patch attention for capturing local interactions and inter-patch attention for modeling global dependencies, reducing computational complexity from quadratic to near-linear scaling. We evaluate PatchSTG on real-world traffic data from Rhode Island and additional large-scale datasets. Experimental results demonstrate that the proposed model achieves stable and competitive forecasting performance across multiple horizons, while significantly improving computational efficiency. Ablation studies further validate the effectiveness of spatial partitioning and dual attention in capturing both local and long-range traffic dynamics. These results suggest that patch-based spatiotemporal modeling provides a scalable and effective framework for traffic forecasting under irregular spatial settings.

交通预测图神经网络时空建模分块结构

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