arXiv:2503.12356cs.CVcs.AI2025-03CVPR被引 30

无需训练即可精准删除图像中特定概念区域,保持其他部分清晰

Localized Concept Erasure for Text-to-Image Diffusion Models Using Training-Free Gated Low-Rank Adaptation

  • 通过轻量门控低秩模块,仅针对目标概念激活删除区域
  • 在不破坏其他内容的前提下,显著提升去概念后图像与提示的匹配度
  • 适合需要安全可控生成的AI绘图应用,尤其适用于批量删除有害概念

基于微调的概念消除方法已在文本到图像扩散模型中展现潜力,能有效防止有害内容生成,同时保留其他概念。为维持模型生成能力,需仅移除图像中目标概念出现的局部区域,而保留其余部分。然而,现有方法常以牺牲其他区域保真度为代价,影响整体生成性能。为此,本文提出局部概念消除框架,实现仅删除特定区域的目标概念,同时保留其他区域。为此设计了一种无需训练的方法——门控低秩适配用于概念消除(GLoCE),其通过注入轻量级模块至扩散模型,由低秩矩阵与简单门控组成,仅依赖少数生成步骤确定,无需训练。通过直接作用于图像嵌入并设计门控仅对目标概念激活,GLoCE可精准选择性移除目标概念所在区域,即使目标与非目标概念共存于同一图像中。大量实验表明,该方法不仅显著提升去概念后的图像保真度,且在有效性、特异性与鲁棒性方面大幅超越现有方法,支持大规模概念删除。

原文摘要 · Abstract (English)

Fine-tuning based concept erasing has demonstrated promising results in preventing generation of harmful contents from text-to-image diffusion models by removing target concepts while preserving remaining concepts. To maintain the generation capability of diffusion models after concept erasure, it is necessary to remove only the image region containing the target concept when it locally appears in an image, leaving other regions intact. However, prior arts often compromise fidelity of the other image regions in order to erase the localized target concept appearing in a specific area, thereby reducing the overall performance of image generation. To address these limitations, we first introduce a framework called localized concept erasure, which allows for the deletion of only the specific area containing the target concept in the image while preserving the other regions. As a solution for the localized concept erasure, we propose a training-free approach, dubbed Gated Low-rank adaptation for Concept Erasure (GLoCE), that injects a lightweight module into the diffusion model. GLoCE consists of low-rank matrices and a simple gate, determined only by several generation steps for concepts without training. By directly applying GLoCE to image embeddings and designing the gate to activate only for target concepts, GLoCE can selectively remove only the region of the target concepts, even when target and remaining concepts coexist within an image. Extensive experiments demonstrated GLoCE not only improves the image fidelity to text prompts after erasing the localized target concepts, but also outperforms prior arts in efficacy, specificity, and robustness by large margin and can be extended to mass concept erasure.

扩散模型概念删除图像生成零样本

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。