202504论文研读-Learning dynamics of multi-level spatiotemporal graph data for traffic flow prediction

作者:Zhongbing Li , Yuli Wei , Guihui Chen , Kai Lu , Xinyu Zheng

发表期刊:Computer Communications

发表日期:2024年

一、背景

问题重要性
交通拥堵是城市化进程中的主要挑战,准确的交通流量预测可优化路网效率,支撑智能交通系统(如路径规划、交通诱导)。
现有方法不足
静态图结构:传统图卷积网络(GCN)依赖预定义邻接矩阵(如地理距离),无法捕捉动态空间相关性(如突发事故、天气变化)。
误差累积:深度模型在长时预测中易因逐层信息传播导致误差累积。
多尺度特征缺失:现有方法未充分融合全局稳定性(长期周期)与局部动态性(短时变化)。

二、创新点

与传统的图卷积网络(GCN)方法相比,MS-GSTGCN具有以下优势:
1、动态邻接矩阵:传统GCN方法使用固定的邻接矩阵来表示节点之间的关系,而MS-GSTGCN通过自学习的邻接矩阵来动态调整这些关系,适应交通流的时变性。
2、多尺度特征提取:MS-GSTGCN能够从多个尺度上捕捉到交通流数据的特征,尤其是在处理复杂交通网络时,能够更好地应对不同类型的变化。
3、时空依赖性建模:模型不仅处理了空间依赖性,还通过时空卷积层有效建模了时间依赖性,特别是针对长时间预测任务,能够提供更准确的结果。

三、MS-GSTGCN模型

MS-GSTGCN模型分为两大核心模块:
自学习交通图结构(G-Conv):动态构建多源邻接矩阵,捕捉复杂空间相关性。
多步交通预测模块:结合时空卷积与门控机制,实现长时精准预测。

输入:三种初始化邻接矩阵(地理、自适应、相似性)。
目标:融合全局稳定性与局部动态性,生成动态更新的自学习邻接矩阵。
1、地理邻接矩阵(Ag​)
反映物理连接,但仅关注静态拓扑,忽略动态变化。
2、自适应邻接矩阵(Ad)
无监督学习节点间潜在关系,但缺乏先验约束,可能不稳定。
3、相似性邻接矩阵(Av)
随时间窗口滑动更新,反映短时交通模式相似性。
4、自学习邻接矩阵(Am)
结合全局稳定性(地理)与局部动态性(相似性),实时适应交通变化。

输入:历史交通数据 X∈R T×N×M(时间步长×节点数×特征维度)与 AM。
1、扩张因果卷积(时间卷积)
作用:扩大感受野,捕捉长时依赖(如早晚高峰周期)。2、多尺度扩散卷积(空间卷积)
捕捉动态空间相关性。
3、层级间信息传递
作用:通过门控单元调节信号强度,抑制无关特征,减少长时预测误差累积。

四、实验设置

数据集:
METR-LA数据集:包含来自洛杉矶高速公路段的传感器数据,共207个传感器,1515条道路边,数据记录时间为2012年3月1日至6月30日。数据记录的交通速度每5分钟更新一次。
PEMS-BAY数据集:包含来自加利福尼亚州的325个传感器数据,2369条道路边,数据记录时间为2017年1月1日至6月3日,数据记录频率为每30秒,5分钟聚合后计算出平均速度。

评价指标
为了客观评估模型性能,论文使用了三种评价指标:
MAE(平均绝对误差):反映预测值与真实值的绝对差异。
RMSE(均方根误差):衡量预测误差的波动程度。
MAPE(平均绝对百分比误差):以百分比的形式衡量误差相对大小。

五、实验结果

实验结果表明,MS-GSTGCN在所有预测时间段(如15、30、60分钟)中均优于传统模型,特别是在60分钟的长时间预测中表现出色。
与其他深度学习模型(如LSTM、STGCN、DCRNN、Graph WaveNet)相比,MS-GSTGCN能够在较短的时间内提供更精确的预测结果。此外,MS-GSTGCN还显示出比基准模型(如历史平均值、ARIMA)更优的预测精度。

实验结果表明,MS-GSTGCN模型通过数据驱动的自学习邻接矩阵和多尺度卷积机制,相比于其他模型(如Graph WaveNet)具有更好的长期预测能力和更强的适应性,尤其是在动态和复杂的交通环境下。
这使得MS-GSTGCN在面对剧烈波动和复杂多变的交通条件时,能够生成更为准确和可靠的预测结果。

六、总结

MS-GSTGCN模型成功解决了传统方法无法处理的交通流预测问题,特别是通过动态调整邻接矩阵和多尺度卷积机制,捕捉到了交通流数据中的复杂时空依赖性。
实验结果表明,该模型在长时间预测任务中表现优异,并且能够处理复杂的交通网络结构。
该模型的提出对于智能交通系统的发展具有重要意义,能够为道路设计、交通流引导以及旅行路径规划提供有价值的预测支持。、