浙江大学学报(工学版), 2026, 60(12): 2765-2774 doi: 10.3785/j.issn.1008-973X.2026.12.022

土木工程、交通工程

因果先验引导的交通流时空预测模型

侯越,, 王艳, 董雨璇, 赵建国, 张琳栋, 马静, 宿增强

1. 兰州交通大学 电子与信息工程学院,甘肃 兰州 730070

2. 西安交通大学 计算机科学与技术学院,陕西 西安 710000

3. 甘肃省交通规划勘察设计院股份有限公司,甘肃 兰州 730070

Causal prior-guided spatiotemporal traffic flow prediction model

HOU Yue,, WANG Yan, DONG Yuxuan, ZHAO Jianguo, ZHANG Lindong, MA Jing, SU Zengqiang

1. School of Electronic and Information Engineering, Lanzhou Jiaotong University, Lanzhou 730070, China

2. School of Computer Science and Technology, Xi’an Jiaotong University, Xi’an 710000, China

3. Gansu Province Transportation Planning Survey andb Design Institute Limited Company, Lanzhou 730070, China

收稿日期: 2025-09-12  

基金资助: 国家自然科学基金资助项目(62363020);甘肃省交通运输厅科技项目(2025-28).

Received: 2025-09-12  

Fund supported: 国家自然科学基金资助项目(62363020);甘肃省交通运输厅科技项目(2025-28).

作者简介 About authors

侯越(1979—),女,教授,博士,从事大数据智能交通研究.orcid.org/0000-0002-8289-329X.E-mail:houyue@mail.lzjtu.cn , E-mail:houyue@mail.lzjtu.cn

摘要

针对交通网络中流量变化驱动因素难以识别、预测性能有待提升的问题,提出因果先验引导的交通流时空预测模型. 采用时空特征融合模块,对空间维度进行离散化建模,并与周期性嵌入相融合,增强模型对复杂非线性特征的表征能力. 利用潜在因果发现模块,引入时滞表征因果卷积和因果筛选器,挖掘时间序列间的潜在因果关系. 采用因果验证模块,在假设检验框架下筛选真实因果关系,计算具有因果效应的节点之间的时延信息. 将经因果验证的结构作为先验知识嵌入动态图卷积模块,指导信息传播与特征聚合. 实验结果表明,与基线模型相比,所提模型在PeMSD4、PeMSD8和成都-滴滴数据集上的MAE指标分别降低了4.9197%、3.9518%、2.1968%,验证了因果机制在交通流预测中的有效性.

关键词: 交通流预测 ; 因果发现 ; 因果验证 ; 深度学习 ; 时空性

Abstract

A causal prior-guided spatiotemporal traffic flow prediction model was proposed in order to address the challenges of identifying the driving factors behind traffic flow variations and improving prediction performance. The spatiotemporal feature fusion module was used to discretize the spatial dimension and integrate it with periodic embeddings in order to enhance the representation of complex nonlinear feature. The latent causal discovery module was utilized to introduce a lag-aware causal convolution and a causal filter in order to capture potential causal relationship among time series. The causal validation module was used to identify true causal relationship under a hypothesis testing framework in order to calculate time delay between nodes with causal effect. The causally validated structure was embedded as prior knowledge into a dynamic graph convolution module in order to guide information propagation and feature aggregation. The experimental results demonstrated that the proposed model achieved reduction in MAE by 4.9197%, 3.9518% and 2.1968% on PeMSD4, PeMSD8 and Chengdu-DiDi datasets, respectively, compared with baseline models. The effectiveness of the causal mechanism in traffic flow prediction was verified.

Keywords: traffic flow prediction ; causal discovery ; causal validation ; deep learning ; spatio-temporality

PDF (1253KB) 元数据 多维度评价 相关文章 导出 EndNote| Ris| Bibtex  收藏本文

本文引用格式

侯越, 王艳, 董雨璇, 赵建国, 张琳栋, 马静, 宿增强. 因果先验引导的交通流时空预测模型. 浙江大学学报(工学版)[J], 2026, 60(12): 2765-2774 doi:10.3785/j.issn.1008-973X.2026.12.022

HOU Yue, WANG Yan, DONG Yuxuan, ZHAO Jianguo, ZHANG Lindong, MA Jing, SU Zengqiang. Causal prior-guided spatiotemporal traffic flow prediction model. Journal of Zhejiang University(Engineering Science)[J], 2026, 60(12): 2765-2774 doi:10.3785/j.issn.1008-973X.2026.12.022

随着城市路网日益复杂、机动车保有量不断增长,城市交通拥堵问题愈加严重. 交通流预测作为智能交通的核心技术[1],通过挖掘多源数据中的时空依赖与非线性规律实现对未来流量的预测. 受多种不确定因素影响,道路间存在显著关联,如何有效建模并量化这种关联性,对提升预测精度与可靠性至关重要.

传统的交通流预测方法由于未能充分挖掘交通数据中的时空关联性且难以建模非线性特征[2],被深度学习方法逐渐取代. 深度学习方法通过多层非线性映射,自适应地学习交通流数据中复杂的非线性规律,大幅提高了交通流预测的精度和效率. 由于交通数据同时具有时间相关性与空间相关性,单纯依赖时间建模难以充分表达路网特征,Li等[3]提出扩散卷积递归神经网络,将图卷积网络[4](GCN)与循环神经网络[5](RNN)相结合,通过扩散卷积建模路网的空间依赖,并利用循环神经网络捕捉时序特征,有效提升了交通流预测的精度. Guo等[6]提出基于注意力机制的图卷积神经网络,通过注意力机制动态建模时空依赖,从而提升了对复杂交通流动态模式的刻画能力. 上述模型的图构建方法大多依赖道路连通性或距离,难以刻画路网中潜在的空间相关性[7]. Song等[8]提出时空同步图卷积网络,通过构建时空图来替代单纯表示空间关系的邻接矩阵,同时引入图卷积操作提取时空特征[9].

上述研究虽然在一定程度上克服了复杂路网中空间依赖与时间依赖的建模障碍,但它们主要关注相关性而非因果性,难以解释道路间因果作用的来源、传播路径以及时延信息. 引入因果建模,可以提升交通流预测准确性并增强决策的可解释性.

