arXiv reaDer
進化する転送可能な剪定機能
Evolving Transferable Pruning Functions
チャネルプルーニングは、効率的な深層学習モデルの設計において大きな進歩を遂げました。従来のアプローチでは、人為的なプルーニング機能を採用して、チャネルプルーニングに対するチャネルの重要性をスコアリングします。これには、ドメインの知識が必要であり、最適ではない可能性があります。この作業では、強力なプルーニングメトリックを自動的に検出するためのエンドツーエンドのフレームワークを提案します。具体的には、剪定機能を表現するための新しいデザインスペースを作成し、進化戦略である遺伝的プログラミングを活用して、高品質で転送可能な剪定機能を進化させます。以前の方法とは異なり、私たちのアプローチは、効率的な推論のためのコンパクトな剪定ネットワークを提供するだけでなく、数学的に説明可能であり、したがってさまざまな剪定タスクに一般化できる新しい閉じた形式の剪定メトリックも提供できます。進化は小さなデータセットで実行されますが、学習した関数は手動で変更することなく大きなデータセットに転送できます。大規模なデータセットでの直接的な進化と比較して、私たちの戦略はより優れた費用対効果を示しています。 ILSVRC-2012など、進化プロセスで使用されるデータセットとは異なる、より困難なデータセットに適用すると、進化した関数は最先端のプルーニング結果を実現します。
Channel pruning has made major headway in the design of efficient deep learning models. Conventional approaches adopt human-made pruning functions to score channels' importance for channel pruning, which requires domain knowledge and could be sub-optimal. In this work, we propose an end-to-end framework to automatically discover strong pruning metrics. Specifically, we craft a novel design space for expressing pruning functions and leverage an evolution strategy, genetic programming, to evolve high-quality and transferable pruning functions. Unlike prior methods, our approach can not only provide compact pruned networks for efficient inference, but also novel closed-form pruning metrics that are mathematically explainable and thus generalizable to different pruning tasks. The evolution is conducted on small datasets while the learned functions are transferable to larger datasets without any manual modification. Compared to direct evolution on a large dataset, our strategy shows better cost-effectiveness. When applied to more challenging datasets, different from those used in the evolution process, e.g., ILSVRC-2012, an evolved function achieves state-of-the-art pruning results.
updated: Thu Oct 21 2021 03:40:51 GMT+0000 (UTC)
published: Thu Oct 21 2021 03:40:51 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アソシエイト