Back to Search Start Over

Layer‐wise hint‐based training for knowledge transfer in a teacher‐student framework

Authors :
Ji‐Hoon Bae
Junho Yim
Nae‐Soo Kim
Cheol‐Sig Pyo
Junmo Kim
Source :
ETRI Journal, Vol 41, Iss 2, Pp 242-253 (2019)
Publication Year :
2019
Publisher :
Electronics and Telecommunications Research Institute (ETRI), 2019.

Abstract

We devise a layer‐wise hint training method to improve the existing hint‐based knowledge distillation (KD) training approach, which is employed for knowledge transfer in a teacher‐student framework using a residual network (ResNet). To achieve this objective, the proposed method first iteratively trains the student ResNet and incrementally employs hint‐based information extracted from the pretrained teacher ResNet containing several hint and guided layers. Next, typical softening factor‐based KD training is performed using the previously estimated hint‐based information. We compare the recognition accuracy of the proposed approach with that of KD training without hints, hint‐based KD training, and ResNet‐based layer‐wise pretraining using reliable datasets, including CIFAR‐10, CIFAR‐100, and MNIST. When using the selected multiple hint‐based information items and their layer‐wise transfer in the proposed method, the trained student ResNet more accurately reflects the pretrained teacher ResNet's rich information than the baseline training methods, for all the benchmark datasets we consider in this study.

Details

Language :
English
ISSN :
12256463, 22337326, and 20180152
Volume :
41
Issue :
2
Database :
Directory of Open Access Journals
Journal :
ETRI Journal
Publication Type :
Academic Journal
Accession number :
edsdoj.ff2810352481469b92afb433d0dbe54e
Document Type :
article
Full Text :
https://doi.org/10.4218/etrij.2018-0152