📄
Abstract - AutoAnchor: Stable Diffusion Unlearning Using Cross-Attention as a Manifold Surrogate
Diffusion unlearning is essential for mitigating the generation of harmful or copyrighted content in text-to-image models. Current diffusion unlearning techniques determine the model update direction by either using alternatives of the target concept as an anchor or using empty prompts. The anchor-based method relies on manually and semantically-chosen anchors that risk biased unlearning, while the anchor-free method inherently suffers from unrobust unlearning due to unconstrained latent updates. In this work, we theoretically formalize such unstable diffusion unlearning issues under the manifold hypothesis and prove that lacking a manifold-proximal anchor inevitably induces significant normal-space drift that degrades unlearning performance. To achieve stable unlearning, we propose \mysysn, a two-stage framework that automatically synthesizes manifold-proximal anchors. However, direct geometric manifold optimization is computationally intractable. To address this challenge, \mysys introduces a novel cross-attention consistency loss which serves as a highly efficient surrogate of manifold proximity. Experimental results demonstrate that \mysys effectively achieves robust and unbiased unlearning across various state-of-the-art baselines, significantly improving targeted concept removal (by up to 31.04\% in CLIP score) and non-target utility (by up to 4.18\% in CLIP score). Moreover, \mysys can also be easily integrated into existing diffusion unlearning methods to enhance their unlearning performance (by 6.30\% for concept removal and 6.65\% for utility on average).
AutoAnchor:利用交叉注意力作为流形代理的稳定扩散模型遗忘学习 /
AutoAnchor: Stable Diffusion Unlearning Using Cross-Attention as a Manifold Surrogate
1️⃣ 一句话总结
本文提出一种名为AutoAnchor的两阶段框架,通过自动生成靠近数据流形的“锚点”并设计高效的交叉注意力一致性损失来替代复杂的几何优化,从而在扩散模型中实现更稳定、无偏且有效的概念遗忘,既能精准移除目标有害内容,又能保持模型对其他内容的生成质量。