arXiv reaDer
GANトレーニング用の既成モデルのアンサンブル
Ensembling Off-the-shelf Models for GAN Training
大規模なトレーニングの出現により、強力な視覚認識モデルの宝庫が生まれました。ただし、GANなどの生成モデルは、従来、教師なしの方法でゼロからトレーニングされてきました。事前にトレーニングされたビジョンモデルの大規模なバンクからの集合的な「知識」を活用して、GANトレーニングを改善できますか?もしそうなら、選択するモデルが非常に多いので、どのモデルを選択する必要があり、どのようにそれらが最も効果的ですか?事前にトレーニングされたコンピュータービジョンモデルは、弁別子のアンサンブルで使用すると、パフォーマンスを大幅に向上させることができることがわかりました。特に、選択したモデルの特定のサブセットは、パフォーマンスに大きく影響します。事前にトレーニングされたモデル埋め込みで実際のサンプルと偽のサンプルの間の線形分離可能性を調査し、最も正確なモデルを選択し、それをディスクリミネーターアンサンブルに段階的に追加することにより、効果的な選択メカニズムを提案します。興味深いことに、私たちの方法は、限られたデータと大規模な設定の両方でGANトレーニングを改善できます。トレーニングサンプルが10kしかない場合、LSUN CatのFIDは、160万枚の画像でトレーニングされたStyleGAN2と一致します。完全なデータセットでは、私たちの方法は、LSUNの猫、教会、馬のカテゴリでFIDを1.5倍から2倍改善します。
The advent of large-scale training has produced a cornucopia of powerful visual recognition models. However, generative models, such as GANs, have traditionally been trained from scratch in an unsupervised manner. Can the collective "knowledge" from a large bank of pretrained vision models be leveraged to improve GAN training? If so, with so many models to choose from, which one(s) should be selected, and in what manner are they most effective? We find that pretrained computer vision models can significantly improve performance when used in an ensemble of discriminators. Notably, the particular subset of selected models greatly affects performance. We propose an effective selection mechanism, by probing the linear separability between real and fake samples in pretrained model embeddings, choosing the most accurate model, and progressively adding it to the discriminator ensemble. Interestingly, our method can improve GAN training in both limited data and large-scale settings. Given only 10k training samples, our FID on LSUN Cat matches the StyleGAN2 trained on 1.6M images. On the full dataset, our method improves FID by 1.5x to 2x on cat, church, and horse categories of LSUN.
updated: Tue Jan 18 2022 18:59:46 GMT+0000 (UTC)
published: Thu Dec 16 2021 18:59:50 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アソシエイト