Knowledge distillation is the method to enhance student network by teacher knowledge. So annually knowledge distillation methods have been proposed, but each paper's do experiments with different networks and compare with different methods. Moreover, each method is implemented by each author, so if a new researcher wants to study knowledge distillation, they have to find or implement all of the methods. Surely it is tough work. To reduce this burden, I publish some codes and modify from my research codes. I'll update the code and knowledge distillation algorithm, and all of the things will be implemented using Tensorflow.
Upgrade version of this Repo. will be available at this link
Please check detail of each category in MHGD and If you think the above categorization is useful, please consider citing the following paper.
@inproceedings{GraphKD,
title = {Graph-based Knowledge Distillation by Multi-head Attention Network},
author = {Seunghyun Lee, Byung Cheol Song},
booktitle = {British Machine Vision Conference (BMVC)},
year = {2019}
}
Defined knowledge by the neural response of the hidden layer or the output layer of the network
- Soft-logit : The first knowledge distillation method for deep neural network. Knowledge is defined by softened logits. Because it is easy to handle it, many applied methods were proposed using it such as semi-supervised learning, defencing adversarial attack and so on.
- Deep Mutual Learning (DML) : train teacher and student network coincidently, to follow not only training results but teacher network's training procedure.
- Factor Transfer (FT) : Encode a teacher network's feature map, and transfer the knowledge by mimicking it.
Increase the quantity of knowledge by sensing several points of the teacher network
- FitNet : To increase amounts of information, knowledge is defined by multi-connected networks and compared feature maps by L2-distance.
- Attention transfer (AT) : Knowledge is defined by attention map which is L2-norm of each feature point.
- Activation boundary (AB) : To soften teacher network's constraint, they propose the new metric function inspired by hinge loss which usually used for SVM.
- VID : Define variational lower boundary as the knowledge, to maximize mutual information between teacher and student network.
Defined knowledge by the shared representation between two feature maps
- Flow of Procedure (FSP) : To soften teacher network's constraint, they define knowledge as relation of two feature maps.
- KD using Singular value decomposition(KD-SVD) : To extract major information in feature map, they use singular value decomposition.
Defined knowledge by intra-data relation
- Relational Knowledge Distillation (RKD): they propose knowledge which contains not only feature information but also intra-data relation information.
- Multi-head Graph Distillation (MHGD): They proposed the distillation module which built with the multi-head attention network. Each attention-head extracts the relation of feature map which contains knowledge about embedding procedure.
The below table and plot are sample results using ResNet and train on CIFAR100.
I use the same hyper-parameter for training each network, and only tune hyper-parameter of each distillation algorithm. However the results may be not optimal. All of the numerical values and plots are averages of five trials.
The teacher network is ResNet32 and Student is ResNet8, and the student network is well-converged (not over and under-fit) for evaluating each distillation algorithm performance precisely. Note that implemented ResNet has doubled depth.
| Methods | Last Accuracy | Best Accuracy |
|---|---|---|
| Student | 71.76 | 71.92 |
| Teacher | 78.96 | 79.08 |
| Soft-logits | 71.79 | 72.08 |
| FitNet | 72.74 | 72.96 |
| AT | 72.31 | 72.60 |
| FSP | 72.65 | 72.91 |
| DML | 73.27 | 73.47 |
| KD-SVD | 73.68 | 73.78 |
| KD-EID | 73.84 | 74.07 |
| FT | 73.35 | 73.50 |
| AB | 73.08 | 73.41 |
| RKD | 73.40 | 73.48 |
| MHGD | 73.98 | 74.30 |
- Upgrade this Repo. to TF2.0. :)