arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2502.18508cs.CRcs.AIcs.CVcs.LG

REFINE:基于模型重编程的无触发器反转后门防御

REFINE: Inversion-Free Backdoor Defense via Model Reprogramming

Zhejiang University(浙江大学);Hangzhou High-Tech Zone (Binjiang) Institute of Blockchain and Data Security(杭州高新区(滨江)区块链与数据安全研究院);Nanyang Technological University(南洋理工大学);IBM Research(IBM研究院)

Yukun Chen, Shuo Shao, Enhao Huang, Yiming Li, Pin-Yu Chen, Zhan Qin, Kui Ren

AI总结:

针对后门攻击,提出基于模型重编程的无反转防御方法REFINE,通过输入变换与输出重映射结合监督对比损失,在保持模型效用同时增强防御能力。

摘要:

深度神经网络(DNN)的后门攻击已成为一种显著的安全威胁,使攻击者能够在模型训练阶段植入隐藏的恶意行为。基于预处理的防御是最重要的防御范式之一,通常侧重于输入变换或后门触发器反转(BTI),以在推理过程中停用或消除嵌入的后门触发器。然而,这些方法存在固有的局限性:基于变换的防御往往难以平衡模型效用与防御性能,而基于BTI的防御在缺乏先验知识的情况下难以准确重建触发器模式。在本文中,我们提出REFINE,一种基于模型重编程的无反转后门防御方法。REFINE由两个关键组件组成:\textbf{(1)} 一个输入变换模块,用于破坏良性模式和后门模式,生成新的良性特征;\textbf{(2)} 一个输出重映射模块,重新定义模型的输出域以有效引导输入变换。通过进一步整合监督对比损失,REFINE在保持模型效用的同时增强了防御能力。在各种基准数据集上的大量实验证明了我们REFINE的有效性及其对潜在自适应攻击的抵抗能力。

英文摘要:

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.

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

↑