arXiv reaDer
学生は最高の教師です:出口-複数出口を備えたアンサンブル蒸留
Students are the Best Teacher: Exit-Ensemble Distillation with Multi-Exits
この論文は、出口アンサンブル蒸留と呼ばれる、事前に訓練された教師ネットワークなしで畳み込みニューラルネットワーク(CNN)の分類性能を改善するための新しい知識蒸留ベースの学習方法を提案します。私たちの方法は、従来のCNNの途中に補助分類子(出口と呼ばれる)を追加する複数出口アーキテクチャを利用します。これにより、早期の推論結果を得ることができます。私たちの方法のアイデアは、出口のアンサンブルを蒸留ターゲットとして使用してネットワークをトレーニングすることです。これにより、ネットワーク全体の分類パフォーマンスが大幅に向上します。私たちの方法は、知識蒸留の新しいパラダイムを示唆しています。教師が生徒にのみ教える従来の蒸留の概念とは異なり、生徒は他の生徒や教師でさえもよりよく学ぶのを助けることができることを示しています。実験結果は、私たちの方法がさまざまな一般的なCNNアーキテクチャ(VGG、ResNet、ResNeXt、WideResNetなど)で分類パフォーマンスの大幅な改善を達成することを示しています。さらに、提案された方法は、改善された安定性で学習の収束を促進することができます。私たちのコードはGithubで入手できます。
This paper proposes a novel knowledge distillation-based learning method to improve the classification performance of convolutional neural networks (CNNs) without a pre-trained teacher network, called exit-ensemble distillation. Our method exploits the multi-exit architecture that adds auxiliary classifiers (called exits) in the middle of a conventional CNN, through which early inference results can be obtained. The idea of our method is to train the network using the ensemble of the exits as the distillation target, which greatly improves the classification performance of the overall network. Our method suggests a new paradigm of knowledge distillation; unlike the conventional notion of distillation where teachers only teach students, we show that students can also help other students and even the teacher to learn better. Experimental results demonstrate that our method achieves significant improvement of classification performance on various popular CNN architectures (VGG, ResNet, ResNeXt, WideResNet, etc.). Furthermore, the proposed method can expedite the convergence of learning with improved stability. Our code will be available on Github.
updated: Mon Apr 05 2021 01:13:20 GMT+0000 (UTC)
published: Thu Apr 01 2021 07:10:36 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アソシエイト