用少量有毒样本让模型快速忘掉后门,提升安全性和准确率
Efficient Backdoor Defense in Multimodal Contrastive Learning: A Token-Level Unlearning Method for Mitigating Threats

- 通过过拟合识别可疑数据,筛选少样本加速模型遗忘后门
- 在CLIP模型上实现攻击成功率降低19%,干净准确率提升2.57%
- 适合关注多模态模型安全、追求高效防御的研究者
多模态对比学习利用多种数据模态生成高质量特征,但其依赖互联网海量数据源,易受后门攻击。此类攻击在训练中植入恶意行为,推理时由特定触发器激活,带来重大安全风险。现有防御方法虽可缓解影响,但常需大量微调时间且损害干净准确率。本文提出一种基于机器遗忘的高效防御机制——无须大规模训练,仅用少量精心构造的有毒样本,即可快速促使模型遗忘后门漏洞,称为UBT。通过过拟合策略强化后门捷径,精准定位潜在中毒样本;再从中挑选少数样本进行针对性遗忘训练,有效消除后门效应。我们设计了一种新的基于令牌的局部遗忘训练范式,聚焦受损模块,分离后门关联同时保持模型整体完整性。大量实验表明,该方法在CLIP模型上对多种后门攻击均具强防御能力:相比最先进方法,攻击成功率下降19%,干净准确率提升2.57%。
原文摘要 · Abstract (English)
Multimodal contrastive learning uses various data modalities to create high-quality features, but its reliance on extensive data sources on the Internet makes it vulnerable to backdoor attacks. These attacks insert malicious behaviors during training, which are activated by specific triggers during inference, posing significant security risks. Despite existing countermeasures through fine-tuning that reduce the malicious impacts of such attacks, these defenses frequently necessitate extensive training time and degrade clean accuracy. In this study, we propose an efficient defense mechanism against backdoor threats using a concept known as machine unlearning. This entails strategically creating a small set of poisoned samples to aid the model's rapid unlearning of backdoor vulnerabilities, known as Unlearn Backdoor Threats (UBT). We specifically use overfit training to improve backdoor shortcuts and accurately detect suspicious samples in the potential poisoning data set. Then, we select fewer unlearned samples from suspicious samples for rapid forgetting in order to eliminate the backdoor effect and thus improve backdoor defense efficiency. In the backdoor unlearning process, we present a novel token-based portion unlearning training regime. This technique focuses on the model's compromised elements, dissociating backdoor correlations while maintaining the model's overall integrity. Extensive experimental results show that our method effectively defends against various backdoor attack methods in the CLIP model. Compared to SoTA backdoor defense methods, UBT achieves the lowest attack success rate while maintaining a high clean accuracy of the model (attack success rate decreases by 19% compared to SOTA, while clean accuracy increases by 2.57%).
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。