近年、生成的敵対的ネットワーク(GAN)の使用は、生成的画像モデリングで非常に人気があります。スタイルベースのGANアーキテクチャは、最新の結果を高忠実度の画像合成にもたらしますが、計算上は非常に複雑です。私たちの仕事では、スタイルベースの生成モデルのパフォーマンスの最適化に焦点を当てています。 StyleGAN2の最も計算が難しい部分を分析し、エッジデバイスにスタイルベースの生成ネットワークを展開できるようにするためのジェネレータネットワークの変更を提案します。 MobileStyleGANアーキテクチャを紹介します。これは、StyleGAN2よりもパラメータが3.5倍少なく、計算の複雑さが9.5倍少なく、同等の品質を提供します。
In recent years, the use of Generative Adversarial Networks (GANs) has become very popular in generative image modeling. While style-based GAN architectures yield state-of-the-art results in high-fidelity image synthesis, computationally, they are highly complex. In our work, we focus on the performance optimization of style-based generative models. We analyze the most computationally hard parts of StyleGAN2, and propose changes in the generator network to make it possible to deploy style-based generative networks in the edge devices. We introduce MobileStyleGAN architecture, which has x3.5 fewer parameters and is x9.5 less computationally complex than StyleGAN2, while providing comparable quality.