LLM2D
STEI-PCN:一种通过空间-时间编码和推断高效实现的纯卷积网络用于交通预测
STEI-PCN: an efficient pure convolutional network for traffic prediction via spatial-temporal encoding and inferring
作者: Kai Hu, Zhidan Zhao, Zhifeng Hao
发布日期: 4/14/2025
arXiv ID: oai:arXiv.org:2504.08061v1

摘要

arXiv:2504.08061v1 Announce Type: cross 摘要:交通数据表现出复杂的时间、空间以及时空相关性。大多数模型要么使用独立模块分别提取时间相关性和空间相关性,要么使用联合模块同步提取它们,而忽略了时空相关性。此外,考虑联合时空相关性(时间、空间和时空相关性)的模型常常面临准确性与计算效率的重大挑战,这使得这些模型无法充分发挥联合时空相关性架构的优势。为了解决这些问题,本文提出了一种通过时空编码与推理(STEI-PCN)的空间交通预测高效纯卷积网络。该模型引入并设计了一个基于绝对时空坐标和相对时空距离编码的动态邻接矩阵推理模块,利用带有门控机制的图卷积网络捕捉局部同步联合时空相关性。此外,还使用了三层时空扩张因果卷积网络来捕获长程时间相关性。最后,通过多视图协作预测模块,该模型将门控激活的原始、局部同步联合时空以及长程时间特征进行集成,以实现综合预测。在流量数据集(PeMS03/04/07/08)和速度数据集(PeMS-Bay)上进行了广泛的实验,涵盖了多个预测时间范围。结果显示,STEI-PCN在训练和推理速度上展现了具有竞争力的计算效率,并在大多数评估指标上达到了或略逊于最先进的(SOTA)模型。