基于卷积神经网络与门控循环单元的交通流预测模型

作者：王博文; 王景升^*; 王统一; 夏天雨; 赵丹婷

来源：重庆大学学报, 2023, 46(08): 132-140.

• 交通流预测
• CNN
• GRU
• 编码器-解码器
• 多步预测

为对交通流进行多步预测，支持智能交通系统的长期决策任务，一种基于编码器-解码器（encoder-decoder,ED）的卷积神经网络(convolutional neural networks,CNN)-门循环单元(gate recurrent unit, GRU)模型，简称ED CNN-GRU。首先使用CNN作为编码器，对交通流序列进行信息捕捉，再将上述信息通过GRU解码器进行解释并输出。实验证明，对比CNN、GRU单个模型，ED框架有效解决了误差的迅速累积问题。对比其他基准模型，CNN、 GRU模型对于交通流序列的特征提取及解释能力较为优秀。对于未来12个步长的交通流量预测任务，对比其他基准模型，单因素输入情况的ED CNN-GRU模型的均方根误差下降约0.344～6.464，平均绝对误差下降约0.192～0.425。对比单因素输入，多因素输入下ED CNN-GRU模型拥有更好的预测能力。证明了ED CNN-GRU模型在不同输入维度的多步交通流预测中任务中均具有良好的预测能力，为数据获取条件不同的城市提供了一个支持单因素及多因素输入情况的多步交通流预测模型。

• 单位
  山东科技大学; 中国人民公安大学