公开(公告)号：CN112905379A

公开(公告)日：2021-06-04

申请号：CN202110260633.6

申请日：2021-03-10

申请人： 南京理工大学

发明人： 张伟斌 ,  张蒲璘 ,  姜影

IPC分类号： G06F11/07 ,  G06F16/9537 ,  G06K9/62 ,  G06N3/04 ,  G06N3/08

摘要： 本发明公开了一种基于自我注意力机制的图自编码器的交通大数据修复方法，包括：确定需要进行交通数据修复的区域，采集该区域的历史交通数据；构建掩码矩阵，同时基于所选区域的路网结构生成邻接矩阵；基于自我注意力机制和图卷积网络构建数据修复模型；对数据修复模型进行训练；针对需要修复的交通数据，利用训练好的数据修复模型获得数据修复结果。本发明在GCN的基础上引入自编码器、多头注意力机制等结构，利用GCN的结构有效地学习城市路网的拓扑图结构，利用多头注意力机制学习交通流数据的时空关联性，并利用自编码器依据存在缺失的交通数据生成完整的交通数据，且通过多头注意力机制和GCN可以有效提高模型数据修复的准确性。

公开(公告)号：CN112905379B

公开(公告)日：2023-07-18

申请号：CN202110260633.6

申请日：2021-03-10

申请人： 南京理工大学

发明人： 张伟斌 ,  张蒲璘 ,  姜影

IPC分类号： G06F11/07 ,  G06F16/9537 ,  G06F18/214 ,  G06N3/0464 ,  G06N3/08

摘要： 本发明公开了一种基于自我注意力机制的图自编码器的交通大数据修复方法，包括：确定需要进行交通数据修复的区域，采集该区域的历史交通数据；构建掩码矩阵，同时基于所选区域的路网结构生成邻接矩阵；基于自我注意力机制和图卷积网络构建数据修复模型；对数据修复模型进行训练；针对需要修复的交通数据，利用训练好的数据修复模型获得数据修复结果。本发明在GCN的基础上引入自编码器、多头注意力机制等结构，利用GCN的结构有效地学习城市路网的拓扑图结构，利用多头注意力机制学习交通流数据的时空关联性，并利用自编码器依据存在缺失的交通数据生成完整的交通数据，且通过多头注意力机制和GCN可以有效提高模型数据修复的准确性。

公开(公告)号：CN112185104B

公开(公告)日：2021-12-10

申请号：CN202010855606.9

申请日：2020-08-22

申请人： 南京理工大学

发明人： 张伟斌 ,  张蒲璘 ,  余英豪

IPC分类号： G08G1/01 ,  G06N3/04 ,  G06N3/08

摘要： 本发明公开了一种基于对抗自编码器的交通大数据修复方法，包括：确定需要进行交通数据修复的路段，采集该路段的历史交通数据；基于历史交通数据，构建掩码矩阵；构建对抗神经网络，包括：构建结合注意力机制的自编码器模型作为生成器，结合注意力机制构建判别器；基于历史交通数据和掩码矩阵，训练对抗神经网络，生成数据修复模型；利用数据修复模型修复所述路段实时采集的交通数据。本发明在GAN的基础上引入自编码器、多头注意力机制，以及线索矩阵等结构，利用对抗神经网络的结构有效地学习交通大数据的分布特征，并利用自编码器依据存在缺失的交通数据生成完整的交通数据，且通过多头注意力机制和线索矩阵可以有效提高模型数据修复的准确性。

公开(公告)号：CN112185104A

公开(公告)日：2021-01-05

申请号：CN202010855606.9

申请日：2020-08-22

申请人： 南京理工大学

发明人： 张伟斌 ,  张蒲璘 ,  余英豪

IPC分类号： G08G1/01 ,  G06N3/04 ,  G06N3/08

摘要： 本发明公开了一种基于对抗自编码器的交通大数据修复方法，包括：确定需要进行交通数据修复的路段，采集该路段的历史交通数据；基于历史交通数据，构建掩码矩阵；构建对抗神经网络，包括：构建结合注意力机制的自编码器模型作为生成器，结合注意力机制构建判别器；基于历史交通数据和掩码矩阵，训练对抗神经网络，生成数据修复模型；利用数据修复模型修复所述路段实时采集的交通数据。本发明在GAN的基础上引入自编码器、多头注意力机制，以及线索矩阵等结构，利用对抗神经网络的结构有效地学习交通大数据的分布特征，并利用自编码器依据存在缺失的交通数据生成完整的交通数据，且通过多头注意力机制和线索矩阵可以有效提高模型数据修复的准确性。