DGL-GAN: Discriminator Guided Learning for GAN Compression

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Tian, Yuesong, Shen, Li, Tian, Xiang, Tao, Dacheng, Li, Zhifeng, Liu, Wei, Chen, Yaowu
Format: Preprint
Published: 2021
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866911810931654656
author Tian, Yuesong
Shen, Li
Tian, Xiang
Tao, Dacheng
Li, Zhifeng
Liu, Wei
Chen, Yaowu
author_facet Tian, Yuesong
Shen, Li
Tian, Xiang
Tao, Dacheng
Li, Zhifeng
Liu, Wei
Chen, Yaowu
contents Generative Adversarial Networks (GANs) with high computation costs, e.g., BigGAN and StyleGAN2, have achieved remarkable results in synthesizing high-resolution images from random noise. Reducing the computation cost of GANs while keeping generating photo-realistic images is a challenging field. In this work, we propose a novel yet simple {\bf D}iscriminator {\bf G}uided {\bf L}earning approach for compressing vanilla {\bf GAN}, dubbed {\bf DGL-GAN}. Motivated by the phenomenon that the teacher discriminator may contain some meaningful information about both real images and fake images, we merely transfer the knowledge from the teacher discriminator via the adversarial interaction between the teacher discriminator and the student generator. We apply DGL-GAN to compress the two most representative large-scale vanilla GANs, i.e., StyleGAN2 and BigGAN. Experiments show that DGL-GAN achieves state-of-the-art (SOTA) results on both StyleGAN2 and BigGAN. Moreover, DGL-GAN is also effective in boosting the performance of original uncompressed GANs. Original uncompressed StyleGAN2 boosted with DGL-GAN achieves FID 2.65 on FFHQ, which achieves a new state-of-the-art performance. Code and models are available at \url{https://github.com/yuesongtian/DGL-GAN}
format Preprint
id arxiv_https___arxiv_org_abs_2112_06502
institution arXiv
publishDate 2021
record_format arxiv
spellingShingle DGL-GAN: Discriminator Guided Learning for GAN Compression
Tian, Yuesong
Shen, Li
Tian, Xiang
Tao, Dacheng
Li, Zhifeng
Liu, Wei
Chen, Yaowu
Computer Vision and Pattern Recognition
Generative Adversarial Networks (GANs) with high computation costs, e.g., BigGAN and StyleGAN2, have achieved remarkable results in synthesizing high-resolution images from random noise. Reducing the computation cost of GANs while keeping generating photo-realistic images is a challenging field. In this work, we propose a novel yet simple {\bf D}iscriminator {\bf G}uided {\bf L}earning approach for compressing vanilla {\bf GAN}, dubbed {\bf DGL-GAN}. Motivated by the phenomenon that the teacher discriminator may contain some meaningful information about both real images and fake images, we merely transfer the knowledge from the teacher discriminator via the adversarial interaction between the teacher discriminator and the student generator. We apply DGL-GAN to compress the two most representative large-scale vanilla GANs, i.e., StyleGAN2 and BigGAN. Experiments show that DGL-GAN achieves state-of-the-art (SOTA) results on both StyleGAN2 and BigGAN. Moreover, DGL-GAN is also effective in boosting the performance of original uncompressed GANs. Original uncompressed StyleGAN2 boosted with DGL-GAN achieves FID 2.65 on FFHQ, which achieves a new state-of-the-art performance. Code and models are available at \url{https://github.com/yuesongtian/DGL-GAN}
title DGL-GAN: Discriminator Guided Learning for GAN Compression
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2112.06502