arXiv reaDer
モデル圧縮のためのピアツーピア相互学習による知識蒸留の強化
Augmenting Knowledge Distillation With Peer-To-Peer Mutual Learning For Model Compression
知識蒸留(KD)は、コンパクトな学生ネットワークが複雑で高度に訓練された教師ネットワークの動作を模倣するように教えられる効果的なモデル圧縮技術です。対照的に、相互学習(ML)は、強力で静的な教師ネットワークがない場合でも、複数の単純な学生ネットワークが知識の共有から利益を得る代替戦略を提供します。これらの調査結果に動機付けられて、KDとMLの両方を活用してパフォーマンスを向上させる、単一教師、複数学生のフレームワークを提案します。さらに、オンライン蒸留戦略を利用して、教師と生徒を同時にトレーニングします。提案されたアプローチのパフォーマンスを評価するために、ベンチマーク生物医学分類(MSI対MSS)およびオブジェクト検出(ポリープ検出)タスクで、教師と学生のネットワークの3つの異なるバージョンを使用して広範な実験が行われました。提案された方法でトレーニングされた学生ネットワークのアンサンブルは、KDまたはMLを個別に使用してトレーニングされた学生のアンサンブルよりも優れた結果を達成し、学生間のピアツーピア学習によって教師から学生への知識移転を強化する利点を確立しました。
Knowledge distillation (KD) is an effective model compression technique where a compact student network is taught to mimic the behavior of a complex and highly trained teacher network. In contrast, Mutual Learning (ML) provides an alternative strategy where multiple simple student networks benefit from sharing knowledge, even in the absence of a powerful but static teacher network. Motivated by these findings, we propose a single-teacher, multi-student framework that leverages both KD and ML to achieve better performance. Furthermore, an online distillation strategy is utilized to train the teacher and students simultaneously. To evaluate the performance of the proposed approach, extensive experiments were conducted using three different versions of teacher-student networks on benchmark biomedical classification (MSI vs. MSS) and object detection (Polyp Detection) tasks. Ensemble of student networks trained in the proposed manner achieved better results than the ensemble of students trained using KD or ML individually, establishing the benefit of augmenting knowledge transfer from teacher to students with peer-to-peer learning between students.
updated: Thu Oct 21 2021 09:59:31 GMT+0000 (UTC)
published: Thu Oct 21 2021 09:59: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アソシエイト