Zhao等[10]提出因果条件隐马尔可夫模型,用于预测多模式交通流,但该模型依赖于静态因果图假设,难以刻画交通网络中随时间动态演化的因果关系,从而在捕捉复杂时空依赖和长时滞因果效应方面存在局限. Liu等[11]结合空间邻近性和时间先后性构建异常因果树,推断因果依赖关系,但忽略了对因果流向的刻画. Kapoor 等[12]针对道路交叉口场景,通过分析交叉点之间的因果关系,预测可能的拥堵传播模式. 上述方法侧重于空间维度的因果依赖,对时间序列间的因果关系关注不足. 格兰杰因果检验[13]能够深刻发掘时间序列数据之间潜在的因果关系,Li等[14]基于格兰杰因果和分解算法提取交通流时间序列的趋势,并结合2轮Lasso回归构建因果依赖图,从而揭示路网结构与交通流的相关性. He等[15]提出时空格兰杰因果关系,结合时空对齐算法与格兰杰因果检验,建模因果关系的全局传输与动态空间依赖. Zhang等[16]提出干预格兰杰因果结构学习模型,利用异构介入性时间序列数据推断格兰杰因果结构并识别未知目标,具有较强的鲁棒性. 格兰杰因果检验依赖线性假设,难以刻画数据中的非线性关系,并且在复杂路网中面临较高的计算开销.

针对上述不足,本文提出因果先验引导的交通流时空预测模型(causality prior guided spatio-temporal traffic flow prediction model, CP-STN). 在空间维度上,引入离散化建模机制,融合日周期嵌入与周周期嵌入,实现时空特征的联合表达. 在时间维度上,提出时滞表征因果卷积,用于从时间序列中挖掘潜在的因果关系. 设计因果验证机制,对潜在因果序列进行筛选,提取具有真实因果效应的节点关系及其因果延迟,并将经验证的因果结构作为动态图卷积的先验知识.

1. CP-STN框架

所提模型的整体框架如图1所示,主要包含以下3个模块.

图 1

图 1   CP-STN模型的整体架构

Fig.1   Overall architecture of CP-STN model


1)时空特征融合模块:将原始输入数据在空间维度进行离散化处理,通过计算Spearman相关性得到相关性矩阵$ \boldsymbol{M}\in {\bf{R}}^{{B}\times {N}\times {N}} $,并将该矩阵与日周期嵌入、周周期嵌入相融合,作为模型的输入.

2)潜在因果发现模块:利用时滞表征因果卷积在多尺度时间窗口中建模时序数据的依赖结构,构建IQR筛选器,并通过基于四分位距法生成的动态阈值对结果进行筛选,从而捕捉潜在因果序列.

3)因果验证模块:对于每个潜在因果变量,构建扰动数据集. 在假设无因果关系的条件下,通过比较干预结果与原始结果的偏差来判断因果关系是否存在,并进一步计算具有因果效应的节点之间的因果时延.

1.1. 时空特征融合

在空间维度上,道路节点间的流量关系不仅受到地理结构与交通环境约束,还表现出显著的非线性依赖特征. 为了有效刻画这种复杂关系,在空间建模中引入基于 Spearman 等级相关系数的相似性度量方法. 该方法能够捕捉节点间的非线性相关性,且在交通流数据常呈非正态分布的情况下具有更高的可靠性. 为了提升相似性表征的精度与稳健性,结合等距离散化策略,对相关性计算结果进行优化,从而构建更准确的空间非线性特征表示. 模块结构如图1(a)所示.

1)节点交通流量离散化.

以某一节点$ a $ 的交通流量为例,将滑动窗口封装得到的每个窗口内的流量数据进行离散化,如下所示:

$ {{L}}^{\prime}=\frac{\max \;({U}_{a})-\min\; ({U}_{a})}{k} . $

式中:$ {U}_{a} $为滑动窗口对节点$ a $ 流量的封装结果,$ k $为离散化后的目标区间数量,$ {{L}}^{\prime} $ 为离散化后的每个区间范围大小. 对于节点$ a $ 的每条交通流数据$ U_{a}^{i} $ ,离散化后的区间索引为

$ I_{a}^{i}=\left\lfloor \frac{U_{a}^{i}-\min\; ({U}_{a})}{{{L}}^{\prime}}\right\rfloor. $

式中:$ \left\lfloor \cdot \cdot \cdot \right\rfloor $表示向下取整,$ I_{\text{a}}^{i} $为节点$ a $的第$ i $条交通流数据离散化后的区间索引. 依次离散化节点$ a $的所有数据,得到列表$ {{\boldsymbol{I}}}_{a}=\left[I_{a}^{1},I_{a}^{2},\cdots ,I_{a}^{i}\right] $. 重复执行1),直至所有窗口内所有节点的交通流数据均被离散化. 如图2(a)~(c)所示分别为原始流量序列、Min-Max归一化后流量序列和离散化后流量序列. 其中,F为流量,N为归一化值,DI为离散索引. 从图2可以看出,离散化后的时间序列与原始序列的时间趋势高度一致.

图 2

图 2   原始流量序列离散前、后的对比图

Fig.2   Comparison of original traffic flow series and its discretized counterpart


2)计算Spearman 相关性$ {\textit{ρ}} $:

$ {\textit{ρ}} =1-\frac{6\displaystyle \sum_{k=1}^n d_{k}^{2}}{{{n(n}}^{2}-1)} . $

式中:$ {n} $为数据点数量;$ {d}_{k} $ 为第$ k $ 个时间步时节点$ {X}_{i} $与$ {X}_{j} $的秩次之差,即两者在排序后序列中的位置差值,

$ {d}_{k}=R({X}_{i})-R({X}_{j}) , $

其中$ {R}\left(\cdot \right) $为节点在排序后序列中的位置.

重复执行1)、2),得到所有样本的Spearman相关性矩阵$ \boldsymbol{M}\in {\bf{R}}^{{B}\times {N}\times {N}} $. 对矩阵$ \boldsymbol{M} $进行投影,

