arXiv reaDer
階層的自己教師あり拡張知識蒸留
Hierarchical Self-supervised Augmented Knowledge Distillation
知識の蒸留には、多くの場合、知識を定義し、教師から生徒に効果的に伝達する方法が含まれます。最近の自己教師あり対照知識は最高のパフォーマンスを達成しますが、ネットワークにそのような知識を学習させると、元のクラス認識タスクの表現学習が損なわれる可能性があります。したがって、元の認識タスクと自己監視補助タスクの同時分布を学習するようにネットワークをガイドするために、代替の自己監視拡張タスクを採用します。通常の分類能力を失うことなく表現力を向上させるためのより豊富な知識として示されています。さらに、以前の方法が確率的知識を最終層の間で転送するだけであるということは不完全です。階層的な中間特徴マップにいくつかの補助分類器を追加して、多様な自己教師あり知識を生成し、1対1の転送を実行して、学生ネットワークを徹底的に教えることを提案します。私たちの方法は、広く使用されているネットワークペア全体で、CIFAR-100で平均2.56%、ImageNetで0.77%の改善で、以前のSOTASSKDを大幅に上回っています。コードはhttps://github.com/winycg/HSAKDで入手できます。
Knowledge distillation often involves how to define and transfer knowledge from teacher to student effectively. Although recent self-supervised contrastive knowledge achieves the best performance, forcing the network to learn such knowledge may damage the representation learning of the original class recognition task. We therefore adopt an alternative self-supervised augmented task to guide the network to learn the joint distribution of the original recognition task and self-supervised auxiliary task. It is demonstrated as a richer knowledge to improve the representation power without losing the normal classification capability. Moreover, it is incomplete that previous methods only transfer the probabilistic knowledge between the final layers. We propose to append several auxiliary classifiers to hierarchical intermediate feature maps to generate diverse self-supervised knowledge and perform the one-to-one transfer to teach the student network thoroughly. Our method significantly surpasses the previous SOTA SSKD with an average improvement of 2.56% on CIFAR-100 and an improvement of 0.77% on ImageNet across widely used network pairs. Codes are available at https://github.com/winycg/HSAKD.
updated: Sat Jul 23 2022 10:19:12 GMT+0000 (UTC)
published: Thu Jul 29 2021 02:57:21 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アソシエイト