arXiv reaDer
GAN圧縮のためのオンラインマルチグラニュラリティ蒸留
Online Multi-Granularity Distillation for GAN Compression
生成的敵対的ネットワーク(GAN)は、優れた画像を生成する上で広く成功していることを目撃していますが、膨大な計算コストと膨大なメモリ使用量のために、リソースに制約のあるデバイスに展開するのは面倒です。 GANの圧縮に関する最近の取り組みは目覚ましい成果を上げていますが、モデルの冗長性が存在する可能性があり、さらに圧縮することができます。この問題を解決するために、軽量のGANを取得するための新しいオンラインマルチグラニュラリティ蒸留(OMGD)スキームを提案します。これは、計算量の少ない高忠実度の画像の生成に貢献します。私たちは、GAN指向の圧縮のための単一段階のオンライン蒸留を普及させる最初の試みを提供します。そこでは、漸進的に促進された教師ジェネレーターが、弁別子のないベースの学生ジェネレーターを改良するのに役立ちます。補完的な教師ジェネレーターとネットワーク層は、多様な次元からの視覚的忠実度を高めるための包括的でマルチグラニュラリティの概念を提供します。 4つのベンチマークデータセットでの実験結果は、OMGDが画質を損なうことなくPix2PixとCycleGANで40倍のMACと82.5倍のパラメーターを圧縮することに成功したことを示しています。これは、OMGDが、リソースに制約のあるデバイスにリアルタイムの画像変換を展開するための実行可能なソリューションを提供することを示しています。コードとモデルは、https://github.com/bytedance/OMGDで公開されています。
Generative Adversarial Networks (GANs) have witnessed prevailing success in yielding outstanding images, however, they are burdensome to deploy on resource-constrained devices due to ponderous computational costs and hulking memory usage. Although recent efforts on compressing GANs have acquired remarkable results, they still exist potential model redundancies and can be further compressed. To solve this issue, we propose a novel online multi-granularity distillation (OMGD) scheme to obtain lightweight GANs, which contributes to generating high-fidelity images with low computational demands. We offer the first attempt to popularize single-stage online distillation for GAN-oriented compression, where the progressively promoted teacher generator helps to refine the discriminator-free based student generator. Complementary teacher generators and network layers provide comprehensive and multi-granularity concepts to enhance visual fidelity from diverse dimensions. Experimental results on four benchmark datasets demonstrate that OMGD successes to compress 40x MACs and 82.5X parameters on Pix2Pix and CycleGAN, without loss of image quality. It reveals that OMGD provides a feasible solution for the deployment of real-time image translation on resource-constrained devices. Our code and models are made public at: https://github.com/bytedance/OMGD.
updated: Thu Aug 26 2021 07:25:46 GMT+0000 (UTC)
published: Mon Aug 16 2021 05:49:50 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アソシエイト