arXiv reaDer
クラス識別CNN圧縮
Class-Discriminative CNN Compression
剪定と蒸留による畳み込みニューラルネットワーク(CNN)の圧縮は、コミュニティでますます注目を集めています。特に、階級差別に基づくアプローチの設計は、CNNのトレーニング目標にシームレスに適合するため望ましいでしょう。この論文では、CNNのトレーニング目標を容易にするために、剪定と蒸留の両方で階級差別を注入する階級差別圧縮(CDC)を提案します。最初に、チャネルプルーニングに対する判別関数のグループの有効性を調査します。ここでは、直観的な一般化により、スチューデントのT検定などのよく知られた単一変量のバイナリクラス統計を調査に含めます。次に、新しいレイヤー適応型階層プルーニングアプローチを提案します。このアプローチでは、初期レイヤーには粗いクラス識別スキームを使用し、後のレイヤーには細かいクラス識別スキームを使用します。この方法は、CNNが初期の層で粗いセマンティクスを処理し、後で細かい概念を抽出するという事実と自然に一致します。さらに、判別成分分析(DCA)を活用して、豊富な判別情報を備えた部分空間の中間表現の知識を抽出します。これにより、隠れ層の線形分離可能性と学生の分類精度が向上します。剪定と蒸留を組み合わせて、CDCはCIFARとILSVRC 2012で評価され、常に最先端の結果を上回っています。
Compressing convolutional neural networks (CNNs) by pruning and distillation has received ever-increasing focus in the community. In particular, designing a class-discrimination based approach would be desired as it fits seamlessly with the CNNs training objective. In this paper, we propose class-discriminative compression (CDC), which injects class discrimination in both pruning and distillation to facilitate the CNNs training goal. We first study the effectiveness of a group of discriminant functions for channel pruning, where we include well-known single-variate binary-class statistics like Student's T-Test in our study via an intuitive generalization. We then propose a novel layer-adaptive hierarchical pruning approach, where we use a coarse class discrimination scheme for early layers and a fine one for later layers. This method naturally accords with the fact that CNNs process coarse semantics in the early layers and extract fine concepts at the later. Moreover, we leverage discriminant component analysis (DCA) to distill knowledge of intermediate representations in a subspace with rich discriminative information, which enhances hidden layers' linear separability and classification accuracy of the student. Combining pruning and distillation, CDC is evaluated on CIFAR and ILSVRC 2012, where we consistently outperform the state-of-the-art results.
updated: Thu Oct 21 2021 02:54:05 GMT+0000 (UTC)
published: Thu Oct 21 2021 02:54:05 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アソシエイト