无需训练即可快速低成本擦除图像生成模型中的特定概念。
Precise, Fast, and Low-cost Concept Erasure in Value Space: Orthogonal Complement Matters
- 通过值空间正交补计算实现概念擦除,无需训练。
- 相比最优方法,先验内容保留提升2至10倍。
- 适用于多种扩散模型,适合需要安全可控生成的场景。
文本到图像生成技术的快速发展及其广泛应用,得益于扩散模型的进展,迫切需要以精确、快速且低成本的方式从预训练模型中删除不希望出现的概念,如版权内容、不当信息和安全隐患。概念擦除需兼顾两个目标:精准移除目标概念(擦除有效性)和对非目标内容影响最小化(先验保留)。现有方法难以平衡二者,且计算成本高。本文提出无需训练的高效擦除方法AdaVD,基于线性代数中的正交补运算,在扩散模型的UNet中每个交叉注意力层的值空间内实现。设计可自适应调节的偏移因子,有效增强先验保留,同时不牺牲擦除效果。与基于训练和无训练的最先进方法对比实验表明,AdaVD在单概念和多概念擦除任务中均表现优异,先验保留优于第二名2至10倍,擦除效果达到最优或接近最优。该方法支持多种扩散模型及下游图像生成任务,代码已公开于https://github.com/WYuan1001/AdaVD。
原文摘要 · Abstract (English)
Recent success of text-to-image (T2I) generation and its increasing practical applications, enabled by diffusion models, require urgent consideration of erasing unwanted concepts, e.g., copyrighted, offensive, and unsafe ones, from the pre-trained models in a precise, timely, and low-cost manner. The twofold demand of concept erasure includes not only a precise removal of the target concept (i.e., erasure efficacy) but also a minimal change on non-target content (i.e., prior preservation), during generation. Existing methods face challenges in maintaining an effective balance between erasure efficacy and prior preservation, and they can be computationally costly. To improve, we propose a precise, fast, and low-cost concept erasure method, called Adaptive Value Decomposer (AdaVD), which is training-free. Our method is grounded in a classical linear algebraic operation of computing the orthogonal complement, implemented in the value space of each cross-attention layer within the UNet of diffusion models. We design a shift factor to adaptively navigate the erasure strength, enhancing effective prior preservation without sacrificing erasure efficacy. Extensive comparative experiments with both training-based and training-free state-of-the-art methods demonstrate that the proposed AdaVD excels in both single and multiple concept erasure, showing 2 to 10 times improvement in prior preservation than the second best, meanwhile achieving the best or near best erasure efficacy. AdaVD supports a series of diffusion models and downstream image generation tasks, with code available on: https://github.com/WYuan1001/AdaVD.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。