arXiv reaDer
Mind2Mind : transfer learning for GANs
高品質(HQ)画像で生成的敵対的ネットワーク(GAN)をトレーニングするには、重要なコンピューティングリソースが必要です。この要件は、GANのアプリケーション開発のボトルネックを表しています。トレーニング時間を大幅に短縮するGANの転移学習手法を提案します。私たちのアプローチは、元のGANの批評家と生成者の両方の低レベルのレイヤーを凍結することで構成されています。評論家とジェネレーターの内部表現の互換性を確保するために、オートエンコーダー制約を想定しています。この仮定は、順方向および逆方向のパス中に低レベルのレイヤーをバイパスできるため、トレーニング時間の増加を説明します。私たちの方法をベースラインと比較し、トレーニングの大幅な加速を観察します。 StyleGANと比較すると、HQデータセットでは2桁に達する可能性があります。最適な輸送の枠組みの中で、転送されたGANの学習の収束を保証する定理を厳密に証明します。さらに、ソースデータセットとターゲットデータセットの間の距離に関して、トレーニングの収束の正確な範囲を提供します。
Training generative adversarial networks (GANs) on high quality (HQ) images involves important computing resources. This requirement represents a bottleneck for the development of applications of GANs. We propose a transfer learning technique for GANs that significantly reduces training time. Our approach consists of freezing the low-level layers of both the critic and generator of the original GAN. We assume an autoencoder constraint in order to ensure the compatibility of the internal representations of the critic and the generator. This assumption explains the gain in training time as it enables us to bypass the low-level layers during the forward and backward passes. We compare our method to baselines and observe a significant acceleration of the training. It can reach two orders of magnitude on HQ datasets when compared with StyleGAN. We prove rigorously, within the framework of optimal transport, a theorem ensuring the convergence of the learning of the transferred GAN. We moreover provide a precise bound for the convergence of the training in terms of the distance between the source and target dataset.
updated: Tue Oct 20 2020 13:15:27 GMT+0000 (UTC)
published: Thu Jun 27 2019 13:19:29 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)アソシエイト