AutoReCon: Neural Architecture Search-based Reconstruction for Data-free Compression

AutoReCon: Neural Architecture Search-based Reconstruction for Data-free Compression

Baozhou Zhu, Peter Hofstee, Johan Peltenburg, Jinho Lee, Zaid Alars

Proceedings of the Thirtieth International Joint Conference on Artificial Intelligence
Main Track. Pages 3470-3476. https://doi.org/10.24963/ijcai.2021/478

Data-free compression raises a new challenge because the original training dataset for a pre-trained model to be compressed is not available due to privacy or transmission issues. Thus, a common approach is to compute a reconstructed training dataset before compression. The current reconstruction methods compute the reconstructed training dataset with a generator by exploiting information from the pre-trained model. However, current reconstruction methods focus on extracting more information from the pre-trained model but do not leverage network engineering. This work is the first to consider network engineering as an approach to design the reconstruction method. Specifically, we propose the AutoReCon method, which is a neural architecture search-based reconstruction method. In the proposed AutoReCon method, the generator architecture is designed automatically given the pre-trained model for reconstruction. Experimental results show that using generators discovered by the AutoRecon method always improve the performance of data-free compression.
Keywords:
Machine Learning: Deep Learning
Machine Learning: Classification