arXiv reaDer
カーネル密度弁別による生成的敵対的学習
Generative Adversarial Learning via Kernel Density Discrimination
生成的敵対的学習の新しい方法であるカーネル密度識別GAN(KDD GAN)を紹介します。 KDD GANは、データ分布が(ローカル)カーネル密度推定(KDE)を介して明示的に書き込まれる、尤度比最適化問題としてトレーニングを定式化します。これは、対照学習の最近の進歩とKDEとの関係に触発されています。 KDEを機能空間で直接定義し、カーネル機能マッピングの可逆性の要件を放棄します。私たちのアプローチでは、元のGANの定式化のように、特徴は線形分離可能性に対して最適化されなくなりましたが、特徴空間内の分布のより一般的な識別のために最適化されています。特徴表現に関して損失の勾配を分析し、元のヒンジ損失よりも適切に動作することを示します。 CIFAR10とImageNetのスケーリングされたバージョンの両方で、トレーニング損失または正則化項として使用される、提案されたKDEベースの損失を使用して実験を実行します。ネットワークのアーキテクチャを設計することに重点を置いていないため、BigGAN / SA-GANをバックボーンおよびベースラインとして使用します。ベースラインと比較して、FIDに関して生成されたサンプルの品質が10%から40%向上していることを示しています。コードが利用可能になります。
We introduce Kernel Density Discrimination GAN (KDD GAN), a novel method for generative adversarial learning. KDD GAN formulates the training as a likelihood ratio optimization problem where the data distributions are written explicitly via (local) Kernel Density Estimates (KDE). This is inspired by the recent progress in contrastive learning and its relation to KDE. We define the KDEs directly in feature space and forgo the requirement of invertibility of the kernel feature mappings. In our approach, features are no longer optimized for linear separability, as in the original GAN formulation, but for the more general discrimination of distributions in the feature space. We analyze the gradient of our loss with respect to the feature representation and show that it is better behaved than that of the original hinge loss. We perform experiments with the proposed KDE-based loss, used either as a training loss or a regularization term, on both CIFAR10 and scaled versions of ImageNet. We use BigGAN/SA-GAN as a backbone and baseline, since our focus is not to design the architecture of the networks. We show a boost in the quality of generated samples with respect to FID from 10% to 40% compared to the baseline. Code will be made available.
updated: Tue Jul 13 2021 15:52:10 GMT+0000 (UTC)
published: Tue Jul 13 2021 15:52:10 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アソシエイト