arXiv reaDer
知識蒸留によるコンパクトなCNN構造学習
Compact CNN Structure Learning by Knowledge Distillation
深い畳み込みニューラルネットワーク(CNN)を圧縮するという概念は、組み込みデバイスで限られた計算、電力、およびメモリリソースを使用するために不可欠です。ただし、既存の方法では、コンピュータービジョンタスクの推論精度が低下するという犠牲を払って、この目的を達成します。このような欠点に対処するために、圧縮とパフォーマンスのトレードオフをより適切に制御しながら、知識の蒸留とカスタマイズ可能なブロック単位の最適化を活用して軽量のCNN構造を学習するフレームワークを提案します。推論ごとの浮動小数点演算(FLOP)やモデルパラメータなど、特定のリソースの制約を考慮すると、この方法では、より優れた推論精度を実現しながら、最先端のネットワーク圧縮が実現します。包括的な評価では、私たちの方法が効果的で堅牢であり、トレーニングのオーバーヘッドがごくわずかで、さまざまなネットワークアーキテクチャとデータセットの結果と一致していることを示しています。特に、すでにコンパクトなネットワークMobileNet_v2の場合、この方法では、ベースラインネットワークよりも1.05%優れたモデルパフォーマンスを実現しながら、FLOPとモデルパラメーターの点でそれぞれ最大2倍と5.2倍優れたモデル圧縮を提供します。
The concept of compressing deep Convolutional Neural Networks (CNNs) is essential to use limited computation, power, and memory resources on embedded devices. However, existing methods achieve this objective at the cost of a drop in inference accuracy in computer vision tasks. To address such a drawback, we propose a framework that leverages knowledge distillation along with customizable block-wise optimization to learn a lightweight CNN structure while preserving better control over the compression-performance tradeoff. Considering specific resource constraints, e.g., floating-point operations per inference (FLOPs) or model-parameters, our method results in a state of the art network compression while being capable of achieving better inference accuracy. In a comprehensive evaluation, we demonstrate that our method is effective, robust, and consistent with results over a variety of network architectures and datasets, at negligible training overhead. In particular, for the already compact network MobileNet_v2, our method offers up to 2x and 5.2x better model compression in terms of FLOPs and model-parameters, respectively, while getting 1.05% better model performance than the baseline network.
updated: Mon Apr 19 2021 10:34:22 GMT+0000 (UTC)
published: Mon Apr 19 2021 10:34:22 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アソシエイト