TERD: A Unified Framework for Safeguarding Diffusion Models Against Backdoors

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.

Acknowlegements

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

Cite this work

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 mo666666@stu.pku.edu.cn!