生成的敵対的ネットワーク(GAN)は、最も強力な生成モデルの1つですが、トレーニングには常に大規模でバランスの取れたデータセットが必要です。従来のGANは、非常に不均衡なデータセットでマイノリティクラスの画像を生成するためには適用できません。この問題を軽減するためにBalancingGAN(BAGAN)が提案されていますが、花や細胞など、異なるクラスの画像が類似している場合は不安定です。この作業では、ラベル付けされた潜在ベクトルを分散させるための中間埋め込みモデルを備えた教師ありオートエンコーダを提案します。改良されたオートエンコーダの初期化により、勾配ペナルティを伴うBAGANのアーキテクチャ(BAGAN-GP)も構築します。私たちが提案するモデルは、元のBAGANの不安定な問題を克服し、高品質の世代にすばやく収束します。私たちのモデルは、MNIST Fashionの不均衡なスケールダウンバージョン、CIFAR-10、および1つの小規模な医用画像データセットで高性能を実現します。
Generative adversarial networks (GANs) are one of the most powerful generative models, but always require a large and balanced dataset to train. Traditional GANs are not applicable to generate minority-class images in a highly imbalanced dataset. Balancing GAN (BAGAN) is proposed to mitigate this problem, but it is unstable when images in different classes look similar, e.g. flowers and cells. In this work, we propose a supervised autoencoder with an intermediate embedding model to disperse the labeled latent vectors. With the improved autoencoder initialization, we also build an architecture of BAGAN with gradient penalty (BAGAN-GP). Our proposed model overcomes the unstable issue in original BAGAN and converges faster to high quality generations. Our model achieves high performance on the imbalanced scale-down version of MNIST Fashion, CIFAR-10, and one small-scale medical image dataset.