arXiv reaDer
AdaPruner:適応チャネルプルーニングと有効な重みの継承
AdaPruner: Adaptive Channel Pruning and Effective Weights Inheritance
チャネルプルーニングは、ディープニューラルネットワークの主要な圧縮アプローチの1つです。以前のプルーニング方法は主に重要でないチャネルの識別に焦点を合わせていましたが、チャネルプルーニングは近年のニューラルアーキテクチャ検索の特殊なケースと見なされています。ただし、既存の方法は複雑であるか、最適ではない剪定が行われる傾向があります。この論文では、各層のチャネル数とサブネットワークのワイト継承基準を適応的に決定する剪定フレームワークを提案します。まず、BNレイヤーのスケーリングパラメーターの平均に基づいて、ネットワーク内の各ブロックの重要性を評価します。次に、二分法を使用して、予算を満たすコンパクトなサブネットワークをすばやく見つけます。最後に、現在のアーキテクチャに適合する重み継承基準を適応的かつ効率的に選択し、プルーニングされたネットワークを微調整してパフォーマンスを回復します。 AdaPrunerを使用すると、構造と初期化の重みの両方を考慮して、プルーニングされたネットワークを迅速、正確、効率的に取得できます。現在人気のあるCNNモデル(VGG、ResNet、MobileNetV2)をさまざまな画像分類データセットで整理し、実験結果は提案された方法の有効性を示しています。 ImageNetでは、MobileNetV2の32.8%FLOPを削減し、トップ1の精度はわずか0.62%減少します。これは、以前のすべての最先端のチャネルプルーニング方法を上回っています。コードがリリースされます。
Channel pruning is one of the major compression approaches for deep neural networks. While previous pruning methods have mostly focused on identifying unimportant channels, channel pruning is considered as a special case of neural architecture search in recent years. However, existing methods are either complicated or prone to sub-optimal pruning. In this paper, we propose a pruning framework that adaptively determines the number of each layer's channels as well as the wights inheritance criteria for sub-network. Firstly, evaluate the importance of each block in the network based on the mean of the scaling parameters of the BN layers. Secondly, use the bisection method to quickly find the compact sub-network satisfying the budget. Finally, adaptively and efficiently choose the weight inheritance criterion that fits the current architecture and fine-tune the pruned network to recover performance. AdaPruner allows to obtain pruned network quickly, accurately and efficiently, taking into account both the structure and initialization weights. We prune the currently popular CNN models (VGG, ResNet, MobileNetV2) on different image classification datasets, and the experimental results demonstrate the effectiveness of our proposed method. On ImageNet, we reduce 32.8% FLOPs of MobileNetV2 with only 0.62% decrease for top-1 accuracy, which exceeds all previous state-of-the-art channel pruning methods. The code will be released.
updated: Tue Sep 14 2021 01:52:05 GMT+0000 (UTC)
published: Tue Sep 14 2021 01:52:05 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アソシエイト