$ {\boldsymbol{M}}_{\mathrm{proj}}=\boldsymbol{M}{\boldsymbol{W}}_{\mathrm{proj}}+{\boldsymbol{b}}_{\mathrm{proj}} . $

式中:$ {\boldsymbol{M}}_{\mathrm{proj}} $为投影后的矩阵,$ {\boldsymbol{W}}_{\text{proj}} $、$ {\boldsymbol{b}}_{ \text{proj}} $为可学习的参数矩阵.

将$ {\boldsymbol{M}}_{\mathrm{proj}} $与初始静态矩阵$ {\boldsymbol{E}}_{0} $相融合,得到空间嵌入$ {\boldsymbol{E}}_{\text{M}} $,

$ {\boldsymbol{E}}_{\mathrm{M}}={\boldsymbol{E}}_{0}+\mathrm{unsqueeze}\;(\tanh \;({\boldsymbol{M}}_{\mathrm{proj}})) . $

将空间嵌入融入日周期嵌入和周周期嵌入,得到模型输入$ \boldsymbol{E} $,

$ \boldsymbol{E}={\boldsymbol{T}}^{{\mathrm{D}}}\odot {\boldsymbol{T}}^{{\mathrm{W}}}\odot {\boldsymbol{E}}_{{\mathrm{M}}} . $

式中:TD、TW分别为广播机制扩维后的日周期嵌入和周周期嵌入.

日周期嵌入和周周期嵌入的计算过程如下:

$ \left.\begin{aligned} & T_{i}^{{\mathrm{D}}}=\dfrac{i\;{\text{mod }}S_{\text{day}}}{{{S}}_{\text{day}}},\\&T_{i}^{{\mathrm{W}}}=\left\lfloor \dfrac{i}{{{S}}_{\text{day}}}\right\rfloor \text{mod 7}.\end{aligned}\right\} $

式中:$ i $表示第$ i $个时间步,$ T_{i}^{{\mathrm{D}}} $、$ T_{i}^{{\mathrm{W}}} $分别为第$ i $个时间步的日周期嵌入和周周期嵌入,$ {{S}}_{\mathrm{day}} $为一天包含的时间步数.

1.2. 潜在因果发现

尽管基于 Spearman 等级相关的空间非线性建模能够有效刻画复杂交通流特征,但该方法仅能反映相关性而难以区分因果驱动关系,存在解释性不足的问题. 本文进一步引入潜在因果发现机制,从因果角度揭示节点间的真实影响路径,从而弥补相关性建模的不足,提升时空依赖表征的准确性.

在潜在因果发现模块中设计基于因果注意力机制的潜在因果发现方法,引导模型在学习各输入时间序列对目标序列关注度的过程中,捕捉具备潜在因果依赖关系的信号,以提升因果结构建模的准确性与对关键影响因素的识别能力. 该模块的结构如图1(b)和图3所示.

图 3

图 3   潜在因果发现模块图

Fig.3   Diagram of latent causal discovery module


为了刻画时间序列的因果依赖,将节点$ i $的时间序列$ {{\boldsymbol{X}}}_{i}=\left[X_{i}^{1},X_{i}^{2},X_{i}^{3},\cdot \cdot \cdot ,X_{i}^{t}\right] $填充为$ {\boldsymbol{X}}_{i}^{\prime}=\left[0,X_{i}^{1},X_{i}^{2}, X_{i}^{3},\cdot \cdot \cdot ,X_{i}^{t}\right] $,利用通道独立的深度扩张卷积扩展感受野,以捕获长时滞的潜在因果关系. 各通道序列通过逐点卷积进行融合,经线性层调整维度,实现跨通道信息整合. 将处理后的序列输入IQR因果筛选器,以识别具有统计显著性的潜在因果序列.

1.2.1. 时滞表征因果卷积

在交通流建模中,不同路段间的信息传播往往具有显著的时间滞后性和非局部依赖性,例如高峰期某一关键节点的拥堵可能在数分钟后蔓延至相邻区域. 传统的卷积神经网络感受野受限,难以捕捉这种长距离的时序影响. 本文在潜在因果发现模块中设计时滞表征因果卷积,通过在卷积核中引入扩张间隔以指数级扩展感受野,实现对长程依赖的有效建模,适用于捕捉交通流中潜在的长时滞因果关系. 第$ l $层的感受野为

$ {H}=1+\sum \limits_{l=0}^{{L}}({K}-1) {{c}}^{l}. $

式中:$ {K} $为卷积核大小,$ {c} $为扩张系数.

为了能够更平滑地处理连续时序中滞后、非线性的交通流量数据,提升模型对复杂交通动态的建模能力,在因果卷积中使用GELU作为激活函数.

$ \text{GELU}(x)=\frac{x}{2}\left(1+\text{erf}\left(\frac{x}{\sqrt{2}}\right)\right) . $

式中:erf为误差函数.

1.2.2. IQR因果筛选器

在复杂时序系统的潜在因果发现任务中,注意力机制虽然能够高效建模高维时序数据的非线性依赖,但注意力分数仅反映节点间的相关性,缺乏因果推断所需的外部扰动与方向性. 注意力分数的高密集性进一步增加了在干扰因素作用下识别真实因果关系的复杂性,显著限制了因果结构学习的准确性与效率. 本文提出基于注意力权重分布的筛选机制,通过非参数统计方法捕捉注意力分数的数值分布差异,自适应生成分离阈值,将对目标变量预测有显著贡献的变量筛选为潜在因果候选集,剔除弱相关干扰项. 具体步骤如下.

1)归一化预处理及排序.

对原始注意力分数$ {{\boldsymbol{Z}}}=\left\{{{{\boldsymbol{z}}}}_{1},{{{\boldsymbol{z}}}}_{2},\cdots ,\left.{{{\boldsymbol{z}}}}_{{N}}\right\}\right. $进行归一化处理,按照数值由高到低排序. 归一化处理如下:

$ {a}_{ij}={\mathrm{softmax}}\;({z}_{ij})=\frac{{{\mathrm{e}}}^{{{{z}}_{{ij}}}}}{\displaystyle \sum \nolimits_{j=1}^{N}{{\mathrm{e}}}^{{{z}_{ij}}}} . $

