论文标题
提炼便携式生成对抗网络,用于图像翻译
Distilling portable Generative Adversarial Networks for Image Translation
论文作者
论文摘要
尽管生成的对抗网络(GAN)已被广泛用于各种图像到图像翻译任务中,但由于其沉重的计算和存储成本,它们几乎不得将其应用于移动设备。传统的网络压缩方法着重于视觉识别任务,但从未处理生成任务。受知识蒸馏的启发,较少参数的学生生成器通过继承原始重型教师生成器的低级和高级信息来训练。为了促进学生生成器的能力,我们包括一个学生歧视者,以测量真实图像和学生和教师生成器产生的图像之间的距离。因此,建立了对抗性学习过程,以优化学生发生者和学生歧视者。通过在基准数据集上进行实验的定性和定量分析表明,所提出的方法可以学习具有较强性能的便携式生成模型。
Despite Generative Adversarial Networks (GANs) have been widely used in various image-to-image translation tasks, they can be hardly applied on mobile devices due to their heavy computation and storage cost. Traditional network compression methods focus on visually recognition tasks, but never deal with generation tasks. Inspired by knowledge distillation, a student generator of fewer parameters is trained by inheriting the low-level and high-level information from the original heavy teacher generator. To promote the capability of student generator, we include a student discriminator to measure the distances between real images, and images generated by student and teacher generators. An adversarial learning process is therefore established to optimize student generator and student discriminator. Qualitative and quantitative analysis by conducting experiments on benchmark datasets demonstrate that the proposed method can learn portable generative models with strong performance.