Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Data-Free Network Quantization With Adversarial Knowledge Distillation

About

Network quantization is an essential procedure in deep learning for development of efficient fixed-point inference models on mobile or edge platforms. However, as datasets grow larger and privacy regulations become stricter, data sharing for model compression gets more difficult and restricted. In this paper, we consider data-free network quantization with synthetic data. The synthetic data are generated from a generator, while no data are used in training the generator and in quantization. To this end, we propose data-free adversarial knowledge distillation, which minimizes the maximum distance between the outputs of the teacher and the (quantized) student for any adversarial samples from a generator. To generate adversarial samples similar to the original data, we additionally propose matching statistics from the batch normalization layers for generated data and the original data in the teacher. Furthermore, we show the gain of producing diverse adversarial samples by using multiple generators and multiple students. Our experiments show the state-of-the-art data-free model compression and quantization results for (wide) residual networks and MobileNet on SVHN, CIFAR-10, CIFAR-100, and Tiny-ImageNet datasets. The accuracy losses compared to using the original datasets are shown to be very minimal.

Yoojin Choi, Jihwan Choi, Mostafa El-Khamy, Jungwon Lee• 2020

Related benchmarks

TaskDatasetResultRank
Image ClassificationCIFAR-100 (val)
Accuracy77.01
661
Image ClassificationCIFAR10 (test)
Accuracy94.61
585
Image ClassificationCIFAR100 (test)
Top-1 Accuracy77.01
377
Image ClassificationTinyImageNet (test)
Accuracy63.73
366
Image ClassificationCIFAR100
Accuracy54.43
331
Knowledge DistillationCIFAR-10
Accuracy92.01
12
Knowledge DistillationCIFAR-100
Accuracy64.79
12
Data-free QuantizationCIFAR10
Accuracy92.66
10
Data-free QuantizationCIFAR100
Accuracy59.42
10
Image ClassificationCIFAR-100 Data-Free (test)
Accuracy77.01
1
Showing 10 of 10 rows

Other info

Follow for update