arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2106.05945cs.LGstat.ML
收藏

知识蒸馏真的有效吗?

Does Knowledge Distillation Really Work?

NYU(纽约大学);Google Research(谷歌研究院)

Samuel Stanton, Pavel Izmailov, Polina Kirichenko, Alexander A. Alemi, Andrew Gordon Wilson

摘要:

知识蒸馏是一种流行的技术,用于训练小型学生网络以模拟较大的教师模型,例如网络的集合。我们发现,虽然知识蒸馏可以提高学生的泛化能力,但它通常并不像人们通常理解的那样起作用:教师和学生的预测分布之间往往存在令人惊讶的巨大差异,即使在学生有能力完全匹配教师的情况下。我们确定优化的困难是学生无法与老师相匹配的关键原因。我们还展示了用于蒸馏的数据集的细节如何在学生与教师的匹配程度上发挥作用-并且更接近地匹配教师并不总是导致更好的学生概括。

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