用均衡分区+线性交互,让交通预测模型更快更准。
SqLinear: Balanced Square Partitioning Makes Linear Interaction Sufficient for Large-Scale Traffic Forecasting

- 用自适应方块分区算法,把传感器分成均衡紧凑的区域。
- 线性交互模块替代注意力机制,计算量低且精度高。
- 适合超大规模交通预测,部署快、省时省力。
交通预测是智能交通系统和城市级决策的核心任务。尽管主流神经网络方法有效,但在数千个交通传感器的实际场景中,其计算可扩展性差,难以部署。现有方法依赖手工几何启发式划分,常导致分区不均、边界碎片化和过度填充开销,影响模型精度。本文提出SqLinear,一种高效的大规模交通预测架构。首先设计Square Partition算法,将海量传感器划分为平衡、非重叠、紧凑的空间区域,理论上保证分区利用率与均衡性,为时空建模奠定高质量基础。其次提出层级线性交互(HLI)模块,摒弃Transformer中昂贵的注意力机制,通过轻量级线性交互实现全局区域间依赖传播与节点级细化,实现线性复杂度的高效时空建模。在四个大规模交通数据集上,11种基线对比实验表明,SqLinear在标准设置下平均降低MAE 2.30%,极端可扩展设置下最高降低6.78%,训练时间减少13.27%–30.84%(空间与时序扩展场景)。
原文摘要 · Abstract (English)
Traffic prediction is a core task in intelligent transportation systems and urban-scale decision making. Despite the effectiveness of mainstream neural network-based methods, their deployment in real-world settings with thousands of traffic sensors is severely jeopardized by their poor computational scalability. To address this, the community has attempted to incorporate spatial database partitioning techniques to improve model scalability. However, these approaches rely on handcrafted geometric heuristics and often produce irregular or imbalanced data partitions, leading to boundary fragmentation, excessive padding overheads, and degraded model accuracy. In this paper, we propose SqLinear, an efficient and effective architecture for large-scale traffic prediction. First, we design Square Partition, a geometry-adaptive algorithm that partitions massive traffic sensors into balanced, non-overlapping, and compact spatial regions. Unlike existing heuristic-based designs, Square Partition is theoretically grounded and provides provable guarantees on partition utilization and split balance, establishing a high-quality foundation for downstream spatio-temporal modeling. Next, we propose a Hierarchical Linear Interaction (HLI) module that abandons the costly attention mechanisms commonly used in Transformer-based spatio-temporal models. HLI efficiently propagates global inter-region dependencies and refines them at the node level through a lightweight linear interaction scheme, enabling effective spatio-temporal modeling with linear computational complexity. Extensive experiments on four large-scale traffic datasets and 11 baselines show that SqLinear reduces MAE by 2.30% on average under the standard setting and by up to 6.78% under extreme scalability settings, while reducing training runtime by 13.27%--30.84% in spatial- and horizon-scaling scenarios.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。