An Empirical Analysis of the Impact of Data Augmentation on Knowledge Distillation
This addresses a problem for practitioners using knowledge distillation, but it is incremental as it builds on existing augmentation and distillation methods.
The paper investigates how data augmentation strategies, particularly mixed sample methods like MixUp and CutMix, affect knowledge distillation, finding that they impair student model generalization by limiting example-specific feature learning.
Generalization Performance of Deep Learning models trained using Empirical Risk Minimization can be improved significantly by using Data Augmentation strategies such as simple transformations, or using Mixed Samples. We attempt to empirically analyze the impact of such strategies on the transfer of generalization between teacher and student models in a distillation setup. We observe that if a teacher is trained using any of the mixed sample augmentation strategies, such as MixUp or CutMix, the student model distilled from it is impaired in its generalization capabilities. We hypothesize that such strategies limit a model's capability to learn example-specific features, leading to a loss in quality of the supervision signal during distillation. We present a novel Class-Discrimination metric to quantitatively measure this dichotomy in performance and link it to the discriminative capacity induced by the different strategies on a network's latent space.