arXiv:2410.18267cs.AI2024-10被引 5

无需训练数据和计算资源,秒级注入模型后门

Backdoor in Seconds: Unlocking Vulnerabilities in Large Pre-trained Models via Model Editing

  • 利用轻量代码本编辑模型嵌入层实现无数据、无训练攻击
  • 在ViT、CLIP等模型上成功触发图像分类、生成等任务的后门
  • 适合安全评估人员研究大模型潜在漏洞

大规模预训练模型在众多下游任务中取得显著成果。然而,近期研究发现,通过污染训练数据可实施对抗性攻击(即后门攻击),对大规模预训练模型的实际应用构成重大威胁,尤其针对定制化模型。因此,探索此类模型脆弱性的独特挑战至关重要。通过对ViT等大型预训练模型进行实证研究,我们发现两大挑战:一是无法操控或访问大规模训练数据;二是训练或微调这些模型需要大量计算资源。为应对这些挑战,我们确立了适用于大型预训练模型的有效且可行的后门攻击新标准。据此,我们提出EDT方法——一种高效、无数据、免训练的后门攻击技术。受模型编辑技术启发,EDT将轻量级代码本注入模型后门,仅通过替换被污染图像的嵌入表示即可完成攻击,无需污染训练数据或训练目标模型。我们在ViT、CLIP、BLIP及Stable Diffusion等多种预训练模型上,针对图像分类、图像描述生成和图像生成等下游任务进行了实验,验证了该方法的有效性。相关代码已提供于补充材料中。

原文摘要 · Abstract (English)

Large pre-trained models have achieved notable success across a range of downstream tasks. However, recent research shows that a type of adversarial attack ($\textit{i.e.,}$ backdoor attack) can manipulate the behavior of machine learning models through contaminating their training dataset, posing significant threat in the real-world application of large pre-trained model, especially for those customized models. Therefore, addressing the unique challenges for exploring vulnerability of pre-trained models is of paramount importance. Through empirical studies on the capability for performing backdoor attack in large pre-trained models ($\textit{e.g.,}$ ViT), we find the following unique challenges of attacking large pre-trained models: 1) the inability to manipulate or even access large training datasets, and 2) the substantial computational resources required for training or fine-tuning these models. To address these challenges, we establish new standards for an effective and feasible backdoor attack in the context of large pre-trained models. In line with these standards, we introduce our EDT model, an \textbf{E}fficient, \textbf{D}ata-free, \textbf{T}raining-free backdoor attack method. Inspired by model editing techniques, EDT injects an editing-based lightweight codebook into the backdoor of large pre-trained models, which replaces the embedding of the poisoned image with the target image without poisoning the training dataset or training the victim model. Our experiments, conducted across various pre-trained models such as ViT, CLIP, BLIP, and stable diffusion, and on downstream tasks including image classification, image captioning, and image generation, demonstrate the effectiveness of our method. Our code is available in the supplementary material.

后门攻击模型编辑大模型安全

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