├── .gitignore ├── ATMC_exps.png ├── LICENSE ├── README.md ├── cifar ├── dataset.py ├── effnets2 │ ├── mobilenetv1.py │ ├── mobilenetv2.py │ ├── mobilenetv2_abcv2.py │ ├── mobilenetv2_dense.py │ ├── mobilenetv2_lr.py │ ├── mobilenetv2_super.py │ ├── nasnet.py │ ├── shufflenetv1.py │ └── shufflenetv2.py ├── inference_model.py ├── inference_model_resnet101.py ├── inference_model_with_sparse_and_quant.py ├── other_attack_test.py ├── prune_model.py ├── resnet │ ├── resnet.py │ ├── resnet_abcv2.py │ ├── resnet_dense.py │ ├── resnet_lr.py │ └── resnet_super.py ├── train_effnets.py ├── train_proj_admm_quant.py ├── train_proj_admm_quant_resnet101.py ├── util.py ├── util_trts.py ├── util_unnormalize.py └── wrn │ ├── wideresnet.py │ ├── wideresnet_abcv2.py │ ├── wideresnet_dense.py │ ├── wideresnet_lr.py │ └── wideresnet_super.py ├── cifar100 ├── dataset.py ├── inference_effnets.py ├── inference_model.py ├── resnet │ ├── resnet.py │ ├── resnet_abcv2.py │ ├── resnet_dense.py │ ├── resnet_lr.py │ └── resnet_super.py ├── train_effnets.py ├── train_proj_admm_quant.py ├── util.py └── util_trts.py ├── materials ├── ATMC_NeurIPS19_poster.pdf ├── ATMC_full.pdf └── video.link ├── mnist ├── caffelenet │ ├── caffelenet.py │ ├── caffelenet_abcv2.py │ ├── caffelenet_dense.py │ ├── caffelenet_lr.py │ ├── caffelenet_quantforward.py │ └── caffelenet_super.py ├── dataset.py ├── inference_model.py ├── train_proj_admm_quant.py ├── util.py └── util_trts.py ├── super_module └── super_class.py ├── svhn ├── dataset.py ├── inference_model.py ├── inference_model_with_sparse_and_quant.py ├── train_proj_admm_quant.py ├── util.py ├── util_trts.py └── wrn │ ├── wideresnet.py │ ├── wideresnet_abcv2.py │ ├── wideresnet_dense.py │ ├── wideresnet_lr.py │ └── wideresnet_super.py └── utee ├── __init__.py ├── kmeans.py ├── misc.py ├── pruning_tools.py └── quantize.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/.gitignore -------------------------------------------------------------------------------- /ATMC_exps.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/ATMC_exps.png -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/README.md -------------------------------------------------------------------------------- /cifar/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/cifar/dataset.py -------------------------------------------------------------------------------- /cifar/effnets2/mobilenetv1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/cifar/effnets2/mobilenetv1.py -------------------------------------------------------------------------------- /cifar/effnets2/mobilenetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/cifar/effnets2/mobilenetv2.py -------------------------------------------------------------------------------- /cifar/effnets2/mobilenetv2_abcv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/cifar/effnets2/mobilenetv2_abcv2.py -------------------------------------------------------------------------------- /cifar/effnets2/mobilenetv2_dense.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/cifar/effnets2/mobilenetv2_dense.py -------------------------------------------------------------------------------- /cifar/effnets2/mobilenetv2_lr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/cifar/effnets2/mobilenetv2_lr.py -------------------------------------------------------------------------------- /cifar/effnets2/mobilenetv2_super.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/cifar/effnets2/mobilenetv2_super.py -------------------------------------------------------------------------------- /cifar/effnets2/nasnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/cifar/effnets2/nasnet.py -------------------------------------------------------------------------------- /cifar/effnets2/shufflenetv1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/cifar/effnets2/shufflenetv1.py -------------------------------------------------------------------------------- /cifar/effnets2/shufflenetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/cifar/effnets2/shufflenetv2.py -------------------------------------------------------------------------------- /cifar/inference_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/cifar/inference_model.py -------------------------------------------------------------------------------- /cifar/inference_model_resnet101.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/cifar/inference_model_resnet101.py -------------------------------------------------------------------------------- /cifar/inference_model_with_sparse_and_quant.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/cifar/inference_model_with_sparse_and_quant.py -------------------------------------------------------------------------------- /cifar/other_attack_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/cifar/other_attack_test.py -------------------------------------------------------------------------------- /cifar/prune_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/cifar/prune_model.py -------------------------------------------------------------------------------- /cifar/resnet/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/cifar/resnet/resnet.py -------------------------------------------------------------------------------- /cifar/resnet/resnet_abcv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/cifar/resnet/resnet_abcv2.py -------------------------------------------------------------------------------- /cifar/resnet/resnet_dense.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/cifar/resnet/resnet_dense.py -------------------------------------------------------------------------------- /cifar/resnet/resnet_lr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/cifar/resnet/resnet_lr.py -------------------------------------------------------------------------------- /cifar/resnet/resnet_super.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/cifar/resnet/resnet_super.py -------------------------------------------------------------------------------- /cifar/train_effnets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/cifar/train_effnets.py -------------------------------------------------------------------------------- /cifar/train_proj_admm_quant.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/cifar/train_proj_admm_quant.py -------------------------------------------------------------------------------- /cifar/train_proj_admm_quant_resnet101.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/cifar/train_proj_admm_quant_resnet101.py -------------------------------------------------------------------------------- /cifar/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/cifar/util.py -------------------------------------------------------------------------------- /cifar/util_trts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/cifar/util_trts.py -------------------------------------------------------------------------------- /cifar/util_unnormalize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/cifar/util_unnormalize.py -------------------------------------------------------------------------------- /cifar/wrn/wideresnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/cifar/wrn/wideresnet.py -------------------------------------------------------------------------------- /cifar/wrn/wideresnet_abcv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/cifar/wrn/wideresnet_abcv2.py -------------------------------------------------------------------------------- /cifar/wrn/wideresnet_dense.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/cifar/wrn/wideresnet_dense.py -------------------------------------------------------------------------------- /cifar/wrn/wideresnet_lr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/cifar/wrn/wideresnet_lr.py -------------------------------------------------------------------------------- /cifar/wrn/wideresnet_super.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/cifar/wrn/wideresnet_super.py -------------------------------------------------------------------------------- /cifar100/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/cifar100/dataset.py -------------------------------------------------------------------------------- /cifar100/inference_effnets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/cifar100/inference_effnets.py -------------------------------------------------------------------------------- /cifar100/inference_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/cifar100/inference_model.py -------------------------------------------------------------------------------- /cifar100/resnet/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/cifar100/resnet/resnet.py -------------------------------------------------------------------------------- /cifar100/resnet/resnet_abcv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/cifar100/resnet/resnet_abcv2.py -------------------------------------------------------------------------------- /cifar100/resnet/resnet_dense.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/cifar100/resnet/resnet_dense.py -------------------------------------------------------------------------------- /cifar100/resnet/resnet_lr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/cifar100/resnet/resnet_lr.py -------------------------------------------------------------------------------- /cifar100/resnet/resnet_super.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/cifar100/resnet/resnet_super.py -------------------------------------------------------------------------------- /cifar100/train_effnets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/cifar100/train_effnets.py -------------------------------------------------------------------------------- /cifar100/train_proj_admm_quant.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/cifar100/train_proj_admm_quant.py -------------------------------------------------------------------------------- /cifar100/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/cifar100/util.py -------------------------------------------------------------------------------- /cifar100/util_trts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/cifar100/util_trts.py -------------------------------------------------------------------------------- /materials/ATMC_NeurIPS19_poster.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/materials/ATMC_NeurIPS19_poster.pdf -------------------------------------------------------------------------------- /materials/ATMC_full.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/materials/ATMC_full.pdf -------------------------------------------------------------------------------- /materials/video.link: -------------------------------------------------------------------------------- 1 | https://youtu.be/DiBzabLY1P8 -------------------------------------------------------------------------------- /mnist/caffelenet/caffelenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/mnist/caffelenet/caffelenet.py -------------------------------------------------------------------------------- /mnist/caffelenet/caffelenet_abcv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/mnist/caffelenet/caffelenet_abcv2.py -------------------------------------------------------------------------------- /mnist/caffelenet/caffelenet_dense.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/mnist/caffelenet/caffelenet_dense.py -------------------------------------------------------------------------------- /mnist/caffelenet/caffelenet_lr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/mnist/caffelenet/caffelenet_lr.py -------------------------------------------------------------------------------- /mnist/caffelenet/caffelenet_quantforward.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/mnist/caffelenet/caffelenet_quantforward.py -------------------------------------------------------------------------------- /mnist/caffelenet/caffelenet_super.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/mnist/caffelenet/caffelenet_super.py -------------------------------------------------------------------------------- /mnist/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/mnist/dataset.py -------------------------------------------------------------------------------- /mnist/inference_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/mnist/inference_model.py -------------------------------------------------------------------------------- /mnist/train_proj_admm_quant.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/mnist/train_proj_admm_quant.py -------------------------------------------------------------------------------- /mnist/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/mnist/util.py -------------------------------------------------------------------------------- /mnist/util_trts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/mnist/util_trts.py -------------------------------------------------------------------------------- /super_module/super_class.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/super_module/super_class.py -------------------------------------------------------------------------------- /svhn/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/svhn/dataset.py -------------------------------------------------------------------------------- /svhn/inference_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/svhn/inference_model.py -------------------------------------------------------------------------------- /svhn/inference_model_with_sparse_and_quant.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/svhn/inference_model_with_sparse_and_quant.py -------------------------------------------------------------------------------- /svhn/train_proj_admm_quant.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/svhn/train_proj_admm_quant.py -------------------------------------------------------------------------------- /svhn/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/svhn/util.py -------------------------------------------------------------------------------- /svhn/util_trts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/svhn/util_trts.py -------------------------------------------------------------------------------- /svhn/wrn/wideresnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/svhn/wrn/wideresnet.py -------------------------------------------------------------------------------- /svhn/wrn/wideresnet_abcv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/svhn/wrn/wideresnet_abcv2.py -------------------------------------------------------------------------------- /svhn/wrn/wideresnet_dense.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/svhn/wrn/wideresnet_dense.py -------------------------------------------------------------------------------- /svhn/wrn/wideresnet_lr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/svhn/wrn/wideresnet_lr.py -------------------------------------------------------------------------------- /svhn/wrn/wideresnet_super.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/svhn/wrn/wideresnet_super.py -------------------------------------------------------------------------------- /utee/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utee/kmeans.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/utee/kmeans.py -------------------------------------------------------------------------------- /utee/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/utee/misc.py -------------------------------------------------------------------------------- /utee/pruning_tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/utee/pruning_tools.py -------------------------------------------------------------------------------- /utee/quantize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/ATMC/HEAD/utee/quantize.py --------------------------------------------------------------------------------