用诱饵分支捕获并清除模型中的后门攻击
Mitigating Backdoors via Decoy Shortcuts and Knowledge Decoupling

- 引入轻量级诱饵分支作为陷阱,吸引后门知识
- 训练后丢弃诱饵分支即可清除后门,无需额外数据
- 适用于多种模型架构,对正常性能无影响
后门攻击严重威胁深度神经网络,尤其在依赖第三方数据训练时,攻击者可通过数据投毒植入恶意行为。本文发现,后门行为在与主网络联合训练时,往往被一个更简单的并行分支吸收。基于此,我们提出训练时防御方法TR:引入轻量级快捷分支作为‘蜜罐’捕获后门知识。训练完成后,通过丢弃该分支即可移除后门,无需额外数据。为进一步增强后门隔离性并保持良性性能,设计基于熵的权重分配策略,引导中毒样本经由蜜罐,使主网络专注良性学习。此外,提出自动快捷分支生成策略,提升跨模型架构的泛化能力。在四个基准数据集和五种模型架构上的大量实验表明,该方法能有效缓解多种后门攻击,同时保持良性数据性能。代码已开源。
原文摘要 · Abstract (English)
Backdoor attacks pose a serious threat to deep neural networks, especially when training relies on third-party data, allowing adversaries to inject malicious behaviors through data poisoning. In this work, we reveal that backdoor behaviors tend to be absorbed by a simpler parallel branch when jointly trained with the main network. Motivated by this insight, we propose Trapping and Removing (TR), a simple yet effective training-time defense that introduces a lightweight shortcut branch as a "honeypot" to trap backdoor knowledge. After training, backdoors can be removed by discarding the shortcut, without requiring any additional data. To further enhance backdoor isolation while maintaining benign performance, we design a knowledge decoupling strategy with entropy-based weight assignment, encouraging poisoned samples to flow through the honeypot while guiding the main network to focus on benign learning. In addition, we introduce an automatic shortcut generation strategy to improve generalization across model architectures. Extensive experiments on four benchmark datasets and five model architectures demonstrate that our approach effectively mitigates a wide range of backdoor attacks while preserving performance on benign data. Code: https://github.com/Zixuan-Zhu/TR}{github.com/Zixuan-Zhu/TR.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。