基于剪枝技术和鲁棒蒸馏融合的轻量对抗攻击防御方法

时间:2023-06-17 11:35:02 公文范文 来源:网友投稿

王滨,李思敏,钱亚冠,张君,李超豪,朱晨鸣,张鸿飞

基于剪枝技术和鲁棒蒸馏融合的轻量对抗攻击防御方法

王滨1,2,李思敏1,钱亚冠1,张君3,李超豪2,朱晨鸣3,张鸿飞3

(1. 浙江科技学院,浙江 杭州 310023;
2. 浙江省多维感知技术应用与安全重点实验室,浙江 杭州 310052;
3. 浙江省电子信息产品检验研究院,浙江 杭州 310007)

对抗训练是一类常用的对抗攻击防御方法,其通过将对抗样本纳入训练过程,从而有效抵御对抗攻击。然而,对抗训练模型的鲁棒性通常依赖于网络容量的提升,即对抗训练所获得的网络为防御对抗攻击而大幅提升网络的模型容量,对其可用性造成较大约束。因此,如何在保证对抗训练模型鲁棒性的同时,降低模型容量,提出轻量对抗攻击防御方法是一大挑战。为解决以上问题,提出一种基于剪枝技术和鲁棒蒸馏融合的轻量对抗攻击防御方法。该方法以对抗鲁棒准确率为优化条件,在对预训练的鲁棒对抗模型进行分层自适应剪枝压缩的基础上,再对剪枝后的网络进行基于数据过滤的鲁棒蒸馏,实现鲁棒对抗训练模型的有效压缩,降低其模型容量。在CIFAR-10和CIFAR-100数据集上对所提出的方法进行性能验证与对比实验,实验结果表明,在相同TRADES对抗训练下,所提出的分层自适应剪枝技术相较于现有剪枝技术,其剪枝所得到的网络结构在多种FLOPs下均表现出更强的鲁棒性。此外,基于剪枝技术和鲁棒蒸馏融合的轻量对抗攻击防御方法相较于其他鲁棒蒸馏方法表现出更高的对抗鲁棒准确率。因此,实验结果证明所提方法在降低对抗训练模型容量的同时,相较于现有方法具有更强的鲁棒性,提升了对抗训练模型在物联网边缘计算环境的适用性。

对抗防御;
剪枝;
鲁棒蒸馏;
轻量网络

随着深度学习技术在图像识别[1]、语音识别[2]、自然语言处理[3]等领域的广泛应用,深度学习模型的安全直接决定了其所应用的系统或业务的安全。然而,研究指出了对抗样本攻击[4,5]对于现有深度学习模型的攻击可行性,即通过在干净样本上添加微小的扰动,造成深度学习模型的功能错误。例如,CW(Carlini and Wagner)攻击可通过在图片上添加人眼不易察觉的对抗扰动,造成智能图像识别模型的分类错误[5],如将停车标志识别为其他标志,可能会造成严重的安全事故[6];
同理,攻击者可通过在音频上叠加微小噪声,构造语音对抗样本,造成智能语音识别系统的错误识别,如将人耳所听的“你好”识别为“打开房门”等敏感指令[7]。

为抵御来自对抗样本的安全威胁,对抗训练[8-10]方法被提出并得到大量研究。对抗训练的核心是在模型训练阶段同时考虑干净样本和对抗样本,从而使训练获得的模型对对抗样本攻击具有较高的鲁棒性。然而,此类方法在提升训练模型的鲁棒性同时,通常会造成模型容量的大幅提升。此外,相关研究表明,模型容量与对抗训练效果成正相关性,即模型容量越大,对抗训练效果越好[9]。然而,该特性对对抗训练的可用性造成较大的约束,如限制对抗训练在计算及存储资源相对受限的边缘设备(如智能手机、智能摄像头等)上的应用部署。因此,亟须设计一种轻量对抗攻击防御方法,尤其在压缩对抗训练模型容量的同时,对模型的鲁棒性影响较小,从而提供轻量级的鲁棒模型,适应物联网边缘计算环境下的智能应用。

