arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2410.19785cs.CRcs.CVcs.LG
收藏

如何对一致性模型植入后门?

How to Backdoor Consistency Models?

University of Texas as Dallas(德克萨斯大学达拉斯分校)

Chengen Wang, Murat Kantarcioglu

AI总结:

本文首次研究一致性模型的后门攻击风险,提出适配其训练机制的投毒框架,并以随机噪声为隐蔽触发器,在保持FID效用的同时实现高特异性后门生成。

摘要:

一致性模型是一类新型模型,通过将噪声直接映射为数据来生成图像,可实现一步生成并显著加快采样过程。然而,其对抗击对抗攻击的鲁棒性尚未得到充分研究。在本工作中,我们首次研究了一致性模型在后门攻击下的脆弱性。尽管此前研究已经探讨了针对扩散模型的后门攻击,但这些研究主要集中于传统扩散模型,采用定制化的后门训练流程和目标函数,而一致性模型具有不同的训练流程和目标函数。我们提出的框架证明了一致性模型易受后门攻击。在图像生成过程中,当从高斯噪声采样时,被投毒的一致性模型生成图像的 Fréchet Inception Distance(FID)与干净模型相当。然而,一旦触发器被激活,它们就会生成后门目标图像。我们探索了多种触发器和目标配置以评估一致性模型的脆弱性,包括使用随机噪声作为触发器。这种新型触发器在视觉上不显眼、更难检测,并且与一致性模型的采样过程高度契合。在所有配置中,我们的框架都成功攻陷了一致性模型,同时保持了高效用性和高特异性。我们还考察了所提攻击的隐蔽性,这种隐蔽性源于一致性模型的独特性质以及高斯噪声触发器难以捉摸的特性。我们的代码可在 https://github.com/chengenw/backdoorCM 获取。

英文摘要:

Consistency models are a new class of models that generate images by directly mapping noise to data, allowing for one-step generation and significantly accelerating the sampling process. However, their robustness against adversarial attacks has not yet been thoroughly investigated. In this work, we conduct the first study on the vulnerability of consistency models to backdoor attacks. While previous research has explored backdoor attacks on diffusion models, those studies have primarily focused on conventional diffusion models, employing a customized backdoor training process and objective, whereas consistency models have distinct training processes and objectives. Our proposed framework demonstrates the vulnerability of consistency models to backdoor attacks. During image generation, poisoned consistency models produce images with a Fréchet Inception Distance (FID) comparable to that of a clean model when sampling from Gaussian noise. However, once the trigger is activated, they generate backdoor target images. We explore various trigger and target configurations to evaluate the vulnerability of consistency models, including the use of random noise as a trigger. This novel trigger is visually inconspicuous, more challenging to detect, and aligns well with the sampling process of consistency models. Across all configurations, our framework successfully compromises the consistency models while maintaining high utility and specificity. We also examine the stealthiness of our proposed attack, which is attributed to the unique properties of consistency models and the elusive nature of the Gaussian noise trigger. Our code is available at \href{https://github.com/chengenw/backdoorCM}{https://github.com/chengenw/backdoorCM}.

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

↑