arXiv reaDer
フィルタ内のフィルタの剪定
Pruning Filter in Filter
剪定は、現代のニューラルネットワークを圧縮および加速するための非常に強力で効果的な手法になりました。既存の剪定方法は、フィルター剪定(FP)と重み剪定(WP)の2つのカテゴリに分類できます。 FPはハードウェアの互換性で勝ちますが、WPと比較して圧縮率では負けます。両方の方法の強度を収束させるために、フィルター内のフィルターを剪定することを提案します。具体的には、フィルターF∈R^ C×K×KをK×Kストライプ、つまり1×1フィルター∈R^ Cとして扱い、フィルター全体ではなくストライプを剪定することで、従来よりも細かい粒度を実現できます。ハードウェアに優しい一方でFP。この方法をSWP(Stripe-Wise Pruning)と呼びます。 SWPは、フィルタースケルトンと呼ばれる新しい学習可能なマトリックスを導入することによって実装されます。このマトリックスの値は、各フィルターの形状を反映しています。最近のいくつかの研究では、剪定されたアーキテクチャが継承された重要な重みよりも重要であることが示されているため、単一のフィルターのアーキテクチャ、つまり形状も重要であると主張します。広範な実験を通じて、SWPが以前のFPベースの方法と比較してより効果的であり、明らかな精度の低下なしにCIFAR-10およびImageNetデータセットで最先端の剪定比を達成することを示します。コードはhttps://github.com/fxmeng/Pruning-Filter-in-Filterで入手できます。
Pruning has become a very powerful and effective technique to compress and accelerate modern neural networks. Existing pruning methods can be grouped into two categories: filter pruning (FP) and weight pruning (WP). FP wins at hardware compatibility but loses at the compression ratio compared with WP. To converge the strength of both methods, we propose to prune the filter in the filter. Specifically, we treat a filter F ∈R^C×K×K as K ×K stripes, i.e., 1×1 filters ∈R^C, then by pruning the stripes instead of the whole filter, we can achieve finer granularity than traditional FP while being hardware friendly. We term our method as SWP (Stripe-Wise Pruning). SWP is implemented by introducing a novel learnable matrix called Filter Skeleton, whose values reflect the shape of each filter. As some recent work has shown that the pruned architecture is more crucial than the inherited important weights, we argue that the architecture of a single filter, i.e., the shape, also matters. Through extensive experiments, we demonstrate that SWP is more effective compared to the previous FP-based methods and achieves the state-of-art pruning ratio on CIFAR-10 and ImageNet datasets without obvious accuracy drop. Code is available at https://github.com/fxmeng/Pruning-Filter-in-Filter
updated: Wed Dec 09 2020 08:35:21 GMT+0000 (UTC)
published: Wed Sep 30 2020 03:35:16 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アソシエイト