针对以上问题,研究者提出基于知识蒸馏(knowledge distillation)[11]对抗训练模型优化方法,即将鲁棒知识从网络容量较大的教师网络迁移到轻量级的学生网络[12-14]。例如,Goldblum等[13]提出ARD(adversarially robust distillation)方法,将对抗样本参与模型的蒸馏,从而进一步提高模型的鲁棒性。在此基础上,Zi等[14]提出了RSLAD(robust soft label adversarial distillation)方法,优化了对抗蒸馏的损失函数,使模型的鲁棒性与经典对抗训练方法相比有所提高。但以上方法都是固定教师网络和学生网络的结构,无法进一步压缩网络容量以适应计算资源相对受限的边缘设备。

因此,本文提出一种基于剪枝技术和鲁棒蒸馏融合的轻量对抗攻击防御方法。首先基于分层自适应的剪枝策略对预训练的鲁棒对抗模型进行剪枝压缩,然后对剪枝后的网络进行基于数据过滤的鲁棒蒸馏,实现鲁棒对抗训练模型的有效压缩,降低其模型容量。

本文主要贡献如下。

1) 提出剪枝技术和鲁棒蒸馏融合的轻量对抗攻击防御方法,其在降低对抗训练模型网络容量的同时,对于对抗训练模型的鲁棒性影响较小。

2) 改进现有剪枝技术,提出分层自适应剪枝技术,提升剪枝过后模型的鲁棒性。

3)提出基于数据过滤的鲁棒蒸馏方法,通过对原始干净样本中被错误识别的样本进行过滤,提升鲁棒蒸馏的有效性。

1.1 深度神经网络与对抗攻击

1.2 对抗训练

近年来,对抗训练作为一种常用的对抗样本攻击防御方法得到广泛的研究与应用[15]。Madry等[9]提出标准对抗训练(SAT,standard adversarial training),其将对抗训练建模成双层优化问题,并基于PGD攻击方法生成对抗样本训练模型。除SAT对抗训练框架之外,Zhang等[16]基于鲁棒性和精度的权衡考虑提出了TRADES框架。另外,还有一类工作是利用没有标签的数据进行对抗训练,进一步提升模型的鲁棒性[17-18]。上述方法在ResNet和WideResNet等大容量深度神经网络上均取得了较高的鲁棒性。此外,研究发现对抗训练通常可使容量较大的网络获得更好的鲁棒性[9]。然而,对于部署到边缘环境中的轻量级神经网络,直接对其进行对抗训练所能取得的鲁棒性效果不如较大网络显著。

1.3 知识蒸馏

为兼顾对抗训练模型的鲁棒性与可用性,本文提出一种基于剪枝技术和鲁棒蒸馏融合的轻量对抗攻击防御方法。首先,本文方法基于分层自适应的剪枝策略对预训练的鲁棒对抗模型进行剪枝压缩。然后,对经过剪枝后的网络进行鲁棒蒸馏。区别于现有鲁棒蒸馏机制,本文方法通过对原始干净样本中被错误识别的样本进行过滤,提升鲁棒知识的高效迁移。剪枝技术和鲁棒蒸馏两类技术的有机结合,既可实现鲁棒对抗训练模型的有效压缩,降低其模型容量,又能降低对模型鲁棒性的影响。下面对以上所述方法进行详细介绍。

2.1 分层自适应剪枝

剪枝技术的核心是将深度神经网络中冗余的卷积核或者通道去除。传统剪枝策略一般分为两类:预定义剪枝策略、基于全局阈值的剪枝策略。首先,预定义剪枝策略通常依赖于人工设定的剪枝率,但往往达不到最优的剪枝效果。其次,基于全局阈值的剪枝策略在使用中具有一定的局限性,如该方法会造成层崩塌,即该层的卷积核几乎被剪完,从而使模型性能骤降。针对以上问题,本文提出一种分层自适应的剪枝策略,其创新点在于依据模型各层输出的软标签信息与模型最终的软标签输出进行比较,进而指导各层剪枝率的分配(如图1所示),使网络结构能够适应鲁棒性的需要。

图1 剪枝分配准则

Figure 1 Pruning allocation guidelines

分层自适应剪枝策略的详细过程如下。

在确定好各层的剪枝率之后,剪枝操作将与蒸馏训练过程交替进行。

2.2 基于数据过滤的鲁棒蒸馏

