arXiv reaDer
自己ラベリングと自己注意による統一された生成的敵対的ネットワークトレーニング
A Unified Generative Adversarial Network Training via Self-Labeling and Self-Attention
あらゆるレベルのラベリングを統一的に処理できる新しいGANトレーニングスキームを提案します。私たちのスキームは、手動で定義されたラベルが利用可能な場合にそれを組み込み、それらの間の位置合わせを誘発できる人工ラベルの形式を導入します。人工ラベルを定義するために、ニューラルネットワークジェネレーターをトレーニングして、個別のカテゴリ間よりも、近くの潜在ベクトルを意味的類似性のあるデータにマッピングする方が簡単であるという仮定を利用します。生成されたデータサンプルとそれに対応する人工条件付けラベルを使用して、分類器をトレーニングします。次に、分類子を使用して、実際のデータに自己ラベルを付けます。自己ラベル付けの精度を高めるために、分類器の指数移動平均も使用します。ただし、特にトレーニングの開始時に分類器が間違いを犯す可能性があるため、分類器が高い分類確率スコアを出力する場合にのみ実際のデータサンプルのラベル付けを使用することにより、自己注意によってラベルを改良します。 CIFAR-10、STL-10、SVHNでのアプローチを評価し、自己ラベル付けと自己注意の両方が生成されたデータの品質を一貫して改善することを示します。さらに驚くべきことに、提案されたスキームはクラス条件付きGANよりも優れたパフォーマンスを発揮できることがわかりました。
We propose a novel GAN training scheme that can handle any level of labeling in a unified manner. Our scheme introduces a form of artificial labeling that can incorporate manually defined labels, when available, and induce an alignment between them. To define the artificial labels, we exploit the assumption that neural network generators can be trained more easily to map nearby latent vectors to data with semantic similarities, than across separate categories. We use generated data samples and their corresponding artificial conditioning labels to train a classifier. The classifier is then used to self-label real data. To boost the accuracy of the self-labeling, we also use the exponential moving average of the classifier. However, because the classifier might still make mistakes, especially at the beginning of the training, we also refine the labels through self-attention, by using the labeling of real data samples only when the classifier outputs a high classification probability score. We evaluate our approach on CIFAR-10, STL-10 and SVHN, and show that both self-labeling and self-attention consistently improve the quality of generated data. More surprisingly, we find that the proposed scheme can even outperform class-conditional GANs.
updated: Fri Jun 18 2021 04:40:26 GMT+0000 (UTC)
published: Fri Jun 18 2021 04:40:26 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アソシエイト