基于时空图卷积循环神经网络的交通流预测
CSTR:
作者:
作者单位:

1. 重庆大学 自动化学院,重庆 400044;2. 重庆城市管理职业学院 商学院,重庆 401331

作者简介:

通讯作者:

E-mail: gzy@cqu.edu.cn.

中图分类号:

TP273

基金项目:

国家自然科学基金项目(62073049);重庆市教委科学技术研究项目(KJQN202003303).


Traffic flow prediction based on STG-CRNN
Author:
Affiliation:

1. School of Automation,Chongqing University,Chongqing 400044 ,China;2. School of Business Administration,Chongqing City Management College,Chongqing 401331,China

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    针对交通流预测模型中路网空间结构刻画和交通流时空特性挖掘不充分的问题,构建一种新型的有向时空图,通过定义节点相对临近度来表征路网结构关系,通过学习邻域节点对预测节点的影响权重来表征节点间时空维度的作用关系,从而能更好表达交通流的时空特性.将时空图作为预测模型的输入,采用图卷积获取交通流数据空间依赖关系,采用门控循环神经网络获取交通流数据的时空依赖关系,建立一种基于时空图卷积循环神经网络的交通流预测模型(STG-CRNN).在美国公路交通数据集上对模型预测效果进行验证,其结果表明:STG-CRNN模型的预测结果在平均绝对误差、均方根误差和平均绝对百分误差方面,均优于自回归移动平均模型、门控循环单元模型,以及扩散卷积循环神经网络模型.

    Abstract:

    Aiming at the problem of insufficient characterization of road network spatial structure and inadequate mining of traffic flow spatiotemporal characteristics in the traffic flow prediction model, a new type of directed spatiotemporal graph is constructed. In which the spatial relationship between nodes are characterized by defining the relative proximity, and the spatiotemporal relationship between nodes are characterized by learning the influence weights of neighborhood nodes on the prediction node, so as to better express the temporal and spatial characteristics of traffic flow. Taking spatiotemporal graphs as the input of the prediction model, graph convolution is used to obtain the spatial dependence of traffic flow data, and the gated recurrent neural network is used to obtain the spatiotemporal dependence of traffic flow data, and a traffic flow based on the spatiotemporal graph convolution recurrent neural network(STG-CRNN) is established. The model prediction effect is verified on the U.S. highway traffic data set, and the results show that the STG-CRNN model is better than the autoregressive moving average model, the gated recurrent unit model, and the diffusion convolutional recurrent neural network model in terms of the mean absolute error, root mean square error, and mean absolute percentage error.

    参考文献
    相似文献
    引证文献
引用本文

谷振宇,陈聪,郑家佳,等.基于时空图卷积循环神经网络的交通流预测[J].控制与决策,2022,37(3):645-653

复制
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2022-01-25
  • 出版日期: 2022-03-20
文章二维码