提出轻量级卷积网络,动态建模交通传感器间时空关联,提升预测效率与精度。
Efficient Traffic State Prediction With Dynamic Joint Spatio-Temporal Relation Inference
- 构建局部因果时空图,结合位置与距离编码动态推断边权重。
- 单层图卷积+门控激活捕捉局部联合依赖,时序空洞卷积建模长程模式。
- 支持多步直接预测,参数少、训练推理成本低,适合实时交通系统部署。
交通状态预测因时间演化、空间交互及道路网中延迟的时空传播而复杂。现有方法或分离建模时空依赖,或使用统一结构,但往往未能充分刻画历史时刻邻近传感器对目标节点的影响,且复杂联合模型计算开销高。本文提出STEI-PCN,一种基于时空编码与关系推理的纯卷积网络。首先构建局部因果联合时空图以限制候选交互;再利用绝对位置与相对距离编码动态推断边权重。单层图卷积配合位置感知门控激活单元捕捉局部联合依赖,时序空洞因果卷积补充长程时间模式。多视角预测模块融合原始、局部传播与长程时间表示,实现直接多步预测。在PeMS03、PeMS04、PeMS07、PeMS08和PeMS-Bay数据集上,多个预测时长远超对比方法,精度媲美先进模型,同时保持参数量适中、训练与推理成本低。消融实验与波动分析验证了核心组件贡献,并实证分析了训练阶段约束在速度突变下的影响。代码已开源于GitHub。
原文摘要 · Abstract (English)
Traffic prediction is difficult due to the complex interplay of temporal evolution, spatial interactions, and delayed spatio-temporal propagation over road networks. Existing methods either model spatial and temporal dependencies separately or employ unified spatio-temporal structures, but they often insufficiently characterize how neighboring sensors at historical timestamps influence a target node, while complex joint models may incur high computation. This paper proposes STEI-PCN, an efficient pure convolutional network based on spatio-temporal encoding and relation inference. It first builds a local causal joint spatio-temporal graph to restrict candidate interactions, then uses absolute position and relative distance encodings to infer dynamic edge weights. A single-layer graph convolution with a position-aware gated activation unit captures local joint dependencies, and temporal dilated causal convolutions complement long-range temporal patterns. A multi-view prediction module fuses raw, local propagation, and long-range temporal representations for direct multi-step forecasting. Experiments on PeMS03, PeMS04, PeMS07, PeMS08, and PeMS-Bay under multiple horizons show that STEI-PCN achieves competitive accuracy with moderate parameters and low training and inference costs. Ablation and fluctuation analyses further verify the contributions of the main components and empirically analyze the effects of the training-stage constraints under sharp speed changes. Our code is available at a GitHub link https://github.com/Jessez2/STEI-PCN.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。