arXiv reaDer
ヘシアンを意識した剪定と最適な神経インプラント
Hessian-Aware Pruning and Optimal Neural Implant
プルーニングは、ニューラルネットワークモデルに関連するメモリフットプリントとFLOPを削減するための効果的な方法です。ただし、既存の構造化プルーニング方法では、中程度のプルーニングレベルで精度が大幅に低下することがよくあります。この問題に対処するために、構造化剪定のメトリックとして2次感度を使用するニューラルインプラントアプローチと組み合わせた新しいヘシアンアウェア剪定(HAP)メソッドを導入します。基本的な考え方は、感度の低いコンポーネントを剪定し、中程度の感度のコンポーネントには完全に剪定するのではなく、ニューラルインプラントを使用することです。後者のアプローチでは、中程度の感度のコンポーネントが、元のコンポーネントよりも小さく、計算コストが低い低ランクのインプラントに置き換えられます。文献で一般的に使用されているマグニチュードベースの感度メトリックとは対照的に、相対ヘッセトレースを使用して感度を測定します。コンピュータービジョンタスクと自然言語タスクの両方についてHAPをテストし、新しい最先端の結果を達成します。具体的には、HAPはPreResNet29 / ResNet50(CIFAR-10 / ImageNet)で0.1%/ 0.5%未満の劣化を達成し、パラメーターの70%/ 50%以上が削除されます。一方、HAPは、トランスベースのモデルでのヘッドプルーニングの勾配ベースの方法と比較して、大幅に優れたパフォーマンス(パラメーターの60%がプルーニングされた状態で最大0.8%)も実現します。フレームワークはオープンソースであり、オンラインで利用できます。
Pruning is an effective method to reduce the memory footprint and FLOPs associated with neural network models. However, existing structured-pruning methods often result in significant accuracy degradation for moderate pruning levels. To address this problem, we introduce a new Hessian Aware Pruning (HAP) method coupled with a Neural Implant approach that uses second-order sensitivity as a metric for structured pruning. The basic idea is to prune insensitive components and to use a Neural Implant for moderately sensitive components, instead of completely pruning them. For the latter approach, the moderately sensitive components are replaced with with a low rank implant that is smaller and less computationally expensive than the original component. We use the relative Hessian trace to measure sensitivity, as opposed to the magnitude based sensitivity metric commonly used in the literature. We test HAP for both computer vision tasks and natural language tasks, and we achieve new state-of-the-art results. Specifically, HAP achieves less than 0.1%/0.5% degradation on PreResNet29/ResNet50 (CIFAR-10/ImageNet) with more than 70%/50% of parameters pruned. Meanwhile, HAP also achieves significantly better performance (up to 0.8% with 60% of parameters pruned) as compared to gradient based method for head pruning on transformer-based models. The framework has been open sourced and available online.
updated: Mon Jun 21 2021 21:41:39 GMT+0000 (UTC)
published: Fri Jan 22 2021 04:08:03 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アソシエイト