知识蒸馏的核心是将教师网络学到的知识迁移到更小的学生网络中,从而使学生网络的概率输出分布接近教师网络的概率输出分布,其形式可表达为

算法1 鲁棒蒸馏算法

3.1 实验设置

(1)数据集和模型

1)基准数据集:本文实验采用两个基准数据集,分别为CIFAR-10和CIFAR-100。这两个数据集各包含60 000张RGB彩色图片,其中训练集50 000张,测试集10 000张,图片大小为32×32像素。CIFAR-10和CIFAR-100分别包含10类和100类图片数据。

2)测试模型:本文选取了VGG-16[19]和resnet56[20]作为待剪枝的教师网络模型。通过对测试模型进行剪枝和鲁棒蒸馏,以提高模型在较低FLOP情况下的鲁棒性。

(2)实验环境

本文实验所用硬件为RTX 2080Ti GPU,运行环境为Ubuntu 16.04.6 LTS,Pytorch版本为1.4.0,CUDA版本为10.0,cuDNN版本为7.6.0,torchattacks库的版本为3.0.0。

(3)评价指标

1)FLOP:即浮点运算数,用来衡量算法/模型的复杂度。FLOP越小,所需计算量也越小,代表越有利于算法/模型在资源相对受限的边缘设备计算环境下进行应用部署。

2)对抗鲁棒准确率:代表算法/模型对于对抗样本的分类精度。本文使用目前较为常用的鲁棒性基准测试平台AutoAttack[21](下面简称AA测试)对模型的鲁棒性进行评估。

3.2 CIFAR-10实验结果

本节基于CIFAR-10数据集,选取VGG-16和resnet56作为待剪枝的教师网络,且都是经过对抗训练的鲁棒网络。在此基础上,实验分别从对抗训练和鲁棒蒸馏两个角度,对比不同剪枝方法得到的模型结构对鲁棒性的影响。图2给出了模型在不同剪枝方法以及在不同训练方法下的鲁棒性对比。

在对抗训练实验中,本文选择TRADES框架[16]作为对抗训练的方法。从图2(a)可以观察到,本文所提的分层自适应剪枝方法得到的模型在相同TRADES对抗训练条件下,相比于其他剪枝方法,几乎在各个FLOP层次上均表现出更高的鲁棒性(AA测试)。这充分表明分层自适应剪枝得到的模型结构更能适应模型鲁棒性的需要,即模型结构更为合理。为了公平对比,图2和图3所有数据点均是模型在实验训练中取得的最好结果。

在鲁棒蒸馏实验中,为更合理地进行对比,所有测试方案均采用剪枝技术与鲁棒蒸馏融合的形式。测试剪枝技术包括L1-norm、Slimming与CHIP。实验中除本方案外,其余剪枝操作后使用的鲁棒蒸馏方法均为RSLAD鲁棒蒸馏方法[14]。该方法是现有较为先进的开源鲁棒蒸馏方法。首先,图2(b)给出了不同方案在VGG-16教师网络上的性能对比。结果表明本文所提出的基于剪枝技术和鲁棒蒸馏融合的轻量对抗攻击防御方法与其余剪枝与鲁棒融合方法相比,在相同的FLOP层次上具有更高的对抗鲁棒准确率;
在相同的对抗鲁棒准确率上需要更少的FLOP。因此,本文所提方法在综合性能上表现更好,尤其是高剪枝率、低FLOP情况下更为明显。此外,实验将相同FLOP条件下的VGG-11模型和VGG-13模型参与横向对比。结果表明,本文所提方法是一种有效的压缩模型手段,且比预设定模型结构进行训练的效果更好。以上的高性能主要得益于分层自适应剪枝策略与鲁棒蒸馏前的数据过滤预处理方法带来网络结构的优化。

图2 模型在不同剪枝方法以及在不同训练方法下的鲁棒性对比

Figure 2 Comparison of the model robustness under different pruning methods and various training methods on CIFAR-10

