English

Knowledge Distillation by On-the-Fly Native Ensemble

Computer Vision and Pattern Recognition 2018-09-11 v2

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

R2 v1 2026-06-23T02:27:34.450Z