arXiv reaDer
パラメータ効率が高く、学生に優しい知識の蒸留
Parameter-Efficient and Student-Friendly Knowledge Distillation
知識蒸留(KD)は、大規模な教師モデルから小規模な学生に知識を転送するために広く採用されており、トレーニング中に教師のパラメータが(または部分的に)固定されます。最近の研究によると、このモードでは、モデルの容量が一致しないため、知識の伝達が困難になる可能性があります。ミスマッチの問題を軽減するために、オンライン蒸留などの教師と生徒の共同トレーニング方法が提案されていますが、それは常に高価な計算コストを必要とします。この論文では、比較的少数の部分パラメータを更新することによって効率的かつ十分な知識伝達を達成するための、パラメータ効率が高く学生に優しい知識蒸留法、すなわちPESF-KDを提示します。技術的には、最初に、予測分布間のシャープネスギャップとして不一致を数学的に定式化します。ここで、このようなギャップは、ソフトラベルの適切な滑らかさで狭めることができることを示します。次に、教師用のアダプタモジュールを紹介し、適切な滑らかさのソフトラベルを取得するためにアダプタを更新するだけです。さまざまなベンチマークでの実験により、PESF-KDは、高度なオンライン蒸留法と比較して、競争力のある結果を得ながら、トレーニングコストを大幅に削減できることが示されています。コードは承認時にリリースされます。
Knowledge distillation (KD) has been extensively employed to transfer the knowledge from a large teacher model to the smaller students, where the parameters of the teacher are fixed (or partially) during training. Recent studies show that this mode may cause difficulties in knowledge transfer due to the mismatched model capacities. To alleviate the mismatch problem, teacher-student joint training methods, e.g., online distillation, have been proposed, but it always requires expensive computational cost. In this paper, we present a parameter-efficient and student-friendly knowledge distillation method, namely PESF-KD, to achieve efficient and sufficient knowledge transfer by updating relatively few partial parameters. Technically, we first mathematically formulate the mismatch as the sharpness gap between their predictive distributions, where we show such a gap can be narrowed with the appropriate smoothness of the soft label. Then, we introduce an adapter module for the teacher and only update the adapter to obtain soft labels with appropriate smoothness. Experiments on a variety of benchmarks show that PESF-KD can significantly reduce the training cost while obtaining competitive results compared to advanced online distillation methods. Code will be released upon acceptance.
updated: Sat May 28 2022 16:11:49 GMT+0000 (UTC)
published: Sat May 28 2022 16:11:49 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アソシエイト