arXiv reaDer
畳み込みニューラル ネットワークの枝刈り用の 1xN パターン
1xN Pattern for Pruning Convolutional Neural Networks
ネットワーク プルーニングは、畳み込みニューラル ネットワーク (CNN) の複雑さを軽減する上で人気がありますが、モデルの精度を維持しながら、一般的な CPU で大幅な高速化を達成することは未解決の問題です。この論文では、この制限を打破するための新しい 1xN 剪定パターンを提案します。特に、同じ入力チャネル インデックスを持つ連続した N 個の出力カーネルは、1 つのブロックにグループ化されます。これは、プルーニング パターンの基本的なプルーニング粒度として機能します。私たちの 1xN パターンは、重要ではないと考えられるこれらのブロックを削除します。また、最初に出力チャネル次元の重み行列を再配置して精度向上のためにより影響力のあるブロックを導き出し、次に入力チャネル次元の次の層の重みに同様の再配置を適用して、正しい畳み込み操作を保証するフィルター再配置のワークフローも提供します。さらに、1xN プルーニング後の出力計算は、並列化されたブロック単位のベクトル化操作によって実現できるため、一般的な CPU で大幅な高速化が実現します。剪定パターンの有効性は、ILSVRC-2012 での実験で証明されています。たとえば、刈り込み率が 50% で N=4 の場合、MobileNet-V2 のトップ 1 精度で、フィルター刈り込みよりも約 3.0% 改善されます。一方、Cortex-A7 CPU では、ウェイト プルーニングよりも 56.04 ミリ秒の推論節約が得られます。私たちのプロジェクトは https://github.com/lmbxmu/1xN で利用できます。
Though network pruning receives popularity in reducing the complexity of convolutional neural networks (CNNs), it remains an open issue to concurrently maintain model accuracy as well as achieve significant speedups on general CPUs. In this paper, we propose a novel 1xN pruning pattern to break this limitation. In particular, consecutive N output kernels with the same input channel index are grouped into one block, which serves as a basic pruning granularity of our pruning pattern. Our 1xN pattern prunes these blocks considered unimportant. We also provide a workflow of filter rearrangement that first rearranges the weight matrix in the output channel dimension to derive more influential blocks for accuracy improvements and then applies similar rearrangement to the next-layer weights in the input channel dimension to ensure correct convolutional operations. Moreover, the output computation after our 1xN pruning can be realized via a parallelized block-wise vectorized operation, leading to significant speedups on general CPUs. The efficacy of our pruning pattern is proved with experiments on ILSVRC-2012. For example, given the pruning rate of 50% and N=4, our pattern obtains about 3.0% improvements over filter pruning in the top-1 accuracy of MobileNet-V2. Meanwhile, it obtains 56.04ms inference savings on Cortex-A7 CPU over weight pruning. Our project is made available at https://github.com/lmbxmu/1xN.
updated: Sun Oct 16 2022 01:34:36 GMT+0000 (UTC)
published: Mon May 31 2021 05:50:33 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アソシエイト