atlas

Research map / Image generation / Text-to-image diffusion

Ring-A-Bell! How Reliable are Concept Removal Methods For Diffusion Models?

ICLR 2024 · Image generation › Text-to-image diffusion · approach: Concept erasure in diffusion models · 4 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes Ring-A-Bell, a model-agnostic red-teaming tool that builds concept embeddings from contrastive prompt pairs in a CLIP text encoder and uses a genetic algorithm to find problematic prompts that revive removed concepts in T2I models and online services.

Problem

Safety mechanisms for text-to-image diffusion models (safety checkers and concept removal methods) are rarely tested against a wide range of prompts, and their reliability under adversarial prompting is unclear, especially for black-box or online services.

Approach

Extract a holistic concept embedding as the averaged difference between CLIP text embeddings of concept-containing and concept-free prompt pairs, add it (scaled by eta) to a target prompt's embedding, and invert it to discrete tokens with a genetic algorithm, so the attack needs only the text encoder and can be prepared offline.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Replace QF-Attack with Ring-A-BellESD · Nudity (w/o SC)ASR: 6.32% -> 35.79%Helped
Replace QF-Attack with Ring-A-Bell-UnionESD · Nudity (w/o SC)ASR: 6.32% -> 55.79%Helped
Replace QF-Attack with Ring-A-BellESD · Violence (w/o SC)ASR: 56% -> 54%No clear effect
Replace QF-Attack with Ring-A-Bell-UnionESD · Violence (w/o SC)ASR: 56% -> 86%Helped
Use the union of three Ring-A-Bell configurationsconcept removal methods · Nudity and violence (w/o SC)ASR: ESD nudity: 35.79% -> 55.79%; ESD violence: 54% -> 86%Helped
Increase prompt length from K=16 to K=77 (ablation)Ring-A-Bell · Nudity, ESD (w/o SC)ASR: 35.79% -> 20%Hurt
Increase prompt length from K=16 to K=77 (ablation)Ring-A-Bell · Violence, ESD (w/o SC)ASR: 38.8% -> 54%Helped
Set η to 5 instead of 4 (ablation)Ring-A-Bell · Violence, ESD (w/o SC)ASR: 50.4% -> 59.2%Helped
Increase η from 5 to 5.5 (ablation)Ring-A-Bell · Violence, ESD (w/o SC)ASR: 59.2% -> 54%Hurt
Use GA instead of PeZ (ablation)Ring-A-Bell · Nudity, ESD (w/o SC)ASR: 11.58% -> 35.79%Helped
Use GA instead of PeZ (ablation)Ring-A-Bell · Violence, ESD (w/o SC)ASR: 32.00% -> 54.00%Helped

Surprising findings

Limitations

Experimental setup

Datasets: I2P ChatGPT-generated prompt pairs

Benchmarks: I2P

Metrics: attack success rate NudeNet detection Q16 classification

Baselines: QF-Attack P4D Original Prompts PeZ

Builds on: Stable Diffusion CLIP Erased Stable Diffusion Safe Latent Diffusion PeZ

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy