无需触发器反演,通过重编程模型防御后门攻击
REFINE: Inversion-Free Backdoor Defense via Model Reprogramming
- 用输入变换和输出重映射重构模型行为,不依赖触发器反演
- 在多个数据集上实现90%以上防御率,同时保持模型准确率
- 适合关注模型安全且追求高实用性的研究人员
深度神经网络的后门攻击已成为重大安全隐患,攻击者可在训练阶段植入隐藏恶意行为。基于预处理的防御方法通常聚焦于输入变换或后门触发器反演(BTI),以在推理阶段消除恶意触发器。然而,这类方法存在固有缺陷:基于变换的防御难以兼顾模型性能与防御效果,而基于BTI的方法在缺乏先验知识时难以准确重建触发模式。本文提出REFINE,一种基于模型重编程的无反演后门防御方法。REFINE包含两个关键组件:(1) 输入变换模块,破坏良性与后门模式,生成新良性特征;(2) 输出重映射模块,重新定义模型输出空间,引导输入变换有效进行。通过引入监督对比损失,进一步增强防御能力并维持模型实用性。大量实验证明,REFINE在多个基准数据集上均表现出色,且对潜在自适应攻击具有鲁棒性。
原文摘要 · Abstract (English)
Backdoor attacks on deep neural networks (DNNs) have emerged as a significant security threat, allowing adversaries to implant hidden malicious behaviors during the model training phase. Pre-processing-based defense, which is one of the most important defense paradigms, typically focuses on input transformations or backdoor trigger inversion (BTI) to deactivate or eliminate embedded backdoor triggers during the inference process. However, these methods suffer from inherent limitations: transformation-based defenses often fail to balance model utility and defense performance, while BTI-based defenses struggle to accurately reconstruct trigger patterns without prior knowledge. In this paper, we propose REFINE, an inversion-free backdoor defense method based on model reprogramming. REFINE consists of two key components: \textbf{(1)} an input transformation module that disrupts both benign and backdoor patterns, generating new benign features; and \textbf{(2)} an output remapping module that redefines the model's output domain to guide the input transformations effectively. By further integrating supervised contrastive loss, REFINE enhances the defense capabilities while maintaining model utility. Extensive experiments on various benchmark datasets demonstrate the effectiveness of our REFINE and its resistance to potential adaptive attacks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。