cmi

The code is based on CMI and OoD distillation is based on single-img-extrapolating.

CMI

• Attack

wandb sweep sweeps/cmi_cifar10_poi.yml

• Defense

wandb sweep sweeps/cmi_cifar10.yml

Customization

Add dataset: Edit get_dataset and NORMALIZE_DICT in cmi/registry.py.

Add model: Edit MODEL_DICT in cmi/registry.py to add model architecture. To set pre-trained model files, edit get_pretrained_path in cmi/.

OoD

The OoD dataset include OoD patches generated from one image. Details follow single-img-explorating.

• Attack

wandb sweep sweeps/ood_cifar10_poi.yml

• Defense

wandb sweep sweeps/ood_cifar10.yml