arXiv reaDer
Latency-Saliency Knapsack による構造的剪定
Structural Pruning via Latency-Saliency Knapsack
構造的プルーニングは、ネットワーク アーキテクチャを簡素化し、推論速度を向上させることができます。我々は、構造的剪定をグローバルなリソース割り当て最適化問題として定式化する Hardware-Aware Latency Pruning (HALP) を提案し、対象デバイスの事前定義された予算の下で遅延を抑制しながら精度を最大化することを目指しています。フィルターの重要度ランキングでは、HALP はレイテンシ ルックアップ テーブルを活用してレイテンシ削減の可能性とグローバルな顕著性スコアを追跡し、精度の低下を測定します。両方のメトリックは、枝刈り中に非常に効率的に評価できるため、ターゲット制約が与えられた場合の報酬最大化問題の下で、グローバルな構造枝刈りを再定式化できます。これにより、拡張ナップザック ソルバーを介して問題を解決できるようになり、HALP は枝刈りの有効性と精度と効率のトレードオフにおいて以前の作業を上回ることができます。さまざまなネットワーク、ImageNet および VOC データセット、さまざまなプラットフォームで、分類タスクと検出タスクの両方で HALP を調べます。特に、ImageNet での ResNet-50/-101 プルーニングの場合、HALP はネットワーク スループットを 1.60 倍/1.90 倍改善し、上位 1 位の精度がそれぞれ +0.3%/-0.2% 変化します。 VOC での SSD プルーニングの場合、HALP はわずか 0.56 mAP の低下でスループットを 1.94 倍向上させます。 HALP は一貫して先行技術よりも優れており、場合によっては大きな差があります。 https://halp-neurips.github.io/ のプロジェクト ページ。
Structural pruning can simplify network architecture and improve inference speed. We propose Hardware-Aware Latency Pruning (HALP) that formulates structural pruning as a global resource allocation optimization problem, aiming at maximizing the accuracy while constraining latency under a predefined budget on targeting device. For filter importance ranking, HALP leverages latency lookup table to track latency reduction potential and global saliency score to gauge accuracy drop. Both metrics can be evaluated very efficiently during pruning, allowing us to reformulate global structural pruning under a reward maximization problem given target constraint. This makes the problem solvable via our augmented knapsack solver, enabling HALP to surpass prior work in pruning efficacy and accuracy-efficiency trade-off. We examine HALP on both classification and detection tasks, over varying networks, on ImageNet and VOC datasets, on different platforms. In particular, for ResNet-50/-101 pruning on ImageNet, HALP improves network throughput by 1.60×/1.90× with +0.3%/-0.2% top-1 accuracy changes, respectively. For SSD pruning on VOC, HALP improves throughput by 1.94× with only a 0.56 mAP drop. HALP consistently outperforms prior art, sometimes by large margins. Project page at https://halp-neurips.github.io/.
updated: Thu Oct 13 2022 01:41:59 GMT+0000 (UTC)
published: Thu Oct 13 2022 01:41:59 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アソシエイト