基于时空卷积神经网络的数据缺失交通流预测  

Traffic Flow Prediction with Missing Data Based on Spatial-Temporal Convolutional Neural Networks

作　　者：张壮壮 屈立成[1] 李翔[1] 张明皓 李昭璐 ZHANG Zhuangzhuang;QU Licheng;LI Xiang;ZHANG Minghao;LI Zhaolu(School of Information Engineering,Chang’an University,Xi’an 710064,China)

机构地区：[1]长安大学信息工程学院,西安710064

出　　处：《计算机工程与应用》2022年第7期259-265,共7页Computer Engineering and Applications

基　　金：国家重点研发计划项目(2019YFE0108300);国家自然科学基金(62001058);陕西省自然科学基础研究计划项目(2020JM-258);长安大学2021年研究生科研创新实践项目(300103714043)。

摘　　要：针对数据连续缺失情况下交通流预测精度下降甚至失效的问题,提出了一种时空卷积神经网络预测模型,根据横向分布的时间相关性和纵向分布的空间相关性,构建路网交通数据时空矩阵,引入掩码矩阵标记数据的缺失状况,利用卷积操作提取路网中各检测器之间隐含的非线性关联,建立当前时刻与未来交通状态的映射关系,实现数据缺失情况下的交通流预测。使用公开数据集,在3个时间尺度上的验证结果表明,所提出的模型在平均误差和预测精度两个方面均优于长短期记忆网络、门控循环单元、扩散卷积神经网络和图马尔可夫网络模型,在交通数据随机缺失和连续缺失两种情况下,均表现出了良好的稳定性和健壮性。In order to improve the accuracy of traffic flow prediction under the condition of continuous missing data,a spatial-temporal convolutional neural network prediction algorithm is proposed.The spatial-temporal matrix of road network traffic data is established by considering the spatial adjacent relationship of horizontal distribution and the time dependence of vertical distribution.A mask matrix is introduced to represent the missing situation of data.The convolution operation is used to mine the implicit nonlinear correlation between the upstream and downstream detectors,establish the mapping relationship between the current and future traffic states,and realize the traffic flow prediction under the condition of continuous data absence.The experimental results show that the accuracy of traffic flow prediction is better than those of LSTM,GRU and GMN models in the absence of continuous traffic data.The model shows good stability and robustness,and improves the accuracy of traffic flow prediction in the absence of data.

关 键 词：智能交通系统 交通流预测 深度学习 时空卷积神经网络 连续数据缺失 

分 类 号：TP183[自动化与计算机技术—控制理论与控制工程]