arXiv reaDer
An Acceleration Framework for High Resolution Image Synthesis
  Generative Adversarial Networks(GAN)を使用した高解像度画像の合成は困難であり、通常、大容量のメモリと長時間のトレーニングを備えた多数のハイエンドグラフィックカードが必要です。この論文では、高解像度画像を合成するトレーニングプロセスを加速する2段階のフレームワークを提案します。高解像度の画像は、まずトレーニングされたエンコーダーおよびデコーダーネットワークを介して小さなコードに変換されます。潜在空間のコードは、元の高解像度画像の数倍です。次に、コード生成ネットワークをトレーニングして、潜在コードの分布を学習します。このようにして、ジェネレーターは、大きな画像の代わりに小さな潜在コードを生成することのみを学習します。最後に、デコーダネットワークを介して生成された潜在コードを画像空間にデコードし、合成された高解像度画像を出力します。実験結果は、提案された方法がトレーニングプロセスを大幅に加速し、生成されたサンプルの品質を向上させることを示しています。提案されたアクセラレーションフレームワークにより、限られたハードウェアリソースで、より短いトレーニング時間で高解像度の画像を生成できます。提案された加速方法を使用した後、1枚のNVIDIA P100グラフィックカードを使用して、Celeba-HQデータセットで1024 * 1024イメージジェネレーターをトレーニングするのに3日しかかかりません。
Synthesis of high resolution images using Generative Adversarial Networks (GANs) is challenging, which usually requires numbers of high-end graphic cards with large memory and long time of training. In this paper, we propose a two-stage framework to accelerate the training process of synthesizing high resolution images. High resolution images are first transformed to small codes via the trained encoder and decoder networks. The code in latent space is times smaller than the original high resolution images. Then, we train a code generation network to learn the distribution of the latent codes. In this way, the generator only learns to generate small latent codes instead of large images. Finally, we decode the generated latent codes to image space via the decoder networks so as to output the synthesized high resolution images. Experimental results show that the proposed method accelerates the training process significantly and increases the quality of the generated samples. The proposed acceleration framework makes it possible to generate high resolution images using less training time with limited hardware resource. After using the proposed acceleration method, it takes only 3 days to train a 1024 *1024 image generator on Celeba-HQ dataset using just one NVIDIA P100 graphic card.
updated: Mon Sep 09 2019 03:19:25 GMT+0000 (UTC)
published: Mon Sep 09 2019 03:19:25 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)アソシエイト