/DeepSweep

An evaluation framework for mitigating DNN backdoor attacks using data augmentations

Primary LanguageJupyter NotebookMIT LicenseMIT

Set up: Please first download and unpack the DeepSweep fine-tuned model and the original infected model following this link (unpack under the main folder). Please download and unpack the clean dataset of the PubFig following this link (unpack under the 'data' folder). Please download and unpack the patched samples of the PubFig following this link (unpack under the 'data/pubfig/wm' folder).