作者投稿和查稿 主编审稿 专家审稿 编委审稿 远程编辑

计算机工程 ›› 2021, Vol. 47 ›› Issue (11): 62-68,76. doi: 10.19678/j.issn.1000-3428.0059064

• 人工智能与模式识别 • 上一篇    下一篇

基于多组件融合与空洞图卷积的车道占用率预测模型

孙秀芳, 李建波, 吕志强, 董传浩   

  1. 青岛大学 计算机科学技术学院, 山东 青岛 266071
  • 收稿日期:2020-07-27 修回日期:2020-11-27 发布日期:2020-12-07
  • 作者简介:孙秀芳(1995-),女,硕士研究生,主研方向为智能交通预测;李建波(通信作者),教授、博士;吕志强、董传浩,硕士研究生。
  • 基金资助:
    国家自然科学基金(61802216);国家重点研发计划重点专项(2018YFB2100303);中国博士后科学基金(2018M642613);山东省高等学校青创科技计划创新团队项目(2020KJN011);山东省博士后创新人才支持计划(40618030001)。

Lane Occupancy Prediction Model Based on Multi-Component Fusion and Dilated Graph Convolution

SUN Xiufang, LI Jianbo, LÜ Zhiqiang, DONG Chuanhao   

  1. College of Computer Science and Technology, Qingdao University, Qingdao, Shandong 266071, China
  • Received:2020-07-27 Revised:2020-11-27 Published:2020-12-07

摘要: 为解决交通拥堵和交通硬件资源分配不足等问题,提出一种基于多组件融合与空洞图卷积的车道占用率预测模型MCFDGCN。针对交通数据的非线性和受多种隐式因素影响的特点,利用图卷积提取交通数据的空间相关性,使用空洞卷积提取时间依赖特征,将车流量和车辆速度作为2个隐式因素引入模型中,对多组件提取的影响车道占用率的多模态特征进行融合,以完成车道占用率预测任务。在PeMS7(O)、PeMS7(4)数据集上进行实验,结果表明,与HA、ARIMA等模型相比,MCFDGCN模型预测误差较低且误差增长较缓慢,能实现更精准的车道占用率预测。

关键词: 智能交通系统, 车道占用率, 交通预测, 图卷积, 空洞卷积

Abstract: In order to solve the problems of traffic congestion and insufficient allocation of traffic hardware resources,a lane occupancy prediction model,MCFDGCN,based on multi-component fusion and dilated graph convolution is proposed.Considering the non-linear feature of traffic data,and that it is influenced by many implicit factors,the model uses graph convolution to extract the spatial correlation of traffic data,and uses dilated convolution to extract time-dependent features.The vehicle flow and vehicle speed are added into the model as two implicit factors,and the multimodal features extracted by multiple components are fused to predict the lane occupancy rate.Experimental results on the datasets of PeMS7(O) and PeMS7(4) show that compared with HA,ARIMA and other models,the MCFDGCN model exhibits a lower average error rate and error growth rate.It can achieve more accurate prediction of the lane occupancy rate.

Key words: Intelligent Traffic System(ITS), lane occupancy, traffic prediction, graph convolution, dilated convolution

中图分类号: