提升图像生成模型去学习后的文本对齐能力,避免内容失真。
Boosting Alignment for Post-Unlearning Text-to-Image Generative Models
- 每轮去学习时动态优化模型更新方向,兼顾去学习与对齐效果。
- 在扩散模型上移除指定类别后,仍保持与原模型高度一致的生成质量。
- 适合需要版权合规与安全可控生成的图像生成系统开发者。
大规模生成模型虽具备强大的图像生成能力,但常无意生成有害或不当内容,引发版权问题。为此,机器去学习成为关键,用于清除模型中的不良知识。然而现有方法常因文本-图像对齐与去学习目标相互冲突,导致去学习效果差或对齐性能下降。本文提出一种框架,在每次去学习迭代中寻找最优模型更新,确保两个目标持续提升,并推导出该更新的理论特征。同时设计策略性数据集多样性增强方法,以提升性能。实验表明,本方法能有效从近期基于扩散的生成模型及Stable Diffusion模型中移除目标类别和概念,同时保持与原始训练状态的高度对齐,显著优于现有最佳基线。代码将开源于https://github.com/reds-lab/Restricted_gradient_diversity_unlearning.git。
原文摘要 · Abstract (English)
Large-scale generative models have shown impressive image-generation capabilities, propelled by massive data. However, this often inadvertently leads to the generation of harmful or inappropriate content and raises copyright concerns. Driven by these concerns, machine unlearning has become crucial to effectively purge undesirable knowledge from models. While existing literature has studied various unlearning techniques, these often suffer from either poor unlearning quality or degradation in text-image alignment after unlearning, due to the competitive nature of these objectives. To address these challenges, we propose a framework that seeks an optimal model update at each unlearning iteration, ensuring monotonic improvement on both objectives. We further derive the characterization of such an update. In addition, we design procedures to strategically diversify the unlearning and remaining datasets to boost performance improvement. Our evaluation demonstrates that our method effectively removes target classes from recent diffusion-based generative models and concepts from stable diffusion models while maintaining close alignment with the models' original trained states, thus outperforming state-of-the-art baselines. Our code will be made available at https://github.com/reds-lab/Restricted_gradient_diversity_unlearning.git.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。