UGC: Unified GAN Compression for Efficient Image-to-Image Translation

Yuxi Ren,Jie Wu,Peng Zhang,Manlin Zhang,Xuefeng Xiao,Qian He,Rui Wang,Min Zheng,Xin Pan
DOI: https://doi.org/10.1109/iccv51070.2023.01585
2023-01-01
Abstract:Recent years have witnessed the prevailing progress of Generative Adversarial Networks (GANs) in image-toimage translation. However, the success of these GAN models hinges on ponderous computational costs and laborexpensive training data. Current efficient GAN learning techniques often fall into two orthogonal aspects: i) model slimming via reduced calculation costs; ii) data/labelefficient learning with fewer training data/labels. To combine the best of both worlds, we propose a new learning paradigm, Unified GAN Compression (UGC), with a unified optimization objective to seamlessly prompt the synergy of model-efficient and label-efficient learning. UGC sets up semi-supervised-driven network architecture search and adaptive online semi-supervised distillation stages sequentially, which formulates a heterogeneous mutual learning scheme to obtain an architecture-flexible, label-efficient, and performance-excellent model. Extensive experiments demonstrate that UGC obtains state-of-the-art lightweight models even with less than 50% labels. UGC that compresses 40x MACs can achieve 21.43 FID on edges.shoes with 25% labels, which even outperforms the original model with 100% labels by 2.75 FID.
What problem does this paper attempt to address?