信息技术2024,Issue(10) :30-36.DOI:10.13274/j.cnki.hdzj.2024.10.005

基于时空图卷积网络的高速路交通流多步预测

Traffic flow multi-step prediction of expressway based on Space-Time Graph Convolutional Network

高铭 梅朵
信息技术2024,Issue(10) :30-36.DOI:10.13274/j.cnki.hdzj.2024.10.005

基于时空图卷积网络的高速路交通流多步预测

Traffic flow multi-step prediction of expressway based on Space-Time Graph Convolutional Network

高铭 1梅朵1
扫码查看

作者信息

  • 1. 渤海大学信息科学与技术学院,辽宁锦州 121000
  • 折叠

摘要

针对图卷积网络容易限制模型在交通流预测上有效的学习时空依赖问题,文中提出了一种时空图卷积循环网络(ST-GCRN)模型,首先通过时间卷积层以消除冗余的时间信息,其次,将图卷积网络与改进的门控循环网络相结合以获取时空依赖,最后通过加入残差的编解码结构解决模型训练梯度消失等问题,从而提高预测准确率,实现多步预测.在加利福尼亚州高速公路数据集上进行了实验,结果表明,该模型的平均绝对误差与均方根误差对比基准模型分别减少了 11%、7.5%.

Abstract

To address the problem that graph convolutional network is easy to restrict the model's effective learning of space-time dependence in traffic flow prediction,this paper proposes a Space-Time Graph Con-volution Recurrent Network(ST-GCRN)model.Firstly,the time convolution layer is used to eliminate re-dundant time information.Secondly,the graph convolutional network is combined with the improved gated recurrent network to obtain the space-time dependence.Finally,the residual encoding and decoding struc-ture is added to solve the problem that the model training gradient disappears,thus improving the prediction accuracy and realize multi-step prediction.The experiment results on the California highway dataset show that the mean absolute error and the root mean square error of the model compared with the benchmark mod-el are reduced by 11%and 7.5%,respectively.

关键词

交通流预测/图卷积网络/时间卷积网络/门控循环网络/编解码结构

Key words

traffic flow prediction/Graph Convolutional Network/Space-Time Convolutional Network/Ga-ted Recurrent Unit/encoder-decoder

引用本文复制引用

基金项目

国家自然科学基金项目(62172057)

科技部国家科技重大专项项目(2019YFD0901605)

出版年

2024
信息技术
黑龙江省信息技术学会 中国电子信息产业发展研究院 中国信息产业部电子信息中心

信息技术

CSTPCD
影响因子:0.413
ISSN:1009-2552
段落导航相关论文