arXiv reaDer
重み依存性に基づく多基準によるチャネルプルーニング
Channel Pruning via Multi-Criteria based on Weight Dependency
チャネルプルーニングは、ConvNetの圧縮におけるその有効性を実証しています。多くの関連技術では、出力フィーチャマップの重要性は、関連するフィルタによってのみ決定されます。ただし、これらのメソッドは、フィーチャマップが削除されると消える次のレイヤーの重みのごく一部を無視します。彼らは体重依存の現象を無視します。さらに、多くの剪定方法は、評価に1つの基準のみを使用し、試行錯誤の方法で剪定構造と精度のスイートスポットを見つけます。これには時間がかかる場合があります。この論文では、事前に訓練されたモデルを直接圧縮できる、重み依存性に基づく多基準CPMCを介したチャネルプルーニングアルゴリズムを提案しました。 CPMCは、関連する重み値、計算コスト、パラメーター量など、3つの側面でチャネルの重要性を定義します。重み依存の現象によると、CPMCは、関連するフィルターと次のレイヤーの対応する部分重みを評価することにより、チャネルの重要性を取得します。次に、CPMCはグローバル正規化を使用してクロスレイヤー比較を実現します。最後に、CPMCは、グローバルランキングによって重要性の低いチャネルを削除します。 CPMCは、さまざまな画像分類データセットでVGGNet、ResNet、DenseNetなどのさまざまなCNNモデルを圧縮できます。広範な実験により、CPMCは他のCPMCよりも大幅に優れていることが示されています。
Channel pruning has demonstrated its effectiveness in compressing ConvNets. In many related arts, the importance of an output feature map is only determined by its associated filter. However, these methods ignore a small part of weights in the next layer which disappears as the feature map is removed. They ignore the phenomenon of weight dependency. Besides, many pruning methods use only one criterion for evaluation and find a sweet spot of pruning structure and accuracy in a trial-and-error fashion, which can be time-consuming. In this paper, we proposed a channel pruning algorithm via multi-criteria based on weight dependency, CPMC, which can compress a pre-trained model directly. CPMC defines channel importance in three aspects, including its associated weight value, computational cost, and parameter quantity. According to the phenomenon of weight dependency, CPMC gets channel importance by assessing its associated filter and the corresponding partial weights in the next layer. Then CPMC uses global normalization to achieve cross-layer comparison. Finally, CPMC removes less important channels by global ranking. CPMC can compress various CNN models, including VGGNet, ResNet, and DenseNet on various image classification datasets. Extensive experiments have shown CPMC outperforms the others significantly.
updated: Tue Apr 20 2021 07:41:11 GMT+0000 (UTC)
published: Fri Nov 06 2020 09:12:00 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アソシエイト