式中:$ {z}_{ij} $为第i个时间序列与第$ j $个时间序列之间的注意力分数,$ {N} $为样本总量. 对于第$ j $个序列,排序后的注意力分数序列为$ {{\boldsymbol{a}}}_{j}=[{a}_{1,j},{a}_{2,j},\cdot \cdot \cdot ,{a}_{n,j}] $.

2)基于四分位距的动态阈值确定.

传统的固定阈值方法往往缺乏对数据分布的动态适应性,可能导致筛选结果过于严格或过于宽松. 为了解决该问题,引入基于四分位距(IQR)的动态阈值策略,捕捉注意力分数分布中的显著偏离值. 具体来说,对于步骤1)得到的注意力分数序列,计算该序列的四分位数.

第一四分位数 (Q1):将数据集分成四等份,处于25%位置上的数值.

第三四分位数 (Q3):将数据集分成四等份,处于75%位置上的数值.

基于Q1和Q3,计算四分位距:

$ \mathrm{IQR}={{Q}}_{3}-{{Q}}_{1} . $

定义动态阈值$ \tau $,该阈值自适应地反映注意力分数的离散程度与偏斜情况,其中$ {m} $为可配置的乘数,用于调节阈值相对于数据主要集中区域的偏离程度.

$ \tau ={{Q}}_{3}+{m}\times \mathrm{IQR} . $

3)因果候选序列筛选.

如果某一序列$ {X}_{i} $与目标序列$ {X}_{j} $的注意力分数$ {a}_{i,j} $大于$ \tau $,则将该序列加入目标时间序列的潜在因果候选序列列表.

为了确保筛选结果的合理性并减少因果判定的误判,限制筛选出的潜在因果变量的数量不超过总输入变量的一半. 这有助于避免模型过度解读噪声或过度依赖弱相关性,从而提升后续因果验证的效率和准确性.

1.3. 因果验证

尽管潜在因果序列能够反映变量间可能的依赖关系,但其本身并不保证真正的因果性. 为了有效剔除非因果依赖,引入基于假设检验的因果验证机制:通过对目标序列的候选因果序列实施扰动干预,观察目标序列预测性能的变化,选出对目标变量具有显著时序影响的因果节点,形成因果关系发现与验证的闭环,从而提升因果结构建模的准确性与可靠性. 该模块的结构如图1(c)和图4所示.

图 4

图 4   因果验证模块图

Fig.4   Causal validation module diagram


对于目标时间序列$ {{\boldsymbol{S}}}_{i} $,设其潜在因果序列为$ {{\boldsymbol{S}}}_{j} $,现引入$ {{\boldsymbol{S}}}_{j} $的扰动数据集$ {\boldsymbol{S}}_{j}^{\prime} $. 执行$ {d} $次顺序扰动,每次都将模型应用于新的置乱数据集,并记录由此产生的损失变化.

$ \Delta {L}^{{(d)}}={L}^{{(1,d)}}-{L}^{{({\mathrm{e}},d)}} . $

式中:$ {L}^{{(1,d)}} $为置乱数据集在第1轮训练中的损失值,$ {L}^{{({\mathrm{e}},d)}} $为置乱数据集在最终轮次的损失值. 原始数据集上的损失值变化量为

$ \Delta {L}^{(0)}={L}^{(1,0)}-{L}^{\text{(e,0)}} . $

式中:$ {L}^{(1,0)} $为原始数据集首轮的损失值,$ {L}^{\text{(e,0)}} $为原始数据集最终轮的损失值. 计算在零假设(假设不存在因果关系)下的概率:

$ {P}=\frac{\text{count}\;(\Delta {L}^{{(d)}}\leqslant \Delta {L}^{(0)})}{{d}} . $

式中:count$(\Delta {L}^{{(d)}}\leqslant \Delta {L}^{(0)}) $为d次扰动中损失变化值小于等于$\Delta {L}^{(0)} $的频次. 若$ {P} $值低于预设的显著性水平$ \alpha $,则拒绝零假设,即认为该因果关系存在. 存在因果关系的序列之间的因果注意力分数矩阵为$ {\boldsymbol{M}}_{\text{y}} $.

在提取目标序列的真实因果序列后,引入因果滞后来量化因果序列对目标序列的时间延迟作用,进而建模节点间的时空依赖特性. 节点$ a $的单层因果滞后时间步delaya,l与总因果滞后时间步TotalDelaya的计算分别如下所示:

$ {\text{delay}}_{a,l}={W}_{l} {{c}}^{l} ,$

$ {\text{TotalDelay}}_a=\sum \nolimits_{l=0}^{{L-1}}{\text{delay}}_{a,l} .$

式中:$ {W}_{\textit{l}} $为在第$ \textit{l} $层的卷积核中最大权重对应的相对位置.

1.4. 时空因果融合

在CP-STN模型中,输入数据先经过空间维度的离散化处理,再与日周期、周周期嵌入融合. 融合后的特征通过多层感知机映射,进而计算节点间的空间相似度,捕捉更高阶的空间相似性结构,并将该结构融入上下文感知门控循环单元(GRU)的输入中,从而增强空间信息的表达能力. 因果验证模块生成的因果注意力矩阵$ {\boldsymbol{M}}_{\text{y}} $被引入历史隐藏状态$ {{\boldsymbol{h}}}_{{t-1}} $,从因果视角对前序驱动因素进行建模,弥补相关性建模在解释性方面的不足. 上下文感知GRU的计算过程如下:

$ {{\boldsymbol{r}}}_{{t}}=\sigma\; ({\boldsymbol{W}}_{\text{ir}}{{\boldsymbol{x}}}_{{t}}+{\boldsymbol{W}}_{\text{hr}}{{\boldsymbol{h}}}_{{t-1}}+{\boldsymbol{b}}_{\text{r}}) , $

$ {{\boldsymbol{z}}}_{{t}}=\sigma\; ({\boldsymbol{W}}_{\text{iz}}{{\boldsymbol{x}}}_{{t}}+{\boldsymbol{W}}_{\text{hz}}{{\boldsymbol{h}}}_{{t-1}}+{\boldsymbol{b}}_{\text{z}}) , $