图2(c)则给出了不同方案在resnet56教师网络上的性能对比。实验结果表明本文所提方法剪枝得到的结构经过相同的对抗训练后,其性能曲线均位于其他方法上方,即其鲁棒性相较于其他方法在整体上具有优势。此外,所提方法的鲁棒准确率在resnet56与VGG-16两种模型上的趋势存在差异,即图2(c)中存在拐点,该现象主要是源于两种模型的内在结构差异。VGG是一种基于直连结构的模型,各层之间具有不同的语义信息。只要未对VGG模型中间的某些层进行过度剪枝,模型表达能力和性能不会有明显的骤降,且训练后的表现较为稳定。因此,本文所提方法的鲁棒准确率在VGG模型上与模型的FLOP大致呈现正相关的趋势。resnet56则是通过特征融合,采取跳层连接结构。这种内在结构的不同,可能会使剪枝操作优先关注某些语义表达丰富的特定层,并且这些层在剪掉相当一部分卷积核之后,出现过渡态的非最优结构,进而出现层与层之间特征表达及语义信息不匹配的情况,造成模型的表达能力受限,即FLOP处于70~80 MFLOP的低谷点。随着迭代剪枝的进行,该方法会动态地优化剩下的模型结构,使其前后跨通道的特征融合更优,即前后层的结构更匹配、更互相适应。

3.3 CIFAR-100实验结果

图3展示了基于CIFAR-100数据集以及VGG-16模型的实验结果。结果表明,如同CIFAR-10的表现,本文所提方法不论在相同的TRADES对抗训练下,还是对比其他的鲁棒蒸馏方法,综合鲁棒性要表现更好。

另外,尽管Slimming方法是基于全局阈值方法的剪枝,但是由该方法剪枝得到的模型在高剪枝条件下会导致模型鲁棒性降。因此该方法存在较大缺陷,无法有效部署于边缘环境中。

针对物联网边缘环境对轻量对抗攻击防御方法的需求,本文提出一种基于剪枝技术和鲁棒蒸馏融合的轻量对抗攻击防御方法,通过分层自适应剪枝技术与基于数据过滤的鲁棒蒸馏方法,实现鲁棒对抗训练模型的有效压缩,即在降低其模型容量的同时,对模型的鲁棒性影响较小。实验结果证明,本文所提方法不仅能使相同FLOP条件下的轻量级网络经过同等对抗训练更具鲁棒性,并且在相同鲁棒对抗准确率下,使压缩的模型具有更低的FLOP。未来将对剪枝技术和鲁棒蒸馏做进一步的研究拓展,继续提高模型的压缩率同时保持较高的鲁棒性。

图3 模型在不同剪枝方法以及在不同训练方法下的鲁棒性对比

Figure 3 Comparison of the model robustness under different pruning methods and various training methods on CIFAR-100

[1] 郑远攀, 李广阳, 李晔. 深度学习在图像识别中的应用研究综述[J]. 计算机工程与应用, 2019, 55(12): 20-36.

ZHENG Y P, LI G Y, LI Y. Survey of application of deep learning in image recognition [J].Computer Engineering and Applications, 2019,55(12):20-36.

[2] 鱼昆, 张绍阳, 侯佳正, 等. 语音识别及端到端技术现状及展望[J]. 计算机系统应用, 2021, 30(3): 14-23.

YU K, ZHANG S Y, HOU J Z, et al. Survey of speech recognition and end-to-end techniques[J]. Computer Systems & Applications, 2021, 30(3): 14-23.

[3] 王睿怡, 罗森林, 吴舟婷, 等. 深度学习在汉语语义分析的应用与发展趋势[J]. 计算机技术与发展, 2019, 29(9): 110-116.

WANG R Y, LUO S L, WU Z T, et al. Application and development trend of deep learning in Chinese semantic analysis[J]. Computer Technology and Development, 2019, 29(9): 110-116.

[4] SZEGEDY C, ZAREMBA W, SUTSKEVER I, et al. Intriguing properties of neural networks[C]//Proceedings of 2nd International Conference on Learning Representations (ICLR 2014). 2014.

[5] CARLINI N, WAGNER D. Towards evaluating the robustness of neural networks[C]//Proceedings of 2017 IEEE Symposium on Security and Privacy (SP). 2017.

[6] SONG D, EYKHOLT K, EVTIMOV I, et al. Physical adversarial examples for object detectors[C]//12th USENIX Workshop on offensive technologies (WOOT 18). 2018.

