├── .gitignore ├── README.md ├── args.py ├── auto_attack_eval.py ├── data ├── __init__.py ├── cifar.py ├── huggingface.py ├── imagenet.py ├── lmdb_dataset.py ├── normalize.py └── tiny_imagenet.py ├── models ├── __init__.py ├── densenet_moe.py ├── densenet_ori.py ├── layers │ ├── __init__.py │ ├── hydra_structured_layers.py │ ├── moe_layer.py │ └── router.py ├── resnet_cifar_moe.py ├── resnet_cifar_ori.py ├── resnet_imagenet_moe.py ├── resnet_imagenet_ori.py ├── vgg_moe.py ├── vgg_ori.py ├── wrn_cifar_moe.py ├── wrn_cifar_ori.py ├── wrn_imagenet_moe.py └── wrn_imagenet_ori.py ├── requirements.txt ├── torchattacks ├── __init__.py ├── attack.py └── attacks │ ├── __init__.py │ ├── _differential_evolution.py │ ├── apgd.py │ ├── apgdt.py │ ├── autoattack.py │ ├── bim.py │ ├── cw.py │ ├── deepfool.py │ ├── eotpgd.py │ ├── fab.py │ ├── ffgsm.py │ ├── fgsm.py │ ├── gn.py │ ├── mifgsm.py │ ├── multiattack.py │ ├── onepixel.py │ ├── pgd.py │ ├── pgddlr.py │ ├── pgdl2.py │ ├── rfgsm.py │ ├── sparsefool.py │ ├── square.py │ ├── tpgd.py │ └── vanila.py ├── train_moe.py ├── train_ori.py └── utils ├── adv_utils.py ├── eval_utils.py ├── general_utils.py ├── model_utils.py └── schedules.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPTML-Group/Robust-MoE-CNN/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPTML-Group/Robust-MoE-CNN/HEAD/README.md -------------------------------------------------------------------------------- /args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPTML-Group/Robust-MoE-CNN/HEAD/args.py -------------------------------------------------------------------------------- /auto_attack_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPTML-Group/Robust-MoE-CNN/HEAD/auto_attack_eval.py -------------------------------------------------------------------------------- /data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPTML-Group/Robust-MoE-CNN/HEAD/data/__init__.py -------------------------------------------------------------------------------- /data/cifar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPTML-Group/Robust-MoE-CNN/HEAD/data/cifar.py -------------------------------------------------------------------------------- /data/huggingface.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPTML-Group/Robust-MoE-CNN/HEAD/data/huggingface.py -------------------------------------------------------------------------------- /data/imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPTML-Group/Robust-MoE-CNN/HEAD/data/imagenet.py -------------------------------------------------------------------------------- /data/lmdb_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPTML-Group/Robust-MoE-CNN/HEAD/data/lmdb_dataset.py -------------------------------------------------------------------------------- /data/normalize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPTML-Group/Robust-MoE-CNN/HEAD/data/normalize.py -------------------------------------------------------------------------------- /data/tiny_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPTML-Group/Robust-MoE-CNN/HEAD/data/tiny_imagenet.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPTML-Group/Robust-MoE-CNN/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/densenet_moe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPTML-Group/Robust-MoE-CNN/HEAD/models/densenet_moe.py -------------------------------------------------------------------------------- /models/densenet_ori.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPTML-Group/Robust-MoE-CNN/HEAD/models/densenet_ori.py -------------------------------------------------------------------------------- /models/layers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/layers/hydra_structured_layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPTML-Group/Robust-MoE-CNN/HEAD/models/layers/hydra_structured_layers.py -------------------------------------------------------------------------------- /models/layers/moe_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPTML-Group/Robust-MoE-CNN/HEAD/models/layers/moe_layer.py -------------------------------------------------------------------------------- /models/layers/router.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPTML-Group/Robust-MoE-CNN/HEAD/models/layers/router.py -------------------------------------------------------------------------------- /models/resnet_cifar_moe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPTML-Group/Robust-MoE-CNN/HEAD/models/resnet_cifar_moe.py -------------------------------------------------------------------------------- /models/resnet_cifar_ori.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPTML-Group/Robust-MoE-CNN/HEAD/models/resnet_cifar_ori.py -------------------------------------------------------------------------------- /models/resnet_imagenet_moe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPTML-Group/Robust-MoE-CNN/HEAD/models/resnet_imagenet_moe.py -------------------------------------------------------------------------------- /models/resnet_imagenet_ori.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPTML-Group/Robust-MoE-CNN/HEAD/models/resnet_imagenet_ori.py -------------------------------------------------------------------------------- /models/vgg_moe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPTML-Group/Robust-MoE-CNN/HEAD/models/vgg_moe.py -------------------------------------------------------------------------------- /models/vgg_ori.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPTML-Group/Robust-MoE-CNN/HEAD/models/vgg_ori.py -------------------------------------------------------------------------------- /models/wrn_cifar_moe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPTML-Group/Robust-MoE-CNN/HEAD/models/wrn_cifar_moe.py -------------------------------------------------------------------------------- /models/wrn_cifar_ori.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPTML-Group/Robust-MoE-CNN/HEAD/models/wrn_cifar_ori.py -------------------------------------------------------------------------------- /models/wrn_imagenet_moe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPTML-Group/Robust-MoE-CNN/HEAD/models/wrn_imagenet_moe.py -------------------------------------------------------------------------------- /models/wrn_imagenet_ori.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPTML-Group/Robust-MoE-CNN/HEAD/models/wrn_imagenet_ori.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPTML-Group/Robust-MoE-CNN/HEAD/requirements.txt -------------------------------------------------------------------------------- /torchattacks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPTML-Group/Robust-MoE-CNN/HEAD/torchattacks/__init__.py -------------------------------------------------------------------------------- /torchattacks/attack.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPTML-Group/Robust-MoE-CNN/HEAD/torchattacks/attack.py -------------------------------------------------------------------------------- /torchattacks/attacks/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torchattacks/attacks/_differential_evolution.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPTML-Group/Robust-MoE-CNN/HEAD/torchattacks/attacks/_differential_evolution.py -------------------------------------------------------------------------------- /torchattacks/attacks/apgd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPTML-Group/Robust-MoE-CNN/HEAD/torchattacks/attacks/apgd.py -------------------------------------------------------------------------------- /torchattacks/attacks/apgdt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPTML-Group/Robust-MoE-CNN/HEAD/torchattacks/attacks/apgdt.py -------------------------------------------------------------------------------- /torchattacks/attacks/autoattack.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPTML-Group/Robust-MoE-CNN/HEAD/torchattacks/attacks/autoattack.py -------------------------------------------------------------------------------- /torchattacks/attacks/bim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPTML-Group/Robust-MoE-CNN/HEAD/torchattacks/attacks/bim.py -------------------------------------------------------------------------------- /torchattacks/attacks/cw.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPTML-Group/Robust-MoE-CNN/HEAD/torchattacks/attacks/cw.py -------------------------------------------------------------------------------- /torchattacks/attacks/deepfool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPTML-Group/Robust-MoE-CNN/HEAD/torchattacks/attacks/deepfool.py -------------------------------------------------------------------------------- /torchattacks/attacks/eotpgd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPTML-Group/Robust-MoE-CNN/HEAD/torchattacks/attacks/eotpgd.py -------------------------------------------------------------------------------- /torchattacks/attacks/fab.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPTML-Group/Robust-MoE-CNN/HEAD/torchattacks/attacks/fab.py -------------------------------------------------------------------------------- /torchattacks/attacks/ffgsm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPTML-Group/Robust-MoE-CNN/HEAD/torchattacks/attacks/ffgsm.py -------------------------------------------------------------------------------- /torchattacks/attacks/fgsm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPTML-Group/Robust-MoE-CNN/HEAD/torchattacks/attacks/fgsm.py -------------------------------------------------------------------------------- /torchattacks/attacks/gn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPTML-Group/Robust-MoE-CNN/HEAD/torchattacks/attacks/gn.py -------------------------------------------------------------------------------- /torchattacks/attacks/mifgsm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPTML-Group/Robust-MoE-CNN/HEAD/torchattacks/attacks/mifgsm.py -------------------------------------------------------------------------------- /torchattacks/attacks/multiattack.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPTML-Group/Robust-MoE-CNN/HEAD/torchattacks/attacks/multiattack.py -------------------------------------------------------------------------------- /torchattacks/attacks/onepixel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPTML-Group/Robust-MoE-CNN/HEAD/torchattacks/attacks/onepixel.py -------------------------------------------------------------------------------- /torchattacks/attacks/pgd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPTML-Group/Robust-MoE-CNN/HEAD/torchattacks/attacks/pgd.py -------------------------------------------------------------------------------- /torchattacks/attacks/pgddlr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPTML-Group/Robust-MoE-CNN/HEAD/torchattacks/attacks/pgddlr.py -------------------------------------------------------------------------------- /torchattacks/attacks/pgdl2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPTML-Group/Robust-MoE-CNN/HEAD/torchattacks/attacks/pgdl2.py -------------------------------------------------------------------------------- /torchattacks/attacks/rfgsm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPTML-Group/Robust-MoE-CNN/HEAD/torchattacks/attacks/rfgsm.py -------------------------------------------------------------------------------- /torchattacks/attacks/sparsefool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPTML-Group/Robust-MoE-CNN/HEAD/torchattacks/attacks/sparsefool.py -------------------------------------------------------------------------------- /torchattacks/attacks/square.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPTML-Group/Robust-MoE-CNN/HEAD/torchattacks/attacks/square.py -------------------------------------------------------------------------------- /torchattacks/attacks/tpgd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPTML-Group/Robust-MoE-CNN/HEAD/torchattacks/attacks/tpgd.py -------------------------------------------------------------------------------- /torchattacks/attacks/vanila.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPTML-Group/Robust-MoE-CNN/HEAD/torchattacks/attacks/vanila.py -------------------------------------------------------------------------------- /train_moe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPTML-Group/Robust-MoE-CNN/HEAD/train_moe.py -------------------------------------------------------------------------------- /train_ori.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPTML-Group/Robust-MoE-CNN/HEAD/train_ori.py -------------------------------------------------------------------------------- /utils/adv_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPTML-Group/Robust-MoE-CNN/HEAD/utils/adv_utils.py -------------------------------------------------------------------------------- /utils/eval_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPTML-Group/Robust-MoE-CNN/HEAD/utils/eval_utils.py -------------------------------------------------------------------------------- /utils/general_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPTML-Group/Robust-MoE-CNN/HEAD/utils/general_utils.py -------------------------------------------------------------------------------- /utils/model_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPTML-Group/Robust-MoE-CNN/HEAD/utils/model_utils.py -------------------------------------------------------------------------------- /utils/schedules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPTML-Group/Robust-MoE-CNN/HEAD/utils/schedules.py --------------------------------------------------------------------------------