arXiv reaDer
効率的な生成的敵対的ネットワークの粗密検索
Coarse-to-Fine Searching for Efficient Generative Adversarial Networks
この論文は、効率的な発電機ネットワークを開発するためのニューラルアーキテクチャ検索(NAS)問題を研究します。視覚認識タスクのディープモデルと比較して、生成的敵対的ネットワーク(GAN)は通常、さまざまな複雑な画像生成を実行するように設計されています。最初に、ネットワークパフォーマンスを完全に掘り下げるための、パス、オペレーター、チャネルの3次元を含むジェネレーターネットワークの完全な検索スペースを発見します。莫大な検索コストを削減するために、検索プロセス全体をそれに応じて3つのサブ最適化問題に分割する粗い検索戦略から細かい検索戦略を検討します。さらに、公正なスーパーネットトレーニングアプローチを利用して、すべてのサブネットワークを公正かつ安定して更新できるようにします。ベンチマークでの実験結果は、最先端の方法よりも優れた画質と低い計算コストを発電機ネットワークに提供できることを示しています。たとえば、私たちの方法では、エッジツーシューズデータセット全体で約8GPU時間で24.13FIDスコアの2.56MBモデルを取得し、Urban100データセット全体で10GPU時間で1.49MBモデルを取得できます。 24.94PSNRスコア。
This paper studies the neural architecture search (NAS) problem for developing efficient generator networks. Compared with deep models for visual recognition tasks, generative adversarial network (GAN) are usually designed to conduct various complex image generation. We first discover an intact search space of generator networks including three dimensionalities, i.e., path, operator, channel for fully excavating the network performance. To reduce the huge search cost, we explore a coarse-to-fine search strategy which divides the overall search process into three sub-optimization problems accordingly. In addition, a fair supernet training approach is utilized to ensure that all sub-networks can be updated fairly and stably. Experiments results on benchmarks show that we can provide generator networks with better image quality and lower computational costs over the state-of-the-art methods. For example, with our method, it takes only about 8 GPU hours on the entire edges-to-shoes dataset to get a 2.56 MB model with a 24.13 FID score and 10 GPU hours on the entire Urban100 dataset to get a 1.49 MB model with a 24.94 PSNR score.
updated: Mon Apr 19 2021 11:46:20 GMT+0000 (UTC)
published: Mon Apr 19 2021 11:46:20 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アソシエイト