原文传递
基于自注意力机制与图自编码器的路网交通流数据修复模型
| 题名： | 基于自注意力机制与图自编码器的路网交通流数据修复模型 |
| 作者： | 张伟斌;张蒲璘;苏子毅;孙锋 |
| 作者单位： | 南京理工大学;山东理工大学 |
| 关键词： | 智能交通;交通流数据修复;图卷积网络;城市路网交通流数据;自注意力机制 |
| 摘要： | 针对城市交通流数据修复问题，提出一种基于图卷积网络和多头自注意力机制的自注意力图自编码器模型。该模型包括基于拓扑图结构和图信号捕获交通流时空关联性的STGCN(Spatial-temporal Grap hConvolutional Networks)网络。在该网络中使用LSTM(Long Short-Term Memory)网络学习数据中时序规律,通过注意力网络计算道路自注意力及一阶临近道路注意力系数,用图卷积网络对图信号重组,达到对缺失数据的精确修复。同时,采用多头自注意力网络计算数据的注意力权值并对数据重组,捕获交通流数据中的二阶及高阶临近道路空间关联性,提取已知数据与缺失数据的时间关系,以残差链的形式加入到模型中,作为对STGCN功能的补充。基于真实数据的实验表明,在多种缺失模式和缺失率下,该模型能够学习路网拓扑关系,捕获数据中的时间规律性和时空关联性,有效地修复缺失数据。 |
| 期刊名称： | 交通运输系统工程与信息 |
| 出版日期： | 202104 |
| 出版年： | 2021 |
| 期： | 04 |
| 页码： | 90-98 |
检索历史
应用推荐
