arXiv reaDer
グループ畳み込みのための効率的な構造化プルーニングおよびアーキテクチャ検索
Efficient Structured Pruning and Architecture Searching for Group Convolution
  畳み込みニューラルネットワークの効率的な推論は、最近盛んなトピックです。事前にトレーニングされたモデルを展開するとき、与えられた推論予算の制約の下で最大のテスト精度を達成することが望ましい。ネットワークプルーニングは一般的に使用される手法ですが、実際のスピードアップをほとんど得られない不規則なスパースモデルを生成する場合があります。グループ畳み込みは、その規則的な構造のために有望な枝刈りターゲットです。ただし、そのような構造をプルーニング手順に組み込むことは困難です。これは、構造上の制約を説明するのが難しく、プルーニングを解決しにくいためです。テストの精度を最大化するグループコンボリューションアーキテクチャ、つまりグループの数を設定する必要もあります。このペーパーでは、この課題に対処する効率的な方法を示します。グループ畳み込みのプルーニングは、最適なチャネル順列を見つけて構造的制約を課し、ヒューリスティックによって効率的に解決するように定式化します。また、テストの精度を最大化するために、推定プルーニングコストに基づいてグループ構成の探索にローカル検索を適用します。以前の作業と比較して、結果は、我々の方法がより短い枝刈り期間内のさまざまなタスクの競合グループたたみ込みモデルを生成し、推論予算の制約を受けて迅速なグループ構成探索を可能にすることを示しています。
Efficient inference of Convolutional Neural Networks is a thriving topic recently. It is desirable to achieve the maximal test accuracy under given inference budget constraints when deploying a pre-trained model. Network pruning is a commonly used technique while it may produce irregular sparse models that can hardly gain actual speed-up. Group convolution is a promising pruning target due to its regular structure; however, incorporating such structure into the pruning procedure is challenging. It is because structural constraints are hard to describe and can make pruning intractable to solve. The need for configuring group convolution architecture, i.e., the number of groups, that maximises test accuracy also increases difficulty. This paper presents an efficient method to address this challenge. We formulate group convolution pruning as finding the optimal channel permutation to impose structural constraints and solve it efficiently by heuristics. We also apply local search to exploring group configuration based on estimated pruning cost to maximise test accuracy. Compared to prior work, results show that our method produces competitive group convolution models for various tasks within a shorter pruning period and enables rapid group configuration exploration subject to inference budget constraints.
updated: Mon Oct 28 2019 05:14:47 GMT+0000 (UTC)
published: Fri Nov 23 2018 01:45:44 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アソシエイト