[7] CARLINI N, WAGNER D. Audio adversarial examples: targeted attacks on speech-to-text[C]//2018 IEEE Security and Privacy Workshops (SPW). 2018: 1-7.

[8] GOODFELLOW I, SHLENS J, SZEGEDY C. Explaining and harnessing adversarial examples[C]//Proceedings of 3rd International Conference on Learning Representations (ICLR 2015). 2015.

[9] MADRY A, MAKELOV A, SCHMIDT L, et al. Towards deep learning models resistant to adversarial attacks[C]//Proceedings of 6th International Conference on Learning Representations (ICLR 2018). 2018.

[10] GOWAL S, QIN C, UESATO J, et al. Uncovering the limits of adversarial training against norm-bounded adversarial examples[J]. arXiv:2010.03593, 2020.

[11] GOU J, YU B, MAYBANK S J, et al. Knowledge distillation: a survey[J]. International Journal of Computer Vision, 2021, 129(6): 1789-1819.

[12] PAPERNOT N, MCDANIEL P, WU X, et al. Distillation as a defense to adversarial perturbations against deep neural networks[C]//IEEE Symposium on Security and Privacy, 2016: 582-597.

[13] GOLDBLUM M, FOWL L, FEIZI S, et al. Adversarially robust distillation[C]//Proceedings of the AAAI Conference on Artificial Intelligence. 2020: 3996-4003.

[14] ZI B, ZHAO S, MA X, et al. Revisiting adversarial robustness distillation: robust soft labels make student better[C]//Proceedings of 2021 IEEE/CVF International Conference on Computer Vision, 2021:16423-16432.

[15] BAI T, LUO J, ZHAO J, et al. Recent advances in adversarial training for adversarial robustness[C]//Proceedings of the Thirtieth International Joint Conference on Artificial Intelligence, 2021: 4312-4321.

[16] ZHANG H, YU Y, JIAO J, et al. Theoretically principled trade-off between robustness and accuracy[C]//International Conference on Machine Learning. 2019: 7472-7482.

[17] ALAYRAC J B, UESATO J, HUANG P S, et al. Are labels required for improving adversarial robustness[C]//Advances in Neural Information Processing Systems. 2019: 12192-12202.

[18] CARMON Y,RAGHUNATHAN A, SCHMIDT L, et al. Unlabeled data improves adversarial robustness[C]//Advances in Neural Information Processing Systems. 2019: 11190-11201.

[19] SIMONYAN K, ZISSERMAN A. Very deep convolutional networks for large-scale image recognition[J]. arXiv:1409.1556, 2014.

[20] HE K, ZHANG X, REN S, et al. Deep residual learning for image recognition[C]//Proceedings of 2016 IEEE/CVF Conference on Computer Vision and Pattern Recognition. 2016: 770-778.

[21] CROCE F, HEIN M. Reliable evaluation of adversarial robustness with an ensemble of diverse parameter-free attacks[C]//Proceedings of the 37th International Conference on Machine Learning. 2020: 2206-2216.

[22] LI H, KADAV A, DURDANOVIC I, et al. Pruning filters for efficient convnets[C]//Proceedings of 5th International Conference on Learning Representations (ICLR 2017). 2017.

[23] LIU Z, LI J, Shen Z, et al. Learning efficient convolutional networks through network slimming[C]//Proceedings of 2017 IEEE/CVF Conference on Computer Vision and Pattern Recognition. 2017: 2755-2763.

[24] SUI Y, YIN M, XIE Y, et al. CHIP: channel Independence-based pruning for compact neural networks[C]//Advances in Neural Information Processing Systems. 2021: 24604-24616.

Lightweight defense mechanism against adversarial attacks via adaptive pruning and robust distillation

WANG Bin1,2, LI Simin1, QIAN Yaguan1, ZHANG Jun3, LI Chaohao2, ZHU Chenming3, ZHANG Hongfei3

1. Zhejiang University of Science and Technology, Hangzhou 310023, China 2. Zhejiang Key Laboratory of Multi-dimensional Perception Technology, Application and Cybersecurity, Hangzhou 310052, China 3. Zhejiang Electronic Information Products Inspection and Research Institute, Hangzhou 310007, China

