Knowledge Distillation by On-the-Fly Native Ensemble
Abstract
Knowledge distillation is effective to train small and generalisable network models for meeting the low-memory and fast running requirements. Existing offline distillation methods rely on a strong pre-trained teacher, which enables favourable knowledge discovery and transfer but requires a complex two-phase training procedure. Online counterparts address this limitation at the price of lacking a highcapacity teacher. In this work, we present an On-the-fly Native Ensemble (ONE) strategy for one-stage online distillation. Specifically, ONE trains only a single multi-branch network while simultaneously establishing a strong teacher on-the- fly to enhance the learning of target network. Extensive evaluations show that ONE improves the generalisation performance a variety of deep neural networks more significantly than alternative methods on four image classification dataset: CIFAR10, CIFAR100, SVHN, and ImageNet, whilst having the computational efficiency advantages.
Keywords
Cite
@article{arxiv.1806.04606,
title = {Knowledge Distillation by On-the-Fly Native Ensemble},
author = {Xu Lan and Xiatian Zhu and Shaogang Gong},
journal= {arXiv preprint arXiv:1806.04606},
year = {2018}
}
Comments
To appear in NIPS2018