Sequence Distillation for Purely Sequence Trained Acoustic Models
Naoyuki Kanda, Yusuke Fujita, Kenji Nagamatsu
Abstract
This paper presents our exploration into teacher-student (TS) training for acoustic models (AMs) based on the lattice-free maximum mutual information technique. Whereas most previous studies of TS training used a frame-level distance between teacher and student models' distributions, we propose using the sequence-level temper-atured Kullback-Leibler divergence as a metric for TS training. In our experiment on the AMI meeting corpus, we prepared a strong teacher model consisting of a convolutional neural network, time delay neural network, and long short-term memory, which had 47.7M parameters and achieved a state-of-the-art word error rate (WER) of 18.05%. Whereas the small student AM (10.8M params. and 19.72% WER) trained by a frame-level TS training was able to fill only 43% of the WER gap between teacher and student AMs, the student AM trained by the proposed method achieved a 18.23% WER, filling 89% of the WER gap from the teacher AM. We also show that the frame-level TS training sometimes even degrades the performance of the student model whereas the proposed method consistently improved the accuracy.
BibTeX
@inproceedings{icassp2018_sequencedistilla,
title = {Sequence Distillation for Purely Sequence Trained Acoustic Models},
author = {Naoyuki Kanda and Yusuke Fujita and Kenji Nagamatsu},
booktitle = {ICASSP 2018},
year = {2018}
}