arXiv reaDer
ScoreMix: 限られたデータで GAN をトレーニングするためのスケーラブルな拡張戦略
ScoreMix: A Scalable Augmentation Strategy for Training GANs with Limited Data
Generative Adversarial Networks (GAN) は、通常、利用可能なトレーニング データが限られている場合にオーバーフィッティングに悩まされます。 GAN トレーニングを容易にするために、現在の方法では、データ固有の拡張技術を使用することが提案されています。有効性にもかかわらず、これらの方法を実用的なアプリケーションに拡張することは困難です。この作業では、さまざまな画像合成タスクのための斬新でスケーラブルなデータ拡張アプローチである ScoreMix を紹介します。最初に、実際のサンプルの凸結合を使用して拡張サンプルを作成します。次に、データスコアのノルム、つまり対数密度関数の勾配を最小化することにより、拡張されたサンプルを最適化します。この手順は、拡張されたサンプルをデータ マニホールドの近くに強制します。スコアを推定するために、マルチスケール スコア マッチングを使用して深い推定ネットワークをトレーニングします。さまざまな画像合成タスクについて、さまざまなデータを使用してスコア推定ネットワークをトレーニングします。ハイパーパラメータの調整やネットワーク アーキテクチャの変更は必要ありません。 ScoreMix メソッドは、データの多様性を効果的に高め、オーバーフィッティングの問題を軽減します。さらに、わずかな修正を加えることで、既存の GAN モデルに簡単に組み込むことができます。多数のタスクに関する実験結果は、ScoreMix メソッドを備えた GAN モデルが大幅な改善を達成することを示しています。
Generative Adversarial Networks (GANs) typically suffer from overfitting when limited training data is available. To facilitate GAN training, current methods propose to use data-specific augmentation techniques. Despite the effectiveness, it is difficult for these methods to scale to practical applications. In this work, we present ScoreMix, a novel and scalable data augmentation approach for various image synthesis tasks. We first produce augmented samples using the convex combinations of the real samples. Then, we optimize the augmented samples by minimizing the norms of the data scores, i.e., the gradients of the log-density functions. This procedure enforces the augmented samples close to the data manifold. To estimate the scores, we train a deep estimation network with multi-scale score matching. For different image synthesis tasks, we train the score estimation network using different data. We do not require the tuning of the hyperparameters or modifications to the network architecture. The ScoreMix method effectively increases the diversity of data and reduces the overfitting problem. Moreover, it can be easily incorporated into existing GAN models with minor modifications. Experimental results on numerous tasks demonstrate that GAN models equipped with the ScoreMix method achieve significant improvements.
updated: Fri Nov 04 2022 01:39:18 GMT+0000 (UTC)
published: Thu Oct 27 2022 02:55:15 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アソシエイト