adv sat:Sinkhorn对抗训练(SAT) 源码

最优运输作为对抗攻击的防御 该存储库包含论文“ (ICPR'20)的参考源代码。 有关更多信息,请查看我们的论文,或。 我们提出了一种基于最新最优运输理论和鲁棒性新指标的鲁棒对抗训练方法。 要求 python == 3.6 + pytorch == 1.6.0 advertorch == 0.2.0 您可以运行conda env create -f environment.yml来安装所需的软件包。 用SAT训练模型 要在CIFAR10上使用SAT训练模型,请使用: python train.py \ --dataset ' cifar10 ' \ -n 60 \ --epoch_adv 0 \ --gpu 0 \ --arch ' resnet ' \ --sgd \ --sink \ --sink_eps 1 \