Backdoor defense of diffusion models @ ICML 2024 "TERD: A Framework for Backdoor Defense on Diffusion Model" by Yichuan Mo, Hui Huang, Mingjie Li, Ang Li, Yisen Wang.
This repository is under reconstruction. Please refer to each directory to access the code for each attack.
This repository is built upon the following repositories:
https://github.com/IBM/BadDiffusion
https://github.com/chenweixin107/TrojDiff
https://github.com/yuanzhi-zhu/mini_edm
If you find our code is useful, please cite our accompanying paper!
@inproceedings{moterd,
title={TERD: A Unified Framework for Safeguarding Diffusion Models Against Backdoors},
author={Mo, Yichuan and Huang, Hui and Li, Mingjie and Li, Ang and Wang, Yisen},
booktitle={ICML}
year={2024}
}
If you have any questions, feel free to contact [email protected]!