Code for reproducing the results in the paper "Ring-A-Bell! How Reliable are Concept Removal Methods For Diffusion Models?".
The code of Concept Extraction is in Get_Concept_Vector.ipynb.
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.
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}
}