分享自:

基于双动态时空图卷积网络的交通预测

期刊:IEEE Transactions on Intelligent Transportation SystemsDOI:10.1109/TITS.2022.3208943

这篇文档属于类型a,即报告了一项原创性研究。以下是基于文档内容生成的学术报告:

作者及发表信息

本文的主要作者包括Yanfeng Sun、Xiangheng Jiang、Yongli Hu、Fuqing Duan、Kan Guo、Boyue Wang、Junbin Gao和Baocai Yin。他们分别来自北京工业大学多媒体与智能软件技术北京市重点实验室、北京师范大学人工智能学院、悉尼大学商学院等机构。该研究于2022年12月发表在《IEEE Transactions on Intelligent Transportation Systems》期刊上,题为“Dual Dynamic Spatial-Temporal Graph Convolution Network for Traffic Prediction”。

学术背景

该研究属于智能交通系统(Intelligent Transportation Systems, ITS)领域,重点关注交通预测问题。随着城市人口和车辆的快速增长,交通系统变得日益复杂,交通拥堵等问题愈发严重。因此,构建智能交通系统成为现代城市发展的迫切需求。交通预测作为ITS中的核心任务之一,旨在通过建模交通数据的动态时空特性,预测未来的交通状况。

传统的交通预测方法,如历史平均法(HA)、移动平均法(MA)、自回归积分移动平均模型(ARIMA)等,虽然简单易用,但预测精度较低。近年来,基于神经网络的方法,尤其是图卷积网络(Graph Convolution Network, GCN)和时序卷积网络(Temporal Convolution Network, TCN),因其在捕捉交通数据时空特性方面的优势,取得了显著的性能提升。然而,现有方法主要关注交通流图中的节点(即路段)特征,而对边的动态时空特性建模不足。为此,本文提出了一种双动态时空图卷积网络(Dual Dynamic Spatial-Temporal Graph Convolution Network, DDSTGCN),旨在同时建模交通流图中节点和边的动态特性,从而提升交通预测的精度。

研究流程

  1. 问题定义与模型框架
    研究首先将交通路网定义为一个加权有向图,其中节点表示路段,边表示路段之间的交通流关系。交通预测任务被形式化为:给定历史交通数据和交通流图,预测未来一段时间内的交通状况。DDSTGCN的核心思想是将交通流图转换为其对偶超图(dual hypergraph),从而捕捉边的动态特性。模型框架包括输入层、多个堆叠的双时空块(DST-Blocks)和输出层。

  2. 双时空块(DST-Blocks)
    DST-Blocks是DDSTGCN的核心组件,用于动态捕捉交通流图中节点和边的时空特征。每个DST-Block包含以下模块:

    • 对偶变换(Dual Transformation):将交通流图映射到其对偶超图,并反向映射。超图的节点特征由交通流图的边特征定义,反之亦然。
    • 动态交互模块(Dynamic Interaction Module, DIM):通过图卷积网络(GCN)和超图卷积网络(HGCN)生成动态边和超边,用于后续的图卷积操作。
    • 动态图卷积(Dynamic Graph Convolution):采用门控时序卷积网络(Gate-TCN)和动态图卷积网络(DGCN)捕捉交通数据的时空特征。
    • 动态超图卷积(Dynamic Hypergraph Convolution):采用类似的卷积操作捕捉超图的时空特征。
  3. 输入层与输出层
    输入层通过线性变换将交通数据映射到高维空间,以提升模型的表达能力。输出层通过跳跃连接(skip-connection)融合所有DST-Blocks的输出,并通过Leaky ReLU和线性变换层生成最终的交通预测结果。

  4. 实验设置
    研究在六个真实交通数据集(METR-LA、PEMS-BAY、PEMS03、PEMS04、PEMS07、PEMS08)上进行了广泛的交通预测实验。实验采用均方误差(MAE)、均方根误差(RMSE)和平均绝对百分比误差(MAPE)作为评价指标。模型参数包括输入时间序列长度、DST-Blocks堆叠层数、中间滤波器数量等。

主要结果

  1. 交通预测性能
    DDSTGCN在所有数据集上均表现出色,显著优于现有的交通预测方法。在交通速度数据集(METR-LA、PEMS-BAY)上,DDSTGCN在15分钟、30分钟和60分钟预测任务中的MAE、RMSE和MAPE指标均优于其他方法。在交通流量数据集(PEMS03、PEMS04、PEMS07、PEMS08)上,DDSTGCN的性能提升更为显著,平均MAE、RMSE和MAPE分别比第二名提高了9.78%、7.80%和5.96%。

  2. 消融实验
    消融实验验证了DDSTGCN各模块的有效性。实验表明,同时使用GCN和HGCN的模型性能优于仅使用其中一种的模型,动态机制和DIM模块的引入进一步提升了模型的性能。

  3. 参数敏感性分析
    研究分析了关键参数(如对偶变换中的top-k边数、DST-Blocks堆叠层数、中间滤波器数量)对模型性能的影响。结果表明,适当的参数设置能够显著提升模型的预测精度。

结论

DDSTGCN通过同时建模交通流图中节点和边的动态时空特性,显著提升了交通预测的精度。其核心贡献包括:
1. 提出了一种新颖的双动态时空图卷积网络框架,通过GCN和HGCN的协同卷积捕捉交通数据的复杂时空特性。
2. 利用对偶超图建模边的动态相关性,揭示了交通系统中潜在的高阶依赖关系。
3. 在多个真实交通数据集上验证了模型的有效性和泛化能力。

研究亮点

  1. 创新性方法:DDSTGCN首次将交通流图转换为对偶超图,并通过动态图卷积和超图卷积捕捉节点和边的动态特性。
  2. 广泛验证:研究在六个真实数据集上进行了全面实验,证明了模型在不同类型交通数据上的优越性能。
  3. 实际应用价值:DDSTGCN为智能交通系统的实时交通预测提供了高效、精确的解决方案,具有广泛的应用前景。

其他有价值内容

研究还提供了源代码(https://github.com/j1o2h3n/ddstgcn),便于其他研究者复现和改进模型。此外,研究对动态机制的直观展示(如动态边和超边的标准差可视化)进一步验证了模型的有效性。

上述解读依据用户上传的学术文献,如有不准确或可能侵权之处请联系本站站长:admin@fmread.com