arXiv reaDer
CATRO:クラス対応トレース比最適化によるチャネルプルーニング
CATRO: Channel Pruning via Class-Aware Trace Ratio Optimization
深い畳み込みニューラルネットワークは、多くのアプリケーションシナリオで高いパラメトリックおよび計算の冗長性を備えて過剰であることが示され、軽量で効率的なネットワークを取得するためにモデルの剪定を検討する作業が増えています。ただし、ほとんどの既存のプルーニングアプローチは、経験的なヒューリスティックによって推進されており、チャネルの共同の影響を考慮することはめったになく、保証されていない最適ではないパフォーマンスにつながります。本論文では、計算負荷を軽減し、モデル推論を加速するために、クラス認識トレース比最適化(CATRO)を介した新しいチャネルプルーニング方法を提案します。 CATROは、いくつかのサンプルからのクラス情報を利用して、特徴空間の識別によって複数のチャネルの共同の影響を測定し、保存されたチャネルのレイヤーごとの影響を統合します。劣モジュラ集合関数の最大化問題としてチャネル剪定を定式化することにより、CATROは2段階の欲張り反復最適化手順を介して効率的にそれを解決します。さらに重要なことは、CATROの収束とパフォーマンスに関する理論的正当化を提示することです。実験結果は、CATROが他の最先端のチャネルプルーニングアルゴリズムよりも同様の計算コストでより高い精度を達成するか、同様の精度でより低い計算コストを達成することを示しています。さらに、CATROはクラス対応の特性を備えているため、さまざまな分類サブタスクに適応して効率的なネットワークをプルーニングするのに適しており、実際のアプリケーションでのディープネットワークの便利な展開と使用を強化します。
Deep convolutional neural networks are shown to be overkill with high parametric and computational redundancy in many application scenarios, and an increasing number of works have explored model pruning to obtain lightweight and efficient networks. However, most existing pruning approaches are driven by empirical heuristics and rarely consider the joint impact of channels, leading to unguaranteed and suboptimal performance. In this paper, we propose a novel channel pruning method via class-aware trace ratio optimization (CATRO) to reduce the computational burden and accelerate the model inference. Utilizing class information from a few samples, CATRO measures the joint impact of multiple channels by feature space discriminations and consolidates the layer-wise impact of preserved channels. By formulating channel pruning as a submodular set function maximization problem, CATRO solves it efficiently via a two-stage greedy iterative optimization procedure. More importantly, we present theoretical justifications on convergence and performance of CATRO. Experimental results demonstrate that CATRO achieves higher accuracy with similar computation cost or lower computation cost with similar accuracy than other state-of-the-art channel pruning algorithms. In addition, because of its class-aware property, CATRO is suitable to prune efficient networks adaptively for various classification subtasks, enhancing handy deployment and usage of deep networks in real-world applications.
updated: Thu Oct 21 2021 06:26:31 GMT+0000 (UTC)
published: Thu Oct 21 2021 06:26:31 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アソシエイト