arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2012.07006cs.CRcs.LG

DeepSweep:一种使用数据增强缓解DNN后门攻击的评估框架

DeepSweep: An Evaluation Framework for Mitigating DNN Backdoor Attacks using Data Augmentation

Tsinghua University(清华大学);University of California San Diego(加利福尼亚大学圣迭戈分校);Chongqing University(重庆大学);Nanyang Technological University(南洋理工大学);Texas A&M University Commerce(得克萨斯A&M大学康默斯分校);The University of Texas at Dallas(得克萨斯大学达拉斯分校)

Han Qiu, Yi Zeng, Shangwei Guo, Tianwei Zhang, Meikang Qiu, Bhavani Thuraisingham

AI总结:

本文提出评估框架DeepSweep,通过全面评估71种数据增强函数寻找最优策略,采用微调受感染模型与推理时预处理输入样本的统一防御方案,有效缓解八种后门攻击并优于五种现有防御方法。

摘要:

公共资源和服务(例如,数据集、训练平台、预训练模型)已被广泛采用以简化基于深度学习的应用的开发。然而,如果第三方提供商不可信,他们可以向数据集中注入中毒样本或在那些模型中嵌入后门。这种完整性破坏可能导致严重的后果,特别是在安全与安保至关重要的应用中。为了实现更高的有效性和隐蔽性,已经提出了各种后门攻击技术。不幸的是,现有的防御解决方案在实践中无法以全面的方式阻止这些攻击。在本文中,我们研究了数据增强技术在缓解后门攻击和增强DL模型鲁棒性方面的有效性。为了实现这一目标,引入了一个评估框架。具体而言,我们考虑了一种统一的防御解决方案,该方案(1)采用数据增强策略对受感染的模型进行微调,并消除嵌入后门的影响;(2)使用另一种增强策略对输入样本进行预处理,并在推理期间使触发器失效。我们提出了一种系统的方法,通过全面评估71种最先进的数据增强函数,来发现抵御不同后门攻击的最优策略。大量实验表明,我们识别出的策略能够有效缓解八种不同类型的后门攻击,并优于五种现有的防御方法。我们设想该框架可以作为一个良好的基准工具,以推动未来的DNN后门研究。

英文摘要:

Public resources and services (e.g., datasets, training platforms, pre-trained models) have been widely adopted to ease the development of Deep Learning-based applications. However, if the third-party providers are untrusted, they can inject poisoned samples into the datasets or embed backdoors in those models. Such an integrity breach can cause severe consequences, especially in safety- and security-critical applications. Various backdoor attack techniques have been proposed for higher effectiveness and stealthiness. Unfortunately, existing defense solutions are not practical to thwart those attacks in a comprehensive way. In this paper, we investigate the effectiveness of data augmentation techniques in mitigating backdoor attacks and enhancing DL models' robustness. An evaluation framework is introduced to achieve this goal. Specifically, we consider a unified defense solution, which (1) adopts a data augmentation policy to fine-tune the infected model and eliminate the effects of the embedded backdoor; (2) uses another augmentation policy to preprocess input samples and invalidate the triggers during inference. We propose a systematic approach to discover the optimal policies for defending against different backdoor attacks by comprehensively evaluating 71 state-of-the-art data augmentation functions. Extensive experiments show that our identified policy can effectively mitigate eight different kinds of backdoor attacks and outperform five existing defense methods. We envision this framework can be a good benchmark tool to advance future DNN backdoor studies.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