Ring-A-Bell

Code for reproducing the results in the paper "Ring-A-Bell! How Reliable are Concept Removal Methods For Diffusion Models?" Link.

Framework of Ring-A-Bell

📌 Concept Extraction

The code of Concept Extraction is in Get_Concept_Vector.ipynb.

📌 Prompt Discovery

The code of Prompt Discovery is in InversePrompt.ipynb.

You can get the problematic prompts after you conduct InversePrompt.ipynb. Then you can use problematic prompts to evaluate the effectiveness of the concept removal methods.

Citation

🔔🔔 If you need InvPrompts for nudity, please visit https://huggingface.co/datasets/Chia15/RingABell-Nudity and send the request.

Please feel free to email us at chiayihsu8315@gmail.com. If this work is useful in your own research, please consider citing our work!

@inproceedings{
ringabell,
title={Ring-A-Bell! How Reliable are Concept Removal Methods For Diffusion Models?},
author={Yu-Lin Tsai*, Chia-Yi Hsu*, Chulin Xie, Chih-Hsun Lin, Jia-You Chen, Bo Li, Pin-Yu Chen, Chia-Mu Yu, Chun-Ying Huang},
booktitle={The Twelfth International Conference on Learning Representations},
year={2024},
url={https://openreview.net/forum?id=lm7MRcsFiS}
}