arXiv reaDer
モデル圧縮のための段階的なチャネルプルーニング
Stage-wise Channel Pruning for Model Compression
自動MLプルーニング手法は、プルーニング戦略を自動的に検索して、深い畳み込みニューラルネットワーク(深いCNN)の計算の複雑さを軽減することを目的としています。ただし、以前のいくつかの作業では、多くのAuto-MLプルーニング方法の結果でさえ、均一プルーニング方法の結果を超えることができないことがわかりました。このホワイトペーパーでは、最初にAuto-MLプルーニングが無効である理由を分析します。続いて、上記の問題を解決するために段階的剪定(SP)法が提案されます。以前のほとんどのAuto-MLプルーニング方法と同様に、SPは、サブネットにプロキシパフォーマンスを提供し、最高のプロキシパフォーマンスを持つ最高のサブネットを検索できるスーパーネットもトレーニングします。以前の作業とは異なり、ディープCNNをいくつかの段階に分割し、すべてのレイヤーがプルーニングされていないフルネットを使用して、サブネットのトレーニングと検索を監視します。注目すべきことに、SPでトレーニングされたサブネットのプロキシパフォーマンスは、以前のAuto-MLプルーニング作業のほとんどよりも実際のパフォーマンスに近くなっています。したがって、SPは、モバイル設定でCIFAR-10とImageNetの両方で最先端を実現します。
Auto-ML pruning methods aim at searching a pruning strategy automatically to reduce the computational complexity of deep Convolutional Neural Networks(deep CNNs). However, some previous works found that the results of many Auto-ML pruning methods even cannot surpass the results of the uniformly pruning method. In this paper, we first analyze the reason for the ineffectiveness of Auto-ML pruning. Subsequently, a stage-wise pruning(SP) method is proposed to solve the above problem. As with most of the previous Auto-ML pruning methods, SP also trains a super-net that can provide proxy performance for sub-nets and search the best sub-net who has the best proxy performance. Different from previous works, we split a deep CNN into several stages and use a full-net where all layers are not pruned to supervise the training and the searching of sub-nets. Remarkably, the proxy performance of sub-nets trained with SP is closer to the actual performance than most of the previous Auto-ML pruning works. Therefore, SP achieves the state-of-the-art on both CIFAR-10 and ImageNet under the mobile setting.
updated: Tue Nov 10 2020 05:19:05 GMT+0000 (UTC)
published: Tue Nov 10 2020 05:19: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アソシエイト