arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2201.06578cs.CVcs.AI
收藏

条件化崩溃:用有限数据训练类条件GANs

Collapse by Conditioning: Training Class-conditional GANs with Limited Data

ETH Zurich(苏黎世联邦理工学院)

Mohamad Shahbazi, Martin Danelljan, Danda Pani Paudel, Luc Van Gool

摘要:

类调节提供了一种基于离散输入变量控制生成对抗网络(GAN)的直接方法。虽然在许多应用中是必要的,但是由类别标签提供的附加信息甚至可以被期望有益于GAN本身的训练。相反,我们观察到,类条件导致模式坍塌在有限的数据设置,无条件的学习导致令人满意的生成能力。基于这一观察,我们提出了一种类条件GANs(cGANs)的训练策略,通过利用无条件学习有效地防止了观察到的模式崩溃。我们的训练策略从一个无条件的GAN开始,并逐渐将类调节注入生成器和目标函数。所提出的用有限数据训练cGAN的方法不仅可以实现稳定的训练,而且可以生成高质量的图像,这要归功于早期利用跨类共享信息。我们分析了观察到的模式坍塌的问题,在综合实验上的四个数据集。与最先进的方法和已建立的基线相比,我们的方法显示出出色的结果。代码可在https://github.com/mshahbazi72/transitional-cGAN上获得

英文摘要:

Class-conditioning offers a direct means to control a Generative Adversarial Network (GAN) based on a discrete input variable. While necessary in many applications, the additional information provided by the class labels could even be expected to benefit the training of the GAN itself. On the contrary, we observe that class-conditioning causes mode collapse in limited data settings, where unconditional learning leads to satisfactory generative ability. Motivated by this observation, we propose a training strategy for class-conditional GANs (cGANs) that effectively prevents the observed mode-collapse by leveraging unconditional learning. Our training strategy starts with an unconditional GAN and gradually injects the class conditioning into the generator and the objective function. The proposed method for training cGANs with limited data results not only in stable training but also in generating high-quality images, thanks to the early-stage exploitation of the shared information across classes. We analyze the observed mode collapse problem in comprehensive experiments on four datasets. Our approach demonstrates outstanding results compared with state-of-the-art methods and established baselines. The code is available at https://github.com/mshahbazi72/transitional-cGAN

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