arXiv reaDer
コンディショニングによる崩壊:限られたデータでクラス条件付きGANをトレーニングする
Collapse by Conditioning: Training Class-conditional GANs with Limited Data
クラス条件付けは、離散入力変数に基づいて生成的敵対的ネットワーク(GAN)を制御する直接的な手段を提供します。多くのアプリケーションで必要ですが、クラスラベルによって提供される追加情報は、GAN自体のトレーニングに役立つと期待することもできます。この信念に反して、クラス条件付けは限られたデータ設定でモード崩壊を引き起こし、無条件の学習が満足のいく生成能力につながることを観察します。この観察に動機付けられて、無条件の学習を活用することにより、観察されたモード崩壊を効果的に防止する条件付きGAN(cGAN)のトレーニング戦略を提案します。私たちのトレーニング戦略は、無条件のGANから始まり、条件付き情報をジェネレーターと目的関数に徐々に注入します。限られたデータでcGANをトレーニングするために提案された方法は、安定したトレーニングだけでなく、クラス間で共有される情報の初期段階の活用のおかげで、高品質の画像を生成することにもなります。 4つのデータセットの包括的な実験で、前述のモード崩壊問題を分析します。私たちのアプローチは、最先端の方法や確立されたベースラインと比較して優れた結果を示しています。コードはhttps://github.com/mshahbazi72/transitional-cGANで入手できます。
Class-conditioning offers a direct means of controlling 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. Contrary to this belief, 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 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 conditional information 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 aforementioned 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
updated: Mon Jan 17 2022 18:59:23 GMT+0000 (UTC)
published: Mon Jan 17 2022 18:59:23 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト