arXiv reaDer
教師クラスネットワーク:ニューラルネットワーク圧縮メカニズム
Teacher-Class Network: A Neural Network Compression Mechanism
ディープニューラルネットワーク(DNN)の圧倒的なサイズを縮小するために、教師と生徒の方法論は、複雑な教師ネットワークから単純な生徒ネットワークに知識を転送しようとします。代わりに、単一の教師と複数の学生ネットワーク(つまり、学生のクラス)で構成される教師クラスネットワークと呼ばれる新しい方法を提案します。提案された方法は、知識を1人の学生だけに転送するのではなく、知識のチャンクを各学生に転送します。私たちの生徒は、問題固有のロジットのトレーニングを受けていません。教師ネットワークで学習した知識(密な表現)を模倣するようにトレーニングされているため、生徒のクラスで学習した知識を組み合わせて、他の問題を解決することもできます。提案された教師クラスのアーキテクチャは、MNIST、Fashion MNIST、IMDB Movie Reviews、CAMVid、CIFAR-10、ImageNetなどのいくつかのベンチマークデータセットで、画像分類、感情分類、セグメンテーションなどの複数のタスクで評価されます。私たちのアプローチは、パラメータの10〜30分の1の削減を達成しながら、精度と計算コストの点で最先端の独身学生のアプローチを上回っています。
To reduce the overwhelming size of Deep Neural Networks (DNN) teacher-student methodology tries to transfer knowledge from a complex teacher network to a simple student network. We instead propose a novel method called the teacher-class network consisting of a single teacher and multiple student networks (i.e. class of students). Instead of transferring knowledge to one student only, the proposed method transfers a chunk of knowledge to each student. Our students are not trained for problem-specific logits, they are trained to mimic knowledge (dense representation) learned by the teacher network thus the combined knowledge learned by the class of students can be used to solve other problems as well. The proposed teacher-class architecture is evaluated on several benchmark datasets such as MNIST, Fashion MNIST, IMDB Movie Reviews, CAMVid, CIFAR-10 and ImageNet on multiple tasks including image classification, sentiment classification and segmentation. Our approach outperforms the state of-the-art single student approach in terms of accuracy as well as computational cost while achieving 10-30 times reduction in parameters.
updated: Fri Oct 29 2021 18:58:25 GMT+0000 (UTC)
published: Tue Apr 07 2020 11:31:20 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アソシエイト