arXiv reaDer
スケジュールされたグローアンドプルーン法による効果的なモデルスパース化
Effective Model Sparsification by Scheduled Grow-and-Prune Methods
ディープニューラルネットワーク(DNN)は、多くの現実の問題を解決するのに効果的です。 DNNモデルが大きいほど、通常は品質(精度など)が高くなりますが、計算が多すぎると、トレーニングと推論に時間がかかります。モデルのスパース化により、モデルの品質を維持しながら、計算とメモリのコストを削減できます。ほとんどの既存のスパース化アルゴリズムは一方向に重みを削除しますが、他のアルゴリズムは各レイヤーの重みの小さなサブセットをランダムまたは貪欲に探索します。アルゴリズムの非効率性により、達成可能なスパース性レベルが低下します。さらに、多くのアルゴリズムでは、事前にトレーニングされた高密度モデルが必要であるため、メモリフットプリントが大きく、トレーニング時間が長くなります。この論文では、高密度モデルを事前にトレーニングすることなく、新しいスケジュールされた成長とプルーン(GaP)の方法論を提案します。これは、レイヤーのサブセットを繰り返し密に成長させ、トレーニング後にスパースに戻すことで、以前の作業の欠点に対処します。実験によると、このようなモデルは、画像分類、客観的検出、3Dオブジェクトパーツのセグメンテーション、変換などのさまざまなタスクで、高度に最適化された高密度モデルの品質に匹敵するか、80%のスパース性で打ち負かすことができます。また、事前にトレーニングされた高密度モデルからの剪定など、他の最先端(SOTA)の剪定方法よりも優れています。例として、GaPを介して取得された90%のスパースResNet-50は、ImageNetで77.9%のトップ1精度を達成し、SOTAの結果を1.5%改善します。
Deep neural networks (DNNs) are effective in solving many real-world problems. Larger DNN models usually exhibit better quality (e.g., accuracy) but their excessive computation results in long training and inference time. Model sparsification can reduce the computation and memory cost while maintaining model quality. Most existing sparsification algorithms unidirectionally remove weights, while others randomly or greedily explore a small subset of weights in each layer. The inefficiency of the algorithms reduces the achievable sparsity level. In addition, many algorithms still require pre-trained dense models and thus suffer from large memory footprint and long training time. In this paper, we propose a novel scheduled grow-and-prune (GaP) methodology without pre-training the dense models. It addresses the shortcomings of the previous works by repeatedly growing a subset of layers to dense and then pruning back to sparse after some training. Experiments have shown that such models can match or beat the quality of highly optimized dense models at 80% sparsity on a variety of tasks, such as image classification, objective detection, 3D object part segmentation, and translation. They also outperform other state-of-the-art (SOTA) pruning methods, including pruning from pre-trained dense models. As an example, a 90% sparse ResNet-50 obtained via GaP achieves 77.9% top-1 accuracy on ImageNet, improving the SOTA results by 1.5%.
updated: Fri Jun 18 2021 01:03:13 GMT+0000 (UTC)
published: Fri Jun 18 2021 01:03:13 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アソシエイト