arXiv reaDer
アダプティブエグザンプラフィルターを使用したネットワークプルーニング
Network Pruning using Adaptive Exemplar Filters
一般的なネットワークプルーニングアルゴリズムは、手作りのモデルを最適化することで冗長な情報を減らし、最適ではないパフォーマンスとフィルターの選択に長い時間をもたらす可能性があります。アルゴリズム設計を簡素化するために適応型エグザンプラフィルターを革新的に導入し、EPrunerと呼ばれる自動で効率的なプルーニングアプローチを実現します。顔認識コミュニティに触発されて、重み行列でメッセージパッシングアルゴリズムAffinity Propagationを使用して、適応数のエグザンプラを取得します。これは、保存されたフィルタとして機能します。 EPrunerは、「重要な」フィルターを決定する際にトレーニングデータへの依存を解消し、GPUベースのSOTAよりも桁違いに高速なCPU実装を数秒で可能にします。さらに、エグザンプラの重みが微調整のためのより良い初期化を提供することを示します。 VGGNet-16では、EPrunerは88.80%のパラメーターを削除することで、76.34%-FLOPsの削減を実現し、CIFAR-10では0.06%の精度が向上しています。 ResNet-152では、EPrunerは64.18%のパラメーターを削除することで65.12%-FLOPsの削減を達成し、ILSVRC-2012では上位5つの精度がわずか0.71%低下します。私たちのコードはhttps://github.com/lmbxmu/EPrunerで入手できます。
Popular network pruning algorithms reduce redundant information by optimizing hand-crafted models, and may cause suboptimal performance and long time in selecting filters. We innovatively introduce adaptive exemplar filters to simplify the algorithm design, resulting in an automatic and efficient pruning approach called EPruner. Inspired by the face recognition community, we use a message passing algorithm Affinity Propagation on the weight matrices to obtain an adaptive number of exemplars, which then act as the preserved filters. EPruner breaks the dependency on the training data in determining the "important" filters and allows the CPU implementation in seconds, an order of magnitude faster than GPU based SOTAs. Moreover, we show that the weights of exemplars provide a better initialization for the fine-tuning. On VGGNet-16, EPruner achieves a 76.34%-FLOPs reduction by removing 88.80% parameters, with 0.06% accuracy improvement on CIFAR-10. In ResNet-152, EPruner achieves a 65.12%-FLOPs reduction by removing 64.18% parameters, with only 0.71% top-5 accuracy loss on ILSVRC-2012. Our code can be available at https://github.com/lmbxmu/EPruner.
updated: Sun Apr 04 2021 07:04:01 GMT+0000 (UTC)
published: Wed Jan 20 2021 06:18:38 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アソシエイト