Adversarial training is one of the commonly used defense methods against adversarial attacks, by incorporating adversarial samples into the training process. However, the effectiveness of adversarial training heavily relied on the size of the trained model. Specially, the size of trained models generated by the adversarial training will significantly increase for defending against adversarial attacks. This imposes constraints on the usability of adversarial training, especially in a resource-constraint environment. Thus, how to reduce the model size while ensuring the robustness of the trained model is a challenge. To address the above issues, a lightweight defense mechanism was proposed against adversarial attacks, with adaptive pruning and robust distillation. A hierarchically adaptive pruning method was applied to the model generated by adversarial training in advance. Then the trained model was further compressed by a modified robust distillation method. Experimental results on CIFAR-10 and CIFAR-100 datasets showed that our hierarchically adaptive pruning method presented stronger robustness under various FLOP than the existing pruning methods. Moreover, the fusion of pruning and robust distillation presented higher robustness than the state-of-art robust distillation methods. Therefore, the experimental results prove that the proposed method can improve the usability of the adversarial training in the IoT edge computing environment.

adversarial defenses; pruning; robust distillation; lightweight network

TP393

A

10.11959/j.issn.2096−109x.2022074

2022−04−24;

2022−06−11

钱亚冠,qianyg@yeah.net

国家自然科学基金(92167203);
浙江省自然科学基金(LZ22F020007)

The National Natural Science Foundation of China (92167203), The Natural Science Foundation of Zhejiang Province (LZ22F020007)

王滨, 李思敏, 钱亚冠, 等. 基于剪枝技术和鲁棒蒸馏融合的轻量对抗攻击防御方法[J]. 网络与信息安全学报, 2022, 8(6): 102-109.

WANG B, LI S M, QIAN Y G, et al. Lightweight defense mechanism against adversarial attacks via adaptive pruning and robust distillation[J]. Chinese Journal of Network and Information Security, 2022, 8(6): 102-109.

王滨(1978− ),男,山东泗水人,博士,浙江省多维感知技术应用与安全重点实验室研究员、博士生导师,主要研究方向为物联网安全、人工智能安全、密码学。

李思敏(1997− ),男,浙江义乌人,浙江科技学院硕士生,主要研究方向为深度学习、人工智能安全和模型压缩。

钱亚冠(1976− ),男,浙江嵊州人,博士,浙江科技学院理学院教授,主要研究方向为深度学习、人工智能安全。

张君(1980− ),女,河南内乡人,浙江省信息化发展中心高级工程师,主要研究方向为网络与信息安全。

李超豪(1995− ),男,浙江温州人,博士,浙江省多维感知技术应用与安全重点实验室副主任,主要研究方向为物联网安全、人工智能安全、感知对抗安全、数据隐私保护。

朱晨鸣(1981− ),男,浙江杭州人,浙江省电子信息产品检验研究院高级工程师,主要研究方向为网络与信息安全。

张鸿飞(1984− ),男,浙江嘉兴人,浙江省电子信息产品检验研究院工程师,主要研究方向为网络与信息安全。

猜你喜欢轻量鲁棒剪枝人到晚年宜“剪枝”保健医苑(2022年5期)2022-06-10基于YOLOv4-Tiny模型剪枝算法成都信息工程大学学报(2021年6期)2021-02-12基于激活-熵的分层迭代剪枝策略的CNN模型压缩计算机应用(2020年5期)2020-06-07轻量新能源汽车应用开关磁阻电机系统匹配的研究时代人物(2019年29期)2019-11-25基于高阶LADRC的V/STOL飞机悬停/平移模式鲁棒协调解耦控制北京航空航天大学学报(2019年9期)2019-10-26基于学习的鲁棒自适应评判控制研究进展自动化学报(2019年6期)2019-07-23我对轻量型并继竿的看法垂钓(2018年6期)2018-09-10目标鲁棒识别的抗旋转HDO 局部特征描述自动化学报(2017年4期)2017-06-15剪枝天津诗人(2017年2期)2017-03-16目标轨迹更新的点到点鲁棒迭代学习控制浙江大学学报(工学版)(2015年1期)2015-03-01

推荐访问:剪枝 蒸馏 防御