一种基于注意力机制的开集对抗样本防御方法

AN OPEN-SET ADVERSARIAL SAMPLE DEFENSE METHOD BASED ON ATTENTION MECHANISM

  • 摘要: 针对开集对抗样本防御成功率不高的问题,提出一种基于注意力机制的自编码器。将基于高斯平均的非局部注意力和双注意力组合并嵌入在加深卷积神经网络的自编码器ResNet网络中,提高开集对抗样本防御成功率。实验结果表明,所提出的方法与OSAD方法相比,开集对抗样本防御成功率得到提高。在CIFAR10数据集和SVHN数据集上,在FGSM攻击下,AUC-ROC分别提高了2.3百分点和13.4百分点,闭集识别准确率分别提高了4.1百分点和0.1百分点;在PGD攻击下,AUC-ROC分别提高了1.0百分点和2.8百分点,闭集识别准确率在SVHN数据集上提高了3.9百分点。

     

    Abstract: As for the low success rate of open-set adversary sample defense, an auto-encoder based on attention mechanism is designed. The non-local attention based on Gaussian mean and dual-attention were combined and embedded in the auto-encoder ResNet network that deepened the convolutional neural network to improve the success rate of open set adversary sample defense. The experimental results show that compared with OSAD, the proposed method get better success rate on open-set adversarial sample. On CIFAR10 dataset and SVHN dataset, under FGSM attack, AUC-ROC is improved by 2.3 and 13.4 percentage points respectively and the accuracy of closed-set recognition is improved by 4.1 and 0.1 percentage points respectively; under PGD attack, AUC-ROC is improved by 1.0 and 2.8 percentage points respectively, and the accuracy of closed-set recognition is improved by 3.9 percentage points on SVHN dataset.

     

/

返回文章
返回