arXiv reaDer
バッチ統計適応による小さなデータセットからの画像生成
Image Generation From Small Datasets via Batch Statistics Adaptation
  ディープ生成モデルの最近の開発のおかげで、忠実度と多様性の両方を備えた高品質の画像を生成することが可能になりつつあります。ただし、このような生成モデルのトレーニングには大きなデータセットが必要です。必要なデータの量を減らすために、大きなデータセットでトレーニングされた事前トレーニング済みジェネレーターの事前知識を別のドメインの小さなデータセットに転送する新しい方法を提案します。このような事前知識を使用して、モデルは小さなデータセットからは取得できない常識を活用した画像を生成できます。この作業では、ジェネレーターの隠れ層のバッチ統計、スケール、シフトのパラメーターに焦点を当てた新しい方法を提案します。これらのパラメーターのみを監視された方法でトレーニングすることで、ジェネレーターの安定したトレーニングを達成し、データセットが小さい場合(〜100)でも、このメソッドは崩壊することなく以前のメソッドと比較して高品質の画像を生成できます。私たちの結果は、事前に訓練されたジェネレーターで取得したフィルターの多様性が、ターゲットドメインでのパフォーマンスにとって重要であることを示しています。このメソッドを使用すると、元のドメインのパフォーマンスを乱すことなく、事前にトレーニングされたジェネレーターに新しいクラスまたはドメインを追加できます。
Thanks to the recent development of deep generative models, it is becoming possible to generate high-quality images with both fidelity and diversity. However, the training of such generative models requires a large dataset. To reduce the amount of data required, we propose a new method for transferring prior knowledge of the pre-trained generator, which is trained with a large dataset, to a small dataset in a different domain. Using such prior knowledge, the model can generate images leveraging some common sense that cannot be acquired from a small dataset. In this work, we propose a novel method focusing on the parameters for batch statistics, scale and shift, of the hidden layers in the generator. By training only these parameters in a supervised manner, we achieved stable training of the generator, and our method can generate higher quality images compared to previous methods without collapsing, even when the dataset is small (~100). Our results show that the diversity of the filters acquired in the pre-trained generator is important for the performance on the target domain. Our method makes it possible to add a new class or domain to a pre-trained generator without disturbing the performance on the original domain.
updated: Wed Oct 23 2019 08:28:38 GMT+0000 (UTC)
published: Wed Apr 03 2019 05:24:02 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アソシエイト