arXiv reaDer
Slimmable 枝刈りされたニューラル ネットワーク
Slimmable Pruned Neural Networks
Slimmable Neural Networks (S-Net) は、現在の計算リソースの可用性に応じて、定義済みの割合のチャネル (サブネットワーク) の 1 つを動的に選択できる新しいネットワークです。ただし、S-Net 上の各サブネットワークの精度は、異なるサブネットワークでの同時最適化が難しいため、同じサイズの個別にトレーニングされたネットワークよりも劣ります。本稿では、S-Net のように各層のチャネルの割合が同じ構造 (幅乗数) を採用する代わりに、枝刈りによってサブネットワーク構造を学習する Slimmable Pruned Neural Networks (SP-Net) を提案し、さらに新しい剪定手順を提案します。ワンショットまたは反復剪定の代わりに複数ベースの剪定を行い、高精度と大幅なトレーニング時間の節約を実現します。また、S-Net と同じくらい高速な計算を実現するためのスリム化可能なチャネル ソーティング (scs) と、より効率的な方法で残余構造を刈り込むためのゼロ パディング マッチ (zpm) 刈り込みを導入しました。 SP-Net は、あらゆる種類のチャネル プルーニング方法と組み合わせることができ、NAS モデルのような複雑な処理や時間のかかるアーキテクチャ検索を必要としません。 S-Net 上の同じ FLOP の各サブネットワークと比較すると、SP-Net は ResNet-50 で 1.2-1.5%、VGGNet で 0.9-4.4%、MobileNetV1 で 1.3-2.7%、MobileNetV2 で 1.4-3.1% 精度が向上します。イメージネット上。さらに、ImageNet での実験結果によると、当社の方法は他の SOTA プルーニング方法よりも優れており、さまざまな NAS モデルと同等です。コードは https://github.com/hideakikuratsu/SP-Net で入手できます。
Slimmable Neural Networks (S-Net) is a novel network which enabled to select one of the predefined proportions of channels (sub-network) dynamically depending on the current computational resource availability. The accuracy of each sub-network on S-Net, however, is inferior to that of individually trained networks of the same size due to its difficulty of simultaneous optimization on different sub-networks. In this paper, we propose Slimmable Pruned Neural Networks (SP-Net), which has sub-network structures learned by pruning instead of adopting structures with the same proportion of channels in each layer (width multiplier) like S-Net, and we also propose new pruning procedures: multi-base pruning instead of one-shot or iterative pruning to realize high accuracy and huge training time saving. We also introduced slimmable channel sorting (scs) to achieve calculation as fast as S-Net and zero padding match (zpm) pruning to prune residual structure in more efficient way. SP-Net can be combined with any kind of channel pruning methods and does not require any complicated processing or time-consuming architecture search like NAS models. Compared with each sub-network of the same FLOPs on S-Net, SP-Net improves accuracy by 1.2-1.5% for ResNet-50, 0.9-4.4% for VGGNet, 1.3-2.7% for MobileNetV1, 1.4-3.1% for MobileNetV2 on ImageNet. Furthermore, our methods outperform other SOTA pruning methods and are on par with various NAS models according to our experimental results on ImageNet. The code is available at https://github.com/hideakikuratsu/SP-Net.
updated: Wed Dec 07 2022 02:54:15 GMT+0000 (UTC)
published: Wed Dec 07 2022 02:54:15 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アソシエイト