arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2108.00352cs.CRcs.CVcs.LG
收藏

BadEncoder:自监督学习中对预训练编码器的后门攻击

BadEncoder: Backdoor Attacks to Pre-trained Encoders in Self-Supervised Learning

Duke University(杜克大学)

Jinyuan Jia, Yupei Liu, Neil Zhenqiang Gong

摘要:

计算机视觉中的自监督学习旨在使用大量未标记的图像或(图像,文本)对来预训练图像编码器。预训练的图像编码器然后可以被用作特征提取器,以利用少量或没有标记的训练数据来构建用于许多下游任务的下游分类器。在这项工作中,我们提出了BadEncoder,这是对自监督学习的第一次后门攻击。特别是,我们的BadEncoder将后门注入到预先训练的图像编码器中,使得基于后门图像编码器为不同下游任务构建的下游分类器同时继承后门行为。我们制定我们的BadEncoder作为一个优化问题,我们提出了一个基于梯度下降的方法来解决它,它产生一个后门的图像编码器从一个干净的。我们对多个数据集的广泛经验评估结果表明,我们的BadEncoder在保持下游分类器准确性的同时,实现了高攻击成功率。我们还展示了BadEncoder使用两个公开可用的真实图像编码器的有效性,即,Google的图像编码器在ImageNet上进行了预训练,OpenAI的对比图像预训练(CLIP)图像编码器在从互联网上收集的4亿对(图像,文本)上进行了预训练。此外,我们考虑的防御包括神经净化和MNTD(经验防御)以及PatchGuard(可证明的防御)。我们的研究结果表明,这些防御措施不足以抵御BadEncoder,突出了对BadEncoder的新防御措施的需求。我们的代码可在https://github.com/jjy1994/BadEncoder上公开获取。

英文摘要:

Self-supervised learning in computer vision aims to pre-train an image encoder using a large amount of unlabeled images or (image, text) pairs. The pre-trained image encoder can then be used as a feature extractor to build downstream classifiers for many downstream tasks with a small amount of or no labeled training data. In this work, we propose BadEncoder, the first backdoor attack to self-supervised learning. In particular, our BadEncoder injects backdoors into a pre-trained image encoder such that the downstream classifiers built based on the backdoored image encoder for different downstream tasks simultaneously inherit the backdoor behavior. We formulate our BadEncoder as an optimization problem and we propose a gradient descent based method to solve it, which produces a backdoored image encoder from a clean one. Our extensive empirical evaluation results on multiple datasets show that our BadEncoder achieves high attack success rates while preserving the accuracy of the downstream classifiers. We also show the effectiveness of BadEncoder using two publicly available, real-world image encoders, i.e., Google's image encoder pre-trained on ImageNet and OpenAI's Contrastive Language-Image Pre-training (CLIP) image encoder pre-trained on 400 million (image, text) pairs collected from the Internet. Moreover, we consider defenses including Neural Cleanse and MNTD (empirical defenses) as well as PatchGuard (a provable defense). Our results show that these defenses are insufficient to defend against BadEncoder, highlighting the needs for new defenses against our BadEncoder. Our code is publicly available at: https://github.com/jjy1994/BadEncoder.

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

↑