Knowledge Distillation for Sequence Model

By Mingkun Huang, Yongbin You, Zhehuai Chen, Yanmin Qian, Kai Yu. 2018

70 citations recorded by the source corpus.

Knowledge distillation , or teacher-student training , has been effectively used to improve the performance of a relatively simpler deep learning model (the student) using a more complex model (the teacher). It is usually done by minimizing the Kullback-Leibler divergence (KLD) between the output distributions of the student and the teacher at each frame. However, the gain from frame-level knowledge distillation is limited for sequence models such as Connectionist Temporal Classification (CTC), due to the mismatch between the sequence-level criterion used in teacher model training and the frame-level criterion used in distillation. In this paper, sequence-level knowledge distillation is proposed to achieve better distillation performance. Instead of calculating a teacher posterior distribution given the feature vector of the current frame, sequence training criterion is employed to calculate the posterior distribution given the whole utterance and the teacher model. Experiments are conducted on both English Switchboard corpus and a large Chinese corpus. The proposed approach achieves significant and consistent improvements over the traditional frame-level knowledge distillation using both labeled and unlabeled data.

Subfields: Artificial Intelligence, Computer Vision and Pattern Recognition, Signal Processing

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. Dacheng Tao
  2. Xinchao Wang
  3. Jianping Gou
  4. Mingli Song
  5. S. Maybank
  6. Baosheng Yu
  7. Jingwen Ye
  8. Jie Song

In Computer Vision and Pattern Recognition

  1. Dacheng Tao
  2. Xinchao Wang
  3. Luc Van Gool
  4. Yu Qiao
  5. Mingli Song
  6. Xiaogang Wang
  7. Junjie Yan
  8. Ming-Hsuan Yang

In Signal Processing

  1. Dacheng Tao
  2. Yu Qiao
  3. S. Maybank
  4. Mingli Song
  5. Junjie Yan
  6. Baosheng Yu
  7. Jingwen Ye
  8. Jianping Gou

Citations in the corpus