[发明专利]多目标类别对抗样本生成方法及相关设备

权利要求书

1.一种多目标类别对抗样本生成方法，其特征在于，包括：

获取原始图像和目标类别，将所述原始图像和所述目标类别输入预先训练好的MTA模型，MTA模型包括编码器、嵌入层、融合层及解码器；

将所述原始图像输入编码器，编码器通过三个卷积层得到三维特征图；

将目标类别的one-hot标签输入嵌入层，嵌入层通过全连接将标签映射成特征向量，然后将所述特征向量平铺得到三维特征向量；

将所述三维特征图和所述三维特征向量输入融合层，在通道层进行拼接得到新特征图，再通过四个残差卷积模块对所述新特征图处理得到解析特征图；

将所述解析特征图输入解码器，通过三个反卷积得到与所述原始图像的大小和通道数相同的噪声；

将所述噪声和所述原始图像叠加得到所述目标类别的对抗样本。

2.根据权利要求1所述的方法，其特征在于，所述将所述噪声和所述原始图像叠加得到所述目标类别的对抗样本，包括：

对所述噪声进行剪裁得到剪裁噪声；

将所述剪裁噪声叠加所述原始图像得到初始对抗样本；

对所述初始对抗样本进行剪裁得到所述对抗样本。

3.根据权利要求1所述的方法，其特征在于，还包括：对所述对抗样本和所述原始图像进行鉴别，生成图像置信度。

4.根据权利要求1所述的方法，其特征在于，还包括：对所述对抗样本进行目标分类，生成所有所述目标类别的回归值并计算损失。

5.根据权利要求4所述的方法，其特征在于，所述损失包括对抗损失和扰动损失。

6.一种多目标类别对抗样本生成装置，其特征在于，包括：

特征提取模块，被配置为获取原始图像和目标类别，将所述原始图像和所述目标类别输入预先训练好的MTA模型；MTA模型包括编码器、嵌入层、融合层及解码器；

将所述原始图像输入编码器，编码器被配置为通过三个卷积层得到三维特征图；

将目标类别的one-hot标签输入嵌入层，嵌入层被配置为通过全连接将标签映射成特征向量，然后将所述特征向量平铺得到三维特征向量；

将所述三维特征图和所述三维特征向量输入融合层，融合层被配置为在通道层进行拼接得到新特征图，再通过四个残差卷积模块对所述新特征图处理得到解析特征图；

将所述解析特征图输入解码器，解码器被配置为通过三个反卷积得到与所述原始图像的大小和通道数相同的噪声；

样本生成模块，被配置为将所述噪声和所述原始图像叠加得到所述目标类别的对抗样本。

7.一种电子设备，包括存储器、处理器及存储在存储器上并可在处理器上运行的计算机程序，所述处理器执行所述程序时实现如权利要求1至5任意一项所述的方法。