交通科技2024,Issue(6) :126-131.DOI:10.3963/j.issn.1671-7570.2024.06.024

基于门控时空图网络和TCN的交通流预测方法

Traffic Flow Forecasting Method Based on Gated Spatial-temporal Spatiotemporal Graph Network and TCN

黄河 谢军义 李志晖 孙霞 彭挺
交通科技2024,Issue(6) :126-131.DOI:10.3963/j.issn.1671-7570.2024.06.024

基于门控时空图网络和TCN的交通流预测方法

Traffic Flow Forecasting Method Based on Gated Spatial-temporal Spatiotemporal Graph Network and TCN

黄河 1谢军义 1李志晖 1孙霞 2彭挺3
扫码查看

作者信息

  • 1. 中铁七局集团第三工程有限公司 西安 710032
  • 2. 西北大学信息科学与技术学院 西安 710127
  • 3. 长安大学特殊地区公路工程教育部重点实验室 西安 710064
  • 折叠

摘要

当前交通流预测研究普遍采用GCN学习空间图结构,但缺乏保留图中重要节点特征的能力,且忽略时间序列之间的长距离依赖关系.针对上述问题,提出一种结合门控时空图神经网络和TCN的交通流预测方法.首先该模型采用门控图神经网络GGNN学习空间图结构并保留关键节点特征信息,然后利用TCN捕获时间序列之间的长距离依赖关系.在PeMSD04和PeMSD08 2种公开的交通流数据集上进行对比实验、消融实验和超参数实验.实验表明,GGNN-TCN模型在MAE、RMSE和MAPE 3种指标上的整体性能明显优于基线模型,消融实验结果验证GGNN和TCN组件有利于提升模型整体性能,参数实验表明当GGNN层数为2模型整体性能最优.

Abstract

Current research on traffic flow prediction commonly employs graph convolutional networks(GCNs)to learn spatial graph structure.However,these approaches often lack the ability to retain important node features within the graph and overlook long-distance dependencies between time se-ries.To address these issues,a traffic flow prediction method that combines gated spatiotemporal graph neural network and temporal convolutional networks(TCNs)was proposed.First,the gated graph neural network(GGNN)was used in the model to learn the spatial graph structure while pre-serving key node feature information.Then,TCN was employed to capture long-distance dependen-cies between time series.Finally,comparison study,ablation study,and hyperparameter study were conducted on the publicly available PeMSD04 and PeMSD08 traffic flow datasets.Experimental re-sults show that the GGNN-TCN model significantly outperforms baseline models in terms of MAE,RMSE,and MAPE.The ablation study confirms that both the GGNN and TCN components contrib-ute positively to the overall model performance,while parameter study indicate that the model achieves optimal performance when the number of GGNN layers is set to 2.

关键词

空间图结构/长距离依赖/交通流预测/时间序列

Key words

spatial graph structure/long-distance dependency/traffic flow prediction/time series

引用本文复制引用

出版年

2024
交通科技
武汉理工大学

交通科技

影响因子:0.495
ISSN:1671-7570
段落导航相关论文