arXiv reaDer
ACGANの再起動:安定したトレーニングを備えた補助分類器GAN
Rebooting ACGAN: Auxiliary Classifier GANs with Stable Training
条件付き生成的敵対的ネットワーク(cGAN)は、クラス情報をGANに組み込むことにより、リアルな画像を生成します。最も人気のあるcGANの1つは、ソフトマックスクロスエントロピー損失(ACGAN)を備えた補助分類器GANですが、データセット内のクラスの数が増えるにつれて、ACGANのトレーニングが困難になることは広く知られています。 ACGANはまた、多様性に欠ける、簡単に分類できるサンプルを生成する傾向があります。この論文では、ACGANの2つの治療法を紹介します。まず、分類器での勾配の爆発が初期のトレーニングで望ましくない崩壊を引き起こす可能性があり、入力ベクトルを単位超球に投影することで問題を解決できることを確認します。次に、クラスラベル付きデータセットの関係情報を活用するために、データ間クロスエントロピー損失(D2D-CE)を提案します。この基盤の上に、再起動された補助分類器生成的敵対的ネットワーク(ReACGAN)を提案します。実験結果は、ReACGANがCIFAR10、Tiny-ImageNet、CUB200、およびImageNetデータセットで最先端の生成結果を達成することを示しています。また、ReACGANが差別化可能な拡張機能の恩恵を受けていること、およびD2D-CEがStyleGAN2アーキテクチャと調和していることを確認します。モデルの重みと、代表的なcGANの実装を提供するソフトウェアパッケージ、および本書のすべての実験は、https://github.com/POSTECH-CVLab/PyTorch-StudioGANで入手できます。
Conditional Generative Adversarial Networks (cGAN) generate realistic images by incorporating class information into GAN. While one of the most popular cGANs is an auxiliary classifier GAN with softmax cross-entropy loss (ACGAN), it is widely known that training ACGAN is challenging as the number of classes in the dataset increases. ACGAN also tends to generate easily classifiable samples with a lack of diversity. In this paper, we introduce two cures for ACGAN. First, we identify that gradient exploding in the classifier can cause an undesirable collapse in early training, and projecting input vectors onto a unit hypersphere can resolve the problem. Second, we propose the Data-to-Data Cross-Entropy loss (D2D-CE) to exploit relational information in the class-labeled dataset. On this foundation, we propose the Rebooted Auxiliary Classifier Generative Adversarial Network (ReACGAN). The experimental results show that ReACGAN achieves state-of-the-art generation results on CIFAR10, Tiny-ImageNet, CUB200, and ImageNet datasets. We also verify that ReACGAN benefits from differentiable augmentations and that D2D-CE harmonizes with StyleGAN2 architecture. Model weights and a software package that provides implementations of representative cGANs and all experiments in our paper are available at https://github.com/POSTECH-CVLab/PyTorch-StudioGAN.
updated: Mon Nov 01 2021 17:51:33 GMT+0000 (UTC)
published: Mon Nov 01 2021 17:51:33 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アソシエイト