Enregistré dans:
Détails bibliographiques
Auteurs principaux: Fan, Wen-Shu, Li, Xin-Chun, Zhan, De-Chuan
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:https://arxiv.org/abs/2405.13078
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866916781440892928
author Fan, Wen-Shu
Li, Xin-Chun
Zhan, De-Chuan
author_facet Fan, Wen-Shu
Li, Xin-Chun
Zhan, De-Chuan
contents Knowledge Distillation (KD) could transfer the ``dark knowledge" of a well-performed yet large neural network to a weaker but lightweight one. From the view of output logits and softened probabilities, this paper goes deeper into the dark knowledge provided by teachers with different capacities. Two fundamental observations are: (1) a larger teacher tends to produce probability vectors with lower distinction among non-ground-truth classes; (2) teachers with different capacities are basically consistent in their cognition of relative class affinity. Through abundant experimental studies we verify these observations and provide in-depth empirical explanations to them. We argue that the distinctness among incorrect classes embodies the essence of dark knowledge. A larger and more accurate teacher lacks this distinctness, which hampers its teaching ability compared to a smaller teacher, ultimately leading to the peculiar phenomenon named "capacity mismatch". Building on this insight, this paper explores multiple simple yet effective ways to address capacity mismatch, achieving superior experimental results compared to previous approaches.
format Preprint
id arxiv_https___arxiv_org_abs_2405_13078
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Exploring Dark Knowledge under Various Teacher Capacities and Addressing Capacity Mismatch
Fan, Wen-Shu
Li, Xin-Chun
Zhan, De-Chuan
Machine Learning
Knowledge Distillation (KD) could transfer the ``dark knowledge" of a well-performed yet large neural network to a weaker but lightweight one. From the view of output logits and softened probabilities, this paper goes deeper into the dark knowledge provided by teachers with different capacities. Two fundamental observations are: (1) a larger teacher tends to produce probability vectors with lower distinction among non-ground-truth classes; (2) teachers with different capacities are basically consistent in their cognition of relative class affinity. Through abundant experimental studies we verify these observations and provide in-depth empirical explanations to them. We argue that the distinctness among incorrect classes embodies the essence of dark knowledge. A larger and more accurate teacher lacks this distinctness, which hampers its teaching ability compared to a smaller teacher, ultimately leading to the peculiar phenomenon named "capacity mismatch". Building on this insight, this paper explores multiple simple yet effective ways to address capacity mismatch, achieving superior experimental results compared to previous approaches.
title Exploring Dark Knowledge under Various Teacher Capacities and Addressing Capacity Mismatch
topic Machine Learning
url https://arxiv.org/abs/2405.13078