arXiv reaDer
教師は教える以上のことをする:画像から画像へのモデルの圧縮
Teachers Do More Than Teach: Compressing Image-to-Image Models
生成的敵対的ネットワーク(GAN)は、忠実度の高い画像の生成に大きな成功を収めていますが、計算コストが非常に高く、メモリ使用量が多いため、効率が低いという問題があります。圧縮GANに関する最近の取り組みは、画質を犠牲にするか、時間のかかる検索プロセスを伴うことにより、より小さなジェネレーターを取得する上で顕著な進歩を示しています。この作業では、知識の蒸留を実行することに加えて、効率的なネットワークアーキテクチャを見つけることができる検索スペースを提供する教師ネットワークを導入することによって、これらの問題に対処することを目指しています。まず、生成モデルの探索空間を再検討し、開始ベースの残余ブロックを生成器に導入します。次に、目標の計算コストを達成するために、教師モデルから学生のアーキテクチャを検索し、検索コストを大幅に削減するワンステップのプルーニングアルゴリズムを提案します。 l1スパース正則化とそれに関連するハイパーパラメータを必要としないため、トレーニング手順が簡素化されます。最後に、Global Kernel Alignment(GKA)という名前のインデックスを介して、教師と生徒の間の機能の類似性を最大化することにより、知識を抽出することを提案します。当社の圧縮ネットワークは、MACなどの計算コストを大幅に削減して、元のモデルと同等またはそれ以上の画像忠実度(FID、mIoU)を実現します。コードはhttps://github.com/snap-research/CATでリリースされます。
Generative Adversarial Networks (GANs) have achieved huge success in generating high-fidelity images, however, they suffer from low efficiency due to tremendous computational cost and bulky memory usage. Recent efforts on compression GANs show noticeable progress in obtaining smaller generators by sacrificing image quality or involving a time-consuming searching process. In this work, we aim to address these issues by introducing a teacher network that provides a search space in which efficient network architectures can be found, in addition to performing knowledge distillation. First, we revisit the search space of generative models, introducing an inception-based residual block into generators. Second, to achieve target computation cost, we propose a one-step pruning algorithm that searches a student architecture from the teacher model and substantially reduces searching cost. It requires no l1 sparsity regularization and its associated hyper-parameters, simplifying the training procedure. Finally, we propose to distill knowledge through maximizing feature similarity between teacher and student via an index named Global Kernel Alignment (GKA). Our compressed networks achieve similar or even better image fidelity (FID, mIoU) than the original models with much-reduced computational cost, e.g., MACs. Code will be released at https://github.com/snap-research/CAT.
updated: Wed Aug 18 2021 07:14:12 GMT+0000 (UTC)
published: Fri Mar 05 2021 04:29:34 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アソシエイト