arXiv reaDer
GAN圧縮における弁別器の再検討:生成器-弁別器協調圧縮スキーム
Revisiting Discriminator in GAN Compression: A Generator-discriminator Cooperative Compression Scheme
最近、GAN圧縮の一連のアルゴリズムが検討されました。これは、リソースに制約のあるエッジデバイスにGANを展開する際に、計算のオーバーヘッドとメモリ使用量を大幅に削減することを目的としています。ただし、既存のGAN圧縮作業のほとんどは、ジェネレーターを圧縮する方法にのみ焦点を当てており、ディスクリミネーターを考慮に入れていません。この作業では、GAN圧縮における弁別器の役割を再検討し、GCCと呼ばれるGAN圧縮用の新しいジェネレーター-弁別器協調圧縮スキームを設計します。 GCC内では、選択的アクティブ化ディスクリミネーターが、ローカル容量制約とグローバル調整制約に従って畳み込みチャネルを自動的に選択してアクティブ化します。これにより、敵対者のトレーニング中に軽量ジェネレーターとのナッシュ均衡を維持し、モードの崩壊を回避できます。元のジェネレーターとディスクリミネーターもゼロから最適化され、教師モデルとして再生され、剪定されたジェネレーターと選択的アクティベーションディスクリミネーターを段階的に改良します。新しいオンライン共同蒸留スキームは、教師ジェネレーターとディスクリミネーターの中間機能を最大限に活用して、軽量ジェネレーターのパフォーマンスをさらに向上させるように設計されています。さまざまなGANベースの生成タスクに関する広範な実験により、GCCの有効性と一般化が実証されています。その中でも、GCCは、画像変換タスクで同等のパフォーマンスを維持しながら、80%の計算コストの削減に貢献します。コードとモデルはhttps://github.com/SJLeo/GCCで入手できます。
Recently, a series of algorithms have been explored for GAN compression, which aims to reduce tremendous computational overhead and memory usages when deploying GANs on resource-constrained edge devices. However, most of the existing GAN compression work only focuses on how to compress the generator, while fails to take the discriminator into account. In this work, we revisit the role of discriminator in GAN compression and design a novel generator-discriminator cooperative compression scheme for GAN compression, termed GCC. Within GCC, a selective activation discriminator automatically selects and activates convolutional channels according to a local capacity constraint and a global coordination constraint, which help maintain the Nash equilibrium with the lightweight generator during the adversarial training and avoid mode collapse. The original generator and discriminator are also optimized from scratch, to play as a teacher model to progressively refine the pruned generator and the selective activation discriminator. A novel online collaborative distillation scheme is designed to take full advantage of the intermediate feature of the teacher generator and discriminator to further boost the performance of the lightweight generator. Extensive experiments on various GAN-based generation tasks demonstrate the effectiveness and generalization of GCC. Among them, GCC contributes to reducing 80% computational costs while maintains comparable performance in image translation tasks. Our code and models are available at https://github.com/SJLeo/GCC.
updated: Tue Nov 09 2021 07:34:21 GMT+0000 (UTC)
published: Wed Oct 27 2021 13:54:55 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アソシエイト