Knowledge Distillation via Instance Relationship Graph

By Yufan Liu, Jiajiong Cao, Bing Li, Chunfen Yuan, Weiming Hu, Yangxi Li, Yun-feng Duan. 2019

329 citations recorded by the source corpus.

The key challenge of knowledge distillation is to extract general, moderate and sufficient knowledge from a teacher network to guide a student network. In this paper, a novel Instance Relationship Graph (IRG) is proposed for knowledge distillation. It models three kinds of knowledge, including instance features, instance relationships and feature space transformation, while the latter two kinds of knowledge are neglected by previous methods. Firstly, the IRG is constructed to model the distilled knowledge of one network layer, by considering instance features and instance relationships as vertexes and edges respectively. Secondly, an IRG transformation is proposed to models the feature space transformation across layers. It is more moderate than directly mimicking the features at intermediate layers. Finally, hint loss functions are designed to force a student's IRGs to mimic the structures of a teacher's IRGs. The proposed method effectively captures the knowledge along the whole network via IRGs, and thus shows stable convergence and strong robustness to different network architectures. In addition, the proposed method shows superior performance over existing methods on datasets of various scales.

Subfields: Artificial Intelligence, Computer Vision and Pattern Recognition

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Artificial Intelligence

  1. Ziwei Liu
  2. Dacheng Tao
  3. Chen Change Loy
  4. Lin Wang
  5. Kuk-Jin Yoon
  6. Rameswar Panda
  7. Chen Qian
  8. Fei Wang

In Computer Vision and Pattern Recognition

  1. Dacheng Tao
  2. Luc Van Gool
  3. Xiaogang Wang
  4. Ming-Hsuan Yang
  5. Chen Change Loy
  6. Yi Yang
  7. Trevor Darrell
  8. Jitendra Malik

Citations in the corpus