提出正交概念擦除方法,高效精准地移除扩散模型中的特定概念。
Orthogonal Concept Erasure for Diffusion Models

- 从几何角度改用乘法更新参数,避免方向与幅度干扰。
- 单次擦除100个概念仅需4.3秒,同时保持生成能力不下降。
- 适合需要快速、安全编辑模型内容的研究者与开发者。
概念擦除已成为缓解扩散模型中不当或危险内容的有前景方法,但现有方法仍存在显著局限。基于训练的方法虽有效却计算成本高,难以扩展;基于编辑的方法更高效且便于部署,却难以兼顾精确擦除与整体生成能力的保留。我们发现编辑类方法的核心问题在于依赖加性参数更新。实证分析表明,概念语义主要依赖神经元方向而非幅度,而整体生成能力则依赖神经元间的角几何关系。加性更新会不可避免地纠缠方向、幅度与角几何,导致擦除与生成性能相互干扰。为此,我们提出正交概念擦除(OCE),将编辑式擦除重构为乘性参数更新,通过层内正交变换实现精确擦除,同时保留神经元幅度与角几何结构。此外,针对多概念擦除中的冲突约束,OCE引入子空间级目标与结构化子空间操作,提升擦除效果与可扩展性。在单概念与多概念擦除上的大量实验表明,OCE在概念擦除与非目标保留方面均优于现有方法,可在4.3秒内擦除多达100个概念。
原文摘要 · Abstract (English)
Concept erasure has emerged as a promising approach to mitigate undesired or unsafe content in diffusion models, yet existing methods still face significant limitations. While training-based methods are effective, their high computational cost limits scalability. Editing-based methods are more efficient and deployment-friendly, yet they struggle to simultaneously achieve precise concept erasure and preserve overall generative capacity. We identify this core limitation of the editing-based methods as reliance on additive parameter updates. Our empirical analysis reveals that concept semantics primarily depend on neuron direction rather than neuron magnitude, while overall generative capacity relies on the angular geometry of neurons. As additive updates inherently entangle direction, magnitude, and angular geometry, they inevitably introduce unintended interference between concept erasure and overall generation performance. To address this, we propose Orthogonal Concept Erasure (OCE), which reformulates editing-based erasure as multiplicative parameter updates from a geometric perspective. Specifically, OCE applies layer-wise orthogonal transformations derived from a closed-form solution to the parameters, enabling precise concept erasure while preserving the neuron magnitude and angular geometry. Furthermore, to address conflicting constraints in multi-concept erasure, OCE introduces a subspace-level objective with structured subspace manipulation, yielding a more effective and scalable erasure. Extensive experiments on single- and multi-concept erasure demonstrate that OCE outperforms existing methods in concept erasure and non-target preservation, erasing up to 100 concepts in 4.3 s. Code: https://github.com/HansSunY/OCE.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。