Does Knowledge Distillation Really Work?Download PDF

21 May 2021, 20:48 (edited 28 Jan 2022)NeurIPS 2021 PosterReaders: Everyone
  • Keywords: distillation, deep ensembles, optimization
  • TL;DR: We show that it is difficult to achieve high agreement between student and teacher in knowledge distillation and explain why
  • Abstract: Knowledge distillation is a popular technique for training a small student network to emulate a larger teacher model, such as an ensemble of networks. We show that while knowledge distillation can improve student generalization, it does not typically work as it is commonly understood: there often remains a surprisingly large discrepancy between the predictive distributions of the teacher and the student, even in cases when the student has the capacity to perfectly match the teacher. We identify difficulties in optimization as a key reason for why the student is unable to match the teacher. We also show how the details of the dataset used for distillation play a role in how closely the student matches the teacher --- and that more closely matching the teacher paradoxically does not always lead to better student generalization.
  • Supplementary Material: pdf
  • Code Of Conduct: I certify that all co-authors of this work have read and commit to adhering to the NeurIPS Statement on Ethics, Fairness, Inclusivity, and Code of Conduct.
  • Code: https://github.com/samuelstanton/gnosis
19 Replies

Loading