公开(公告)号：CN118115818A

公开(公告)日：2024-05-31

申请号：CN202410392054.0

申请日：2024-04-02

Applicant: 杭州市滨江区浙工大网络空间安全创新研究院

Inventor： 翔云 ,  周俊 ,  宣琦

IPC: G06V10/764 ,  G06V10/82 ,  G06N3/0464 ,  G06N3/08

Abstract: 本发明公开了一种基于注意力机制的神经网络故障修复方法、图像分类模型、图像分类方法，涉及深度神经网络修复技术领域。包括：利用修复数据集，计算故障数据和正常数据在神经网络不同层之间的神经元激活差异；根据余弦相似度选择神经元激活差异最大的层，插入注意力模块；通过修复数据集，训练注意力模块，从而在中间层过滤错误信息，进而修复神经网络。本发明通过在模型中插入注意力模块，利用注意力机制，过滤导致故障的样本信息，从而修复模型。

公开(公告)号：CN117115536A

公开(公告)日：2023-11-24

申请号：CN202311084659.5

申请日：2023-08-28

Applicant: 杭州市滨江区浙工大网络空间安全创新研究院

Inventor： 翔云 ,  周俊 ,  陈作辉 ,  宣琦

IPC: G06V10/764 ,  G06V10/82 ,  G06N3/094 ,  G06N3/0475 ,  G06N3/084

Abstract: 本发明公开了一种基于对抗训练神经网络图像分类方法、装置及可存储介质，涉及图像处理技术领域，其中方法包括以下步骤：获取待分类图像；构建待修复分类神经网络以及探针网络，利用所述探针网络对所述待修复分类神经网络进行对抗训练修复；利用经过对抗训练修复的所述待修复分类神经网络对所述待分类图像进行分类，得到分类结果；本发明通过在模型中间层插入探针网络的方式获取模型中间层的信息，学习两个领域之间的差异知识，并通过对抗攻击的方法对输入数据进行对抗增强，结合对抗训练的方法，对模型进行迭代修复，进一步提升模型故障修复效果，从而提高图像分类的精度。