中文

Receler:通过轻量级擦除器实现文本到图像扩散模型的可靠概念擦除

计算机视觉与模式识别 2024-07-19 v3 机器学习

摘要

文本到图像扩散模型中的概念擦除旨在使预训练扩散模型无法生成与目标概念相关的图像。为实现可靠的概念擦除,鲁棒性与局部性这两类性质是所期望的。前者阻止模型针对任何改写或习得提示生成与目标概念关联的图像,后者则保留其生成含非目标概念图像的能力。本文提出通过轻量级擦除器实现可靠概念擦除(Receler)。该方法学习一个轻量级擦除器来执行概念擦除,同时通过所提出的概念局部化正则化与对抗提示学习方案满足上述期望性质。针对多种概念的试验验证了 Receler 相对于先前方法的优越性。

关键词

引用

@article{arxiv.2311.17717,
  title  = {Receler: Reliable Concept Erasing of Text-to-Image Diffusion Models via Lightweight Erasers},
  author = {Chi-Pin Huang and Kai-Po Chang and Chung-Ting Tsai and Yung-Hsuan Lai and Fu-En Yang and Yu-Chiang Frank Wang},
  journal= {arXiv preprint arXiv:2311.17717},
  year   = {2024}
}

备注

ECCV 2024. Project page: https://jasper0314-huang.github.io/receler-concept-erasing/