UFID: A Unified Framework for Input-level Backdoor Detection on Diffusion Models

Environment

  • Python 3.8.5
  • PyTorch 1.10.1+cu11 or 1.11.0+cu102

Usage

Install Require Packages and Prepare Essential Data

Please check VillanDiffusion for more details.

Demo Example with Pokemon-Caption Dataset

bash run_conditional.sh