Mar, 2024

利用可学习的提示在文本到图像生成模型中去除不需要的概念

TL;DR我们提出了一种新的方法,通过将可学习提示引入交叉注意力模块,从文本到图像生成模型中去除不良概念。这种可学习提示作为额外的记忆来传递不良概念的知识,并减少这些概念对模型参数和对应文本输入的依赖。通过这种知识传递,擦除这些不良概念更稳定,并对其他概念的负面影响较小。我们在稳定扩散模型上展示了我们方法的有效性,展示了它在去除不良内容同时保留其他无关元素方面的优越性。