$ {{\boldsymbol{g}}}_{\text{c}}=\sigma\; ({\boldsymbol{W}}_{\text{ic}}{\boldsymbol{C}}_{{t}}+{\boldsymbol{W}}_{\text{hc}}{{\boldsymbol{h}}}_{{t-1}}+{\boldsymbol{b}}_{\text{c}}) , $

$ \widetilde{{{\boldsymbol{h}}}_{{t}}}=\tanh \;({\boldsymbol{W}}_{\text{in}}{{\boldsymbol{x}}}_{{t}}+{{\boldsymbol{r}}}_{{t}}\odot ({\boldsymbol{W}}_{\text{hn}}{{\boldsymbol{h}}}_{{t-1}})+{{\boldsymbol{g}}}_{\text{c}}\odot {\boldsymbol{C}}_{{t}}) , $

$ {{\boldsymbol{h}}}_{{t}}=({\boldsymbol{1}}-{{\boldsymbol{z}}}_{{t}})\odot \widetilde{{{\boldsymbol{h}}}_{{t}}}+{{\boldsymbol{z}}}_{{t}}\odot {{\boldsymbol{h}}}_{{t-1}} . $

式中:$ {{\boldsymbol{x}}}_{{t}} $为当前时间步输入,$ {{\boldsymbol{h}}}_{{t-1}} $为上一个时间步隐藏状态,$ {\boldsymbol{C}}_{{t}} $为当前时间步的上下文向量,$ \widetilde{{{\boldsymbol{h}}}_{{t}}} $为候选隐藏状态,$ {{\boldsymbol{r}}}_{{t}} $、$ {{\boldsymbol{z}}}_{{t}} $分别为重置门与更新门,$ {{\boldsymbol{g}}}_{\text{c}} $为上下文向量的门控,$ {\boldsymbol{W}}_{\text{ir}} $、$ {\boldsymbol{W}}_{\text{hr}} $、$ {\boldsymbol{W}}_{\text{iz}} $、$ {\boldsymbol{W}}_{\text{hz}} $、$ {\boldsymbol{W}}_{\text{ic}} $、$ {\boldsymbol{W}}_{\text{hc}} $、$ {\boldsymbol{W}}_{\text{in}} $、$ {\boldsymbol{W}}_{\text{hn}} $为可学习的参数向量,$ {\boldsymbol{b}}_{\text{r}} $、$ {\boldsymbol{b}}_{\text{z}} $、$ {\boldsymbol{b}}_{\text{c}} $为偏置向量.

经上下文感知GRU处理后得到的隐状态,既融合了时间序列特征,又结合了因果上下文信息. 在此基础上,引入动态图卷积对节点之间的高阶空间依赖与动态拓扑结构进行进一步建模. 动态图卷积利用二阶邻接相似度及图结构对节点进行加权更新,从而在全局范围内捕捉复杂的时空交互模式,实现时间依赖与空间结构的深度耦合,显著提升交通流预测的精准度. 该模块的结构如图1(d)所示.

2. 实验分析

2.1. 数据集和实验设置

采用PeMSD4、PeMSD8的流量数据和成都-滴滴的速度数据进行实验. PeMSD4与PeMSD8数据集均来自加利福尼亚州公路交通传感器网络,其中PeMSD4包含2018年1月至2月期间共307个交通传感器的流量数据,采样间隔为5 min;PeMSD8包含2016年7月至8月期间共170个交通传感器的流量数据,采样间隔同样为5 min. 成都-滴滴数据长度为17 280个时间步,包含524个节点,时间为2018年1月至4月,采样间隔为10 min. 训练集、验证集、测试集按6∶2∶2的比例进行划分.

使用PyTorch 3.10,在配备NVIDIA GeForce RTX 4090显卡的设备上进行实验. 实验中,PeMSD4和PeMSD8的历史滑动窗口及未来滑动窗口均设为12,成都-滴滴的历史与未来滑动窗口均设为6,批大小为16,初始学习率为0.001,并设置指数衰减率为0.001,在训练过程中动态更新学习率,优化器选用Adam. 实验采用平均绝对误差MAE、均方根误差RMSE和决定系数R2作为评价指标,计算如下:

$ \text{MAE}=\frac{1}{{N}}\sum \limits_{i=1}^{{N}}\left| {y}_{i}-\left.{\overline{y}}_{i}\right| \right. , $

$ \text{RMSE}=\sqrt{\frac{1}{{N}}\sum \limits_{i=1}^{{N}}{({{y}_{i}}-{{\overline{y}}_{i}})}^{2}} , $

$ {{R}}^{2}=1-\frac{\displaystyle \sum \limits_{j=1}^{{N}}{({{\overline{y}}_{j}}-{{y}_{j}})}^{2}}{\displaystyle \sum \limits_{j=1}^{{N}}{(\overline{y}-{{y}_{j}})}^{2}} . $

式中:$ {y}_{i} $为真实值,$ {\overline{y}}_{i} $为预测值.

2.2. 对比实验

为了验证所提CP-STN模型的有效性并考察其在不同交通结构下的适应性,选取高速公路与城市出行2类典型交通场景,分别对应PeMSD4、PeMSD8和成都-滴滴3个真实交通数据集,开展针对未来一小时交通流的平均性能指标对比实验,将CP-STN与10种基准模型进行比较. 10种基准模型包括时间卷积网络(TCN)、长短期记忆神经网络(LSTM)、扩散卷积递归神经网络[3](DCRNN)、基于注意力机制的时空图卷积网络[6](ASTGCN)、下采样卷积交互网络[17](SCINet)、图神经控制微分方程[18](STG-NCDE)、深度时空不确定性量化[19](DeepSTUQ)、贝叶斯图卷积网络[20](BGCN)、考虑路网虚拟空间的交通流预测模型[21](FGI)、基于信息瓶颈与原型的时间回归任务图可解释性研究[22](GINTRIP). 实验结果如表1所示.

表 1   不同模型在3个交通流数据集上的预测准确度

Tab.1  Prediction accuracy of different models on three traffic flow datasets

模型PeMSD4PeMSD8成都-滴滴
MAERMSER2MAERMSER2MAERMSER2
TCN23.0636.030.945619.2230.160.95462.794.050.8234
LSTM22.1436.110.946920.1731.500.95372.784.040.8263
DCRNN22.2834.290.916616.7425.630.92882.784.090.5818
ASTGCN20.4432.810.919316.4925.360.92852.353.490.6920
SCINet19.7531.690.959816.5025.710.96792.253.450.6905
STG-NCDE20.2931.360.950915.8324.890.97082.323.440.6932
DeepSTUQ(23)19.1531.360.933915.0423.970.93882.293.440.6933
BGCN(24)18.8331.730.959414.8223.840.97322.353.340.7523
FGI(24)19.3131.680.928214.9323.640.93962.193.310.7142
GINTRIP(24)18.6631.010.957214.9124.050.97382.293.370.7075
CP-STN18.3630.230.967814.3423.600.96992.143.180.8469

新窗口打开| 下载CSV


分析表1可知,TCN和LSTM虽能较好地捕捉交通流数据中的时序特征,但由于未建模空间特征,两者的预测性能较差. DCRNN将交通流预测建模为有向图上的扩散过程,ASTGCN是基于注意力机制的图卷积网络,同时对空间和时间特征建模,这2个模型的预测性能均有所提升. SCINet将时间序列下采样为2个子序列,并采用不同的卷积滤波器提取各子序列特征,以保留多尺度信息. STG-NCDE基于神经控制微分方程,对时空序列分别建模. DeepSTUQ关注交通预测的不确定性量化,通过对任意不确定性和认知不确定性进行估计,提升预测精度. BGCN引入贝叶斯方法以增强模型的泛化能力,解决了确定性模型面临的不确定性捕获及过拟合问题. FGI通过自适应加权邻接矩阵和特征增强机制学习道路节点的时变关联特性,取得了较高的预测精度. GINTRIP结合信息瓶颈原理与原型表示机制,有效增强了时间图回归模型的可解释性,在模型可理解性与预测精度上均优于FGI.

本文提出的CP-STN模型引入潜在时序因果发现机制,挖掘交通流因果关系并建模关键路径,不仅显著提升了模型预测的准确性,也增强了模型的可靠性. 与FGI模型相比,CP-STN模型在PeMSD4和PeMSD8数据集上的MAE指标分别降低了4.9197%、3.9518%,验证了CP-STN优越的预测精度.

为了更直观地评估CP-STN模型在未来一小时内的多步预测性能,图5可视化了该模型与8个基准模型在3个数据集上的预测结果. 在成都-滴滴数据集上,DCRNN的预测性能随时间推移波动较大,稳定性不足;ASTGCN因同时建模空间与时间特征,预测性能较DCRNN更稳定且效果更优. 在PeMSD4数据集上,FGI的R2指标下降较明显,而DeepSTUQ则表现更平稳. CP-STN模型的误差曲线上升最缓慢,性能最稳定.

图 5

图 5   CP-STN与基准模型在3个数据集上的预测性能对比

Fig.5   Prediction performance comparison of CP-STN and baseline models on three datasets


2.3. 消融实验

在PeMSD4和PeMSD8数据集上进行消融实验,验证CP-STN模型各组成部分的有效性. 实验结果如表2、3所示. 各消融变体设置如下.

表 2   CP-STN模型在PeMSD4数据集上的消融实验结果

Tab.2  Ablation study result of CP-STN model on PeMSD4 dataset

模型MAERMSER2
w/o D18.3730.650.9634
w/o DW19.3231.510.9417
w/o CD18.4230.730.9601
w/o IQR18.4330.710.9599
w/o CV18.4030.600.9631
CP-STN18.3630.230.9678

新窗口打开| 下载CSV


w/o D:取消时空特征融合模块中的离散化处理.

w/o DW:时空特征融合模块中去除日周期嵌入和周周期嵌入.

w/o CD:使用均匀分布的注意力分数替换因果注意力分数.

w/o IQR:取消潜在因果发现模块中的IQR因果筛选器,原始注意力分数直接进入因果验证模块.

w/o CV:在保留潜在因果发现模块的基础上取消因果验证模块.

如表2、3所示,去除离散化处理(w/o D)后,模型在PeMSD4数据集上的MAE升高至18.37,表明空间离散化处理对捕捉非线性依赖、提升预测精度具有积极作用. 在去除日周期嵌入和周周期嵌入(w/o DW)后,模型在PeMSD4数据集上的MAE相较于完整模型提高了4.9689%,这表明时空数据具有显著的周期性和规律性. 用均匀分布的注意力分数替代因果注意力分数(w/o CD)后,模型在PeMSD8数据集上的MAE升高至14.51,印证了识别潜在因果的有效性. 在去除因果验证模块中的IQR因果筛选器(w/o IQR)后,模型在PeMSD8数据集上的MAE和RMSE较完整模型分别提高了0.8984%和1.6659%,说明在因果验证前进行因果筛选的必要性. 在去除因果验证模块(w/o CV)后,模型在PeMSD4数据集上的MAE升高至18.40,揭示了因果验证模块在筛选真实因果关系与提升模型准确性方面的积极作用. 综上所述,CP-STN模型在3项评估指标上均优于该模型的消融变体,在预测未来一小时交通流时具有更优的表现.

表 3   CP-STN模型在PeMSD8数据集上的消融实验结果

Tab.3  Ablation study result of CP-STN model on PeMSD8 dataset

模型MAERMSER2
w/o D14.4423.850.9683
w/o DW14.5924.930.9510
w/o CD14.5123.980.9613
w/o IQR14.4724.010.9611
w/o CV14.4923.910.9612
CP-STN14.3423.610.9699

新窗口打开| 下载CSV


2.4. 因果注意力可视化

对PeMSD8数据集中部分节点的因果注意力分数Sa进行可视化分析. 如图6所示,颜色越深表示相应传感器之间的因果作用越显著. 可以看出,对角线颜色整体较深,表明模型在预测时主要依赖时间序列自身的历史数值,这反映了模型对时间序列历史依赖关系的有效建模能力. 此外,部分非对角线位置也呈现出明显的深色区域,例如节点16与节点9、节点2与节点5之间存在较强的注意力分数,揭示了它们在路网中存在显著的跨节点因果关系. 上述结果表明,模型不仅能够捕捉节点间的空间依赖,还能量化关键路段之间的相互影响,凸显了该模型对复杂空间依赖性的刻画能力.

图 6

图 6   部分节点因果注意力分数热力图

Fig.6   heatmap of causal attention scores for partial nodes


2.5. 因果时延

为了深入揭示交通网络中节点流量变化与因果关系的动态演化特征,选取PeMSD8数据集中的节点{37,106,134,136}作为典型研究对象. 图7展示了这些节点在14:55—15:55这1 h内的流量变化,图8呈现了节点间的多源因果关系及对应的因果时延.

图 7

图 7   部分节点在不同时间段的流量变化

Fig.7   Traffic flow variations of partial nodes across different time periods


图 8

图 8   部分节点间的因果时延图

Fig.8   Causal time delay diagram among partial nodes


从图8可以看出,节点37与节点134的流量变化在一个时间片(5 min)后会共同作用于节点136. 例如在15:20,节点37与节点134的共同作用使节点136在15:25时的交通流量呈现下降趋势. 这种多源因果效应揭示了交通网络中节点间复杂的交互机制. 此外,节点134在5 min后对自身交通状况产生影响,说明节点的历史状态也是影响节点未来状态的重要因素. 上述结果不仅验证了模型对因果延迟和多节点交互关系的刻画能力,也为理解交通流在时空维度上的传播规律提供了依据.

2.6. 超参数实验

为了确定模型层数$ {L} $的最佳值,并验证不同超参数对模型性能的影响,实验结果如表4所示. 随着模型层数的增加,模型指标MAE和RMSE呈现先下降后上升的趋势. 当L = 2时模型性能最优,在PeMSD8数据集上的MAE和RMSE分别为14.34、23.60,较L = 1时分别降低了1.6461%和0.1269%.

表 4   不同模型层数在PeMSD8数据集上的平均预测结果

Tab.4  Average prediction result of different model layers on PeMSD8 dataset

LMAERMSER2
114.5823.630.9651
214.3423.600.9699
314.4523.610.9613

新窗口打开| 下载CSV


如表5所示,当超参数m取值为0.8、1.0时,模型性能均低于最优值;当m = 1.2时,模型性能最优. 在PeMSD8数据集上,与 m = 1.0 相比,m = 1.2时的MAE 降低 0.4858%,RMSE降低 0.5059%;与 m = 1.4 相比,MAE降低 0.3475%,RMSE降低 0.3799%.

表 5   不同超参数m在PeMSD8数据集上的平均预测结果

Tab.5  Average prediction result of different hyperparameter m on PeMSD8 dataset

mMAERMSER2
0.814.5323.910.9591
1.014.4123.720.9621
1.214.3423.600.9699
1.414.3923.690.9617

新窗口打开| 下载CSV


如表6所示,当显著性水平α取值为0.25时,模型性能最优:在PeMSD8数据集上,MAE和RMSE相较于α=0.20时分别降低了0.069 97%和0.463 9%,相较于α = 0.30时分别降低了0.278 2%和0.923 6%.

表 6   不同超参数α在PeMSD8数据集上的平均预测结果

Tab.6  Average prediction result of different hyperparameter α on PeMSD8 dataset

αMAERMSER2
0.2014.3523.710.9683
0.2514.3423.600.9699
0.3014.3823.820.9613
0.3514.4723.930.9551

新窗口打开| 下载CSV


2.7. 效率实验

为了评估 CP-STN 模型的计算性能,从平均每轮训练时间ttr、MAE指标及参数量Np 3个方面对模型进行效率评估. 实验基于PeMSD8数据集,在固定硬件环境下记录各模型的训练时间. 从表7可知,CP-STN的参数量显著少于SCINet与STG-NCDE模型. 尽管DeepSTUQ与GINTRIP模型训练时间较短,但预测精度相对较低,难以兼顾效率与性能.

表 7   基于时间、精度与参数量的模型效率对比

Tab.7  Efficiency comparison of models by time, accuracy and parameter count

模型ttr/sMAENp/103
ASTGCN12716.49179.46
SCINet6416.502 778.62
STG-NCDE24115.832 547.28
DeepSTUQ3515.04198.64
GINTRIP2114.91415.87
CP-STN6314.34218.63

新窗口打开| 下载CSV


3. 结 论

(1)时空融合模块在空间维度上结合Spearman等级相关系数与等距离散化,有效刻画了节点间复杂非线性依赖,为潜在因果发现提供了更可靠的特征输入,增强了空间依赖建模能力.

(2)设计时滞表征因果卷积,逐层扩展感受野以捕捉多时间尺度依赖,结合基于四分位距的自适应阈值,实现了对潜在因果关系的准确识别.

(3)在零假设下对潜在因果关系进行统计检验,并进一步量化时间延迟特征,为因果结构的精细化分析提供了依据.

未来工作将引入天气、节假日、突发事故等外部因素,并将其融入因果建模框架,以更全面地刻画交通流演化规律.

参考文献

崔建勋, 要甲, 赵泊媛

基于深度学习的短期交通流预测方法综述

[J]. 交通运输工程学报, 2024, 24 (2): 50- 64

DOI:10.19818/j.cnki.1671-1637.2024.02.003      [本文引用: 1]

Cui Jianxun, Yao Jia, Zhao Boyuan

A review of short-term traffic flow prediction methods based on deep learning

[J]. Journal of Transportation Engineering, 2024, 24 (2): 50- 64

DOI:10.19818/j.cnki.1671-1637.2024.02.003      [本文引用: 1]

秦凡. 基于时空图神经网络的交通流量预测方法研究[D]. 北京: 北京化工大学, 2025.

[本文引用: 1]

Qin Fan. Research on traffic flow prediction method based on spatio-temporal graph neural network [D]. Beijing: Beijing University of Chemical Technology, 2025.

[本文引用: 1]

Li Y G, Yu R, Shahabi C, et al. Diffusion convolutional recurrent neural network: data-driven traffic forecasting [PP/OL]. arXiv (2017-07-06) [2025-05-27]. https://arxiv.org/abs/1707.01926.

[本文引用: 2]

Zhang H, Lu G, Zhan M, et al

Semi-supervised classification of graph convolutional networks with Laplacian rank constraints

[J]. Neural Process Lett 54, 2022, 54 (4): 2645- 2656

DOI:10.1007/s11063-020-10404-7      [本文引用: 1]

ZAREMBA W, SUTSKEVER I, VINYALS O. Recurrent neural network regularization [PP/OL]. arXiv (2015-02-19) [2025-05-27]. https://arxiv.org/abs/1409.2329.

[本文引用: 1]

Guo S, Lin Y, Feng N, et al. Attention based spatial-temporal graph convolutional networks for traffic flow forecasting [C]//Proceedings of the AAAI Conference on Artificial Intelligence. Honolulu: AAAI Press, 2019: 922−929.

[本文引用: 2]

林霆堉, 邬群勇, 王珂月

基于时间图卷积网络的交通流时空预测模型

[J]. 贵州大学学报(自然科学版), 2025, 42 (4): 88- 93

DOI:10.15958/j.cnki.gdxbzrb.2025.04.11      [本文引用: 1]

Lin Tingyu, Wu Qunyong, Wang Keyue

Traffic flow spatiotemporal prediction model based on temporal graph convolutional network

[J]. Journal of Guizhou University (Natural Science Edition), 2025, 42 (4): 88- 93

DOI:10.15958/j.cnki.gdxbzrb.2025.04.11      [本文引用: 1]

Song C, Lin Y, Guo S, et al. Spatial-temporal synchronous graph convolutional networks: a new framework for spatial-temporal network data forecasting [C]// Proceedings of the AAAI Conference on Artificial Intelligence. Palo Alto: AAAI Press, 2020: 914–921.

[本文引用: 1]

贺文武, 裴博彧, 李雅婷, 等

基于双向自适应门控图卷积网络的交通流预测

[J]. 交通运输系统工程与信息, 2023, 23 (1): 187- 197

DOI:10.16097/j.cnki.1009-6744.2023.01.020      [本文引用: 1]

He Wenwu, Pei Boyu, Li Yating, et al

Traffic flow prediction based on bidirectional adaptive gated graph convolutional network

[J]. Journal of Transportation Systems Engineering and Information, 2023, 23 (1): 187- 197

DOI:10.16097/j.cnki.1009-6744.2023.01.020      [本文引用: 1]

Zhao Y, Deng P, Liu J, et al. Causal conditional hidden Markov model for multimodal traffic prediction [C]//Proceedings of the AAAI Conference on Artificial Intelligence. Washington, DC: AAAI Press, 2023: 4929-4936.

[本文引用: 1]

Liu W, Zheng Y, Chawla S, et al. Discovering spatio-temporal causal interactions in traffic data streams [C]//Proceedings of the 17th ACM SIGKDD International Conference on Knowledge Discovery and Data Mining. New York: ACM, 2011: 1010–1018.

[本文引用: 1]

Kapoor V, Saxena D, Raychoudhury V, et al. Real time building and maintaining causal congestion graph for intelligent traffic management [C]//2018 IEEE International Conference on Pervasive Computing and Communications Workshops. Athens: IEEE, 2018: 770-775.

[本文引用: 1]

Granger C W J

Investigating causal relations by econometric models and cross-spectral methods

[J]. Econometrica: Journal of the Econometric Society, 1969, 37 (3): 424- 438

[本文引用: 1]

Li L, Su X, Wang Y, et al

Robust causal dependence mining in big data network and its application to traffic flow predictions

[J]. Transportation Research Part C: Emerging Technologies, 2015, 58: 292- 307

DOI:10.1016/j.trc.2015.03.003      [本文引用: 1]

He S, Luo Q, Du R, et al

STGC-GNNs: a GNN-based traffic prediction framework with a spatial–temporal Granger causality graph

[J]. Physica A: Statistical Mechanics and its Applications, 2023, 623: 128913

DOI:10.1016/j.physa.2023.128913      [本文引用: 1]

Zhang Z, Ren S, Qian X, et al. Learning flexible time-windowed granger causality integrating heterogeneous interventional time series data [C]//Proceedings of the 30th ACM SIGKDD Conference on Knowledge Discovery and Data Mining. New York: ACM, 2024: 4408-4418.

[本文引用: 1]

Liu M, Zeng A, Chen M, et al

Scinet: time series modeling and forecasting with sample convolution and interaction

[J]. Advances in Neural Information Processing Systems, 2022, 35: 5816- 5828

[本文引用: 1]

Choi J, Choi H, Hwang J, et al

Graph neural controlled differential equations for traffic forecasting

[J]. Proceedings of the AAAI Conference on Artificial Intelligence, 2022, 36 (6): 6367- 6374

DOI:10.21203/rs.3.rs-3972614/v1      [本文引用: 1]

Qian W Z, Zhang D L, Zhao Y, et al. Uncertainty quantification for traffic forecasting: a unified approach [C]//2023 IEEE 39th International Conference on Data Engineering. Anaheim: IEEE, 2023: 992−1004.

[本文引用: 1]

Fu J, Zhou W, Chen Z

Bayesian graph convolutional network for traffic prediction

[J]. Neurocomputing, 2024, 582: 127507

DOI:10.1016/j.neucom.2024.127507      [本文引用: 1]

Hou Y, Zhang D, Li D, et al

Regional traffic flow combination prediction model considering virtual space of the road network

[J]. Physica A: Statistical Mechanics and its Applications, 2024, 637: 129598

DOI:10.1016/j.physa.2024.129598      [本文引用: 1]

Royat A, Moghadas S M, De Cruz L, et al. GINTRIP: interpretable temporal graph regression using information bottleneck and prototype-based method [PP/OL]. arXiv (2025-12-27) [2025-05-27]. https://arxiv.org/abs/2409.10996.

[本文引用: 1]

/

〈 〉