arXiv reaDer
データ効率の高いGANトレーニングのための拡張対応の自己監視
Augmentation-Aware Self-Supervision for Data-Efficient GAN Training
限られたデータで敵対的生成ネットワーク(GAN)をトレーニングすることは価値がありますが、そのような状況では弁別器が過剰適合する傾向があるため、困難です。最近提案された弁別器用の微分可能なデータ拡張技術は、GANのトレーニングのデータ効率の向上を示しています。ただし、単純なデータ拡張は、ディスクリミネーターへの拡張に望ましくない不変性をもたらします。不変性は、弁別器の表現学習能力を低下させ、それによって生成器の生成モデリング性能に影響を与える可能性があります。データ拡張の利点を継承しながら不変性を軽減するために、拡張された元のデータが与えられた場合に拡張のパラメータを予測する、拡張を意識した新しい自己監視型弁別器を提案します。さらに、予測タスクは、トレーニング中に異なるため、実際のデータと生成されたデータを区別できるようにする必要があります。さらに、拡張が予測可能な実際のデータを生成することにより、生成者が提案された弁別子から学習することをお勧めします。提案された方法を、CIFAR-10/100およびいくつかのローショットデータセットのクラス条件付きBigGANおよび無条件StyleGAN2アーキテクチャ全体で最新のものとそれぞれ比較します。実験結果は、データ効率の高いGANをトレーニングするための競合する方法よりも、私たちの方法の生成パフォーマンスが大幅に向上していることを示しています。
Training generative adversarial networks (GANs) with limited data is valuable but challenging because discriminators are prone to over-fitting in such situations. Recently proposed differentiable data augmentation techniques for discriminators demonstrate improved data efficiency of training GANs. However, the naive data augmentation introduces undesired invariance to augmentation into the discriminator. The invariance may degrade the representation learning ability of the discriminator, thereby affecting the generative modeling performance of the generator. To mitigate the invariance while inheriting the benefits of data augmentation, we propose a novel augmentation-aware self-supervised discriminator that predicts the parameter of augmentation given the augmented and original data. Moreover, the prediction task is required to distinguishable between real data and generated data since they are different during training. We further encourage the generator to learn from the proposed discriminator by generating augmentation-predictable real data. We compare the proposed method with state-of-the-arts across the class-conditional BigGAN and unconditional StyleGAN2 architectures on CIFAR-10/100 and several low-shot datasets, respectively. Experimental results show a significantly improved generation performance of our method over competing methods for training data-efficient GANs.
updated: Tue May 31 2022 10:35:55 GMT+0000 (UTC)
published: Tue May 31 2022 10:35:55 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アソシエイト