クラス条件付けは、離散入力変数に基づいて生成的敵対的ネットワーク(GAN)を制御する直接的な手段を提供します。多くのアプリケーションで必要ですが、クラスラベルによって提供される追加情報は、GAN自体のトレーニングに役立つことさえ期待できます。それどころか、クラス条件付けは限られたデータ設定でモード崩壊を引き起こし、無条件の学習が満足のいく生成能力につながることを観察します。この観察に動機付けられて、無条件の学習を活用することによって観察されたモード崩壊を効果的に防ぐクラス条件付きGAN(cGAN)のトレーニング戦略を提案します。私たちのトレーニング戦略は、無条件のGANから始まり、クラス条件付けをジェネレーターと目的関数に徐々に注入します。限られたデータでcGANをトレーニングするために提案された方法は、安定したトレーニングだけでなく、クラス間で共有される情報の初期段階の活用のおかげで、高品質の画像を生成することにもなります。 4つのデータセットでの包括的な実験で、観測されたモード崩壊の問題を分析します。私たちのアプローチは、最先端の方法や確立されたベースラインと比較して優れた結果を示しています。コードは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