arXiv:2409.17946cs.CRcs.AI2024-09被引 7

用小模型教大模型学坏,实现高效后门攻击

Breaking PEFT Limitations: Leveraging Weak-to-Strong Knowledge Transfer for Backdoor Attacks in LLMs

  • 先用全参微调毒化小模型作教师,再通过知识蒸馏传递后门
  • 在4个大模型上实现接近100%的后门触发成功率
  • 突破参数高效微调限制,适合研究模型安全的学者

尽管大型语言模型(LLMs)表现出色,但易受后门攻击。传统攻击需全参数微调,计算开销大;而参数高效微调(PEFT)因更新参数有限,难以对齐触发词与目标标签。本文验证了PEFT下后门攻击面临可行性挑战。为此提出基于特征对齐增强知识蒸馏(FAKD)的新算法:先用全参数微调毒化小型教师模型,再通过FAKD将后门隐式迁移至大规模学生模型(采用PEFT)。理论分析表明FAKD可提升攻击有效性。实验在4个语言模型、4种攻击算法及2类教师架构上验证,针对PEFT的后门攻击成功率接近100%。

原文摘要 · Abstract (English)

Despite being widely applied due to their exceptional capabilities, Large Language Models (LLMs) have been proven to be vulnerable to backdoor attacks. These attacks introduce targeted vulnerabilities into LLMs by poisoning training samples and full-parameter fine-tuning (FPFT). However, this kind of backdoor attack is limited since they require significant computational resources, especially as the size of LLMs increases. Besides, parameter-efficient fine-tuning (PEFT) offers an alternative but the restricted parameter updating may impede the alignment of triggers with target labels. In this study, we first verify that backdoor attacks with PEFT may encounter challenges in achieving feasible performance. To address these issues and improve the effectiveness of backdoor attacks with PEFT, we propose a novel backdoor attack algorithm from the weak-to-strong based on Feature Alignment-enhanced Knowledge Distillation (FAKD). Specifically, we poison small-scale language models through FPFT to serve as the teacher model. The teacher model then covertly transfers the backdoor to the large-scale student model through FAKD, which employs PEFT. Theoretical analysis reveals that FAKD has the potential to augment the effectiveness of backdoor attacks. We demonstrate the superior performance of FAKD on classification tasks across four language models, four backdoor attack algorithms, and two different architectures of teacher models. Experimental results indicate success rates close to 100% for backdoor attacks targeting PEFT.

后门攻击知识蒸馏大模型安全

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