about
A Comprehensive Overhaul of Feature Distillation (arxiv.org)
2 points by pplonski86 on Apr 8, 2019 | hide | past | pdf | discuss on HN

In plain words: A small network learns from a big one by copying its internal features through a filter that clips unhelpful values and a distance measure that skips redundant parts. The student reached 21.65% error on a large image-recognition test, beating the bigger teacher it learned from.

Abstract

We investigate the design aspects of feature distillation methods achieving network compression and propose a novel feature distillation method in which the distillation loss is designed to make a synergy among various aspects: teacher transform, student transform, distillation feature position and distance function. Our proposed distillation loss includes a feature transform with a newly designed margin ReLU, a new distillation feature position, and a partial L2 distance function to skip redundant information giving adverse effects to the compression of student. In ImageNet, our proposed method achieves 21.65% of top-1 error with ResNet50, which outperforms the performance of the teacher network, ResNet152. Our proposed method is evaluated on various tasks such as image classification, object detection and semantic segmentation and achieves a significant performance improvement in all tasks. The code is available at https://sites.google.com/view/byeongho-heo/overhaul

Byeongho Heo, Jeesoo Kim, Sangdoo Yun, Hyojin Park, Nojun Kwak, Jin Young Choi
arXiv:1904.01866 · cs.CV, cs.LG · submitted Apr 3, 2019 · updated Aug 9, 2019
abstract · pdf · html · Accepted at ICCV 2019

add comment on HN