arXiv reaDer
リソース再割り当てによるネットワークプルーニング
Network Pruning via Resource Reallocation
チャネルプルーニングは、大規模で扱いにくいネットワークから重要でないチャネルを排除することにより、小さなコンパクトなモデルを取得するための効果的なアプローチとして広く認識されています。現代の方法は通常、元の過剰にパラメータ化されたモデルから反復的な剪定手順を実行します。これは、特に剪定が積極的である場合、面倒で費用がかかります。この論文では、rEsource rEalLocation(PEEL)を介したネットワークプルーニングと呼ばれるシンプルで効果的なチャネルプルーニング手法を提案し、ごくわずかなコストで目的のスリムモデルを迅速に作成します。具体的には、PEELは最初に事前定義されたバックボーンを構築し、次にリソースの再割り当てを実行して、パラメーターを情報量の少ないレイヤーから重要なレイヤーに1ラウンドでシフトし、これらの情報量の多いレイヤーのプラスの効果を増幅します。 PEELの有効性を実証するために、ResNet-18、ResNet-50、MobileNetV2、MobileNetV3-small、およびEfficientNet-B0を使用してImageNetで広範な実験を実行します。実験結果は、PEELによってカバーされていない構造が、さまざまな剪定設定の下で最先端の剪定アルゴリズムと競争力のあるパフォーマンスを示すことを示しています。私たちのコードはhttps://github.com/cardwing/Codes-for-PEELで入手できます。
Channel pruning is broadly recognized as an effective approach to obtain a small compact model through eliminating unimportant channels from a large cumbersome network. Contemporary methods typically perform iterative pruning procedure from the original over-parameterized model, which is both tedious and expensive especially when the pruning is aggressive. In this paper, we propose a simple yet effective channel pruning technique, termed network Pruning via rEsource rEalLocation (PEEL), to quickly produce a desired slim model with negligible cost. Specifically, PEEL first constructs a predefined backbone and then conducts resource reallocation on it to shift parameters from less informative layers to more important layers in one round, thus amplifying the positive effect of these informative layers. To demonstrate the effectiveness of PEEL , we perform extensive experiments on ImageNet with ResNet-18, ResNet-50, MobileNetV2, MobileNetV3-small and EfficientNet-B0. Experimental results show that structures uncovered by PEEL exhibit competitive performance with state-of-the-art pruning algorithms under various pruning settings. Our code is available at https://github.com/cardwing/Codes-for-PEEL.
updated: Tue Mar 02 2021 16:28:10 GMT+0000 (UTC)
published: Tue Mar 02 2021 16:28:10 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アソシエイト