中文

TinyGAN:蒸馏 BigGAN 用于条件图像生成

计算机视觉与模式识别 2020-09-30 v1

摘要

生成对抗网络(GANs)已成为生成式图像建模的有力方法。然而,GANs 以训练不稳定著称,尤其在大规模复杂数据集上。近期 BigGAN 的工作显著提升了 ImageNet 上图像生成质量,但其模型庞大,难以部署于资源受限设备。为缩减模型规模,我们提出一种黑盒知识蒸馏框架用于压缩 GANs,其具备稳定高效的训练过程。以 BigGAN 为教师网络,我们训练出小得多的学生网络来模仿其功能,在 Inception 与 FID 分数上取得有竞争力性能,且生成器参数减少 16×16\times

关键词

引用

@article{arxiv.2009.13829,
  title  = {TinyGAN: Distilling BigGAN for Conditional Image Generation},
  author = {Ting-Yun Chang and Chi-Jen Lu},
  journal= {arXiv preprint arXiv:2009.13829},
  year   = {2020}
}

备注

accepted by ACCV 2020