arXiv reaDer
伝達可能なニューラル プルーニング関数の進化
Evolving Transferable Neural Pruning Functions
ニューラル ネットワークの構造設計は、ディープ ラーニングの成功にとって非常に重要です。進化的学習におけるこれまでの研究のほとんどは、ネットワークの構造を直接検索することを目的としていましたが、最近、効率的な深層学習モデルの設計において大きな進歩を遂げた、別の有望なトラックであるチャネルプルーニングに関する試みはほとんど行われていません。実際、従来のプルーニング方法は、人間が作成したプルーニング関数を採用して、チャネルのプルーニングに対するチャネルの重要性をスコアリングしますが、これにはドメインの知識が必要であり、最適ではない可能性があります。この目的のために、遺伝的プログラミング (GP) を使用して強力なプルーニング メトリクスを自動的に発見するパイオニアです。具体的には、高品質で転送可能なプルーニング関数を表現するための新しいデザイン空間を作成します。これにより、進化後の転送可能性のために進化した関数を手動で変更する必要がないエンドツーエンドの進化プロセスが保証されます。以前の方法とは異なり、私たちのアプローチは、効率的な推論のためのコンパクトな剪定ネットワークと、数学的に説明可能であり、したがってさまざまな剪定タスクに一般化できる新しい閉形式剪定メトリックの両方を提供できます。進化は小さなデータセットで行われますが、私たちの関数は、進化プロセスで使用されるものとは異なり、より困難なデータセットに適用すると有望な結果を示します。たとえば、ILSVRC-2012 では、進化した関数によって最先端の剪定結果が得られます。
Structural design of neural networks is crucial for the success of deep learning. While most prior works in evolutionary learning aim at directly searching the structure of a network, few attempts have been made on another promising track, channel pruning, which recently has made major headway in designing efficient deep learning models. In fact, prior pruning methods adopt human-made pruning functions to score a channel's importance for channel pruning, which requires domain knowledge and could be sub-optimal. To this end, we pioneer the use of genetic programming (GP) to discover strong pruning metrics automatically. Specifically, we craft a novel design space to express high-quality and transferable pruning functions, which ensures an end-to-end evolution process where no manual modification is needed on the evolved functions for their transferability after evolution. Unlike prior methods, our approach can provide both compact pruned networks for efficient inference and novel closed-form pruning metrics which are mathematically explainable and thus generalizable to different pruning tasks. While the evolution is conducted on small datasets, our functions shows promising results when applied to more challenging datasets, different from those used in the evolution process. For example, on ILSVRC-2012, an evolved function achieves state-of-the-art pruning results.
updated: Thu Aug 04 2022 02:10:56 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アソシエイト