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

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

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

traffic flow predictionGraph Convolutional NetworkSpace-Time Convolutional NetworkGa-ted Recurrent Unitencoder-decoder

高铭、梅朵

展开 >

渤海大学信息科学与技术学院,辽宁锦州 121000

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

国家自然科学基金项目科技部国家科技重大专项项目

621720572019YFD0901605

2024

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

信息技术

CSTPCD
影响因子:0.413
ISSN:1009-2552
年,卷(期):2024.(10)