通过风格扰动防御文本生成图像的风格模仿攻击
StyleGuard: Preventing Text-to-Image-Model-based Style Mimicry Attacks by Style Perturbations
- 在隐空间优化风格特征,使图像偏离原风格
- 对抗多种净化方法,对不同模型均有良好迁移性
- 适用于DreamBooth、Textual Inversion等主流方法
近期,基于扩散模型的文本到图像生成技术广泛用于风格模仿与个性化定制,如DreamBooth和Textual Inversion,引发知识产权保护与虚假内容生成的担忧。已有研究如Glaze和Anti-DreamBooth采用对抗噪声进行防护,但近期基于净化的方法(如DiffPure和Noise Upscaling)已成功攻破这些防御,暴露其脆弱性。此外,现有方法跨模型迁移能力有限,难以应对未知模型。为此,本文提出新型反模仿方法StyleGuard。我们设计一种新颖的风格损失,优化隐空间中的风格相关特征,使其偏离原始图像,提升模型无关的迁移性;同时,为增强扰动对扩散净化方法的规避能力,引入一种包含集成净化器与上采样器的上采样损失。在WikiArt和CelebA数据集上的大量实验表明,StyleGuard在多种变换与净化策略下均表现更优,有效抵御各类模型的风格模仿攻击,且对DreamBooth和Textual Inversion等方法同样有效。代码已开源。
原文摘要 · Abstract (English)
Recently, text-to-image diffusion models have been widely used for style mimicry and personalized customization through methods such as DreamBooth and Textual Inversion. This has raised concerns about intellectual property protection and the generation of deceptive content. Recent studies, such as Glaze and Anti-DreamBooth, have proposed using adversarial noise to protect images from these attacks. However, recent purification-based methods, such as DiffPure and Noise Upscaling, have successfully attacked these latest defenses, showing the vulnerabilities of these methods. Moreover, present methods show limited transferability across models, making them less effective against unknown text-to-image models. To address these issues, we propose a novel anti-mimicry method, StyleGuard. We propose a novel style loss that optimizes the style-related features in the latent space to make it deviate from the original image, which improves model-agnostic transferability. Additionally, to enhance the perturbation's ability to bypass diffusion-based purification, we designed a novel upscale loss that involves ensemble purifiers and upscalers during training. Extensive experiments on the WikiArt and CelebA datasets demonstrate that StyleGuard outperforms existing methods in robustness against various transformations and purifications, effectively countering style mimicry in various models. Moreover, StyleGuard is effective on different style mimicry methods, including DreamBooth and Textual Inversion. The code is available at https://github.com/PolyLiYJ/StyleGuard.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。