about
Knowledge Distillation from Internal Representations (arxiv.org)
2 points by sel1 on Oct 11, 2019 | hide | past | pdf | discuss on HN

In plain words: Instead of only teaching a small model to copy a big model's final answer guesses, this also makes it copy the big model's inner workings. Across language tasks, that inner-copying version consistently beat copying answers alone.

Abstract

Knowledge distillation is typically conducted by training a small model (the student) to mimic a large and cumbersome model (the teacher). The idea is to compress the knowledge from the teacher by using its output probabilities as soft-labels to optimize the student. However, when the teacher is considerably large, there is no guarantee that the internal knowledge of the teacher will be transferred into the student; even if the student closely matches the soft-labels, its internal representations may be considerably different. This internal mismatch can undermine the generalization capabilities originally intended to be transferred from the teacher to the student. In this paper, we propose to distill the internal representations of a large model such as BERT into a simplified version of it. We formulate two ways to distill such representations and various algorithms to conduct the distillation. We experiment with datasets from the GLUE benchmark and consistently show that adding knowledge distillation from internal representations is a more powerful method than only using soft-label distillation.

Gustavo Aguilar, Yuan Ling, Yu Zhang, Benjamin Yao, Xing Fan, Chenlei Guo
arXiv:1910.03723 · cs.CL · submitted Oct 8, 2019 · updated Jan 16, 2020
abstract · pdf · html · To appear in AAAI-2020

add comment on HN