arXiv:2411.16746cs.CRcs.AI2024-11被引 14

用少量资源通过LoRA实现隐蔽的模型融合后门攻击

LoBAM: LoRA-Based Backdoor Attack on Model Merging

  • 基于LoRA设计智能权重放大机制,降低攻击资源需求
  • 在多种融合场景下攻击成功率显著提升,达90%以上
  • 攻击极难被检测,适合资源受限的隐蔽攻击场景

模型融合是一种新兴技术,可将针对不同任务微调的多个模型整合,生成在多领域表现优异的通用模型。然而,该方法可能引入后门攻击风险——一个恶意模型即可破坏融合模型的完整性。现有研究通常假设攻击者具备充足计算资源,能对预训练模型进行全量微调,但随着模型规模增大,此假设难以成立。在实际资源受限场景中,攻击者仅能使用如低秩适应(LoRA)等轻量化技术生成恶意模型,此时攻击是否仍有效尚不明确。本文首次发现,采用LoRA微调时攻击效果显著下降。为此,我们提出LoBAM方法,在极低训练资源下实现高攻击成功率。其核心思想是智能放大恶意权重,有效增强攻击效力。大量实验证明,该设计在多种模型融合场景中均显著提升攻击成功率。此外,我们的方法具有高度隐蔽性,难以被检测和防御。

原文摘要 · Abstract (English)

Model merging is an emerging technique that integrates multiple models fine-tuned on different tasks to create a versatile model that excels in multiple domains. This scheme, in the meantime, may open up backdoor attack opportunities where one single malicious model can jeopardize the integrity of the merged model. Existing works try to demonstrate the risk of such attacks by assuming substantial computational resources, focusing on cases where the attacker can fully fine-tune the pre-trained model. Such an assumption, however, may not be feasible given the increasing size of machine learning models. In practice where resources are limited and the attacker can only employ techniques like Low-Rank Adaptation (LoRA) to produce the malicious model, it remains unclear whether the attack can still work and pose threats. In this work, we first identify that the attack efficacy is significantly diminished when using LoRA for fine-tuning. Then, we propose LoBAM, a method that yields high attack success rate with minimal training resources. The key idea of LoBAM is to amplify the malicious weights in an intelligent way that effectively enhances the attack efficacy. We demonstrate that our design can lead to improved attack success rate through extensive empirical experiments across various model merging scenarios. Moreover, we show that our method is highly stealthy and is difficult to detect and defend against.

后门攻击模型融合LoRA安全漏洞

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。