Internal Language Model Training for Domain-Adaptive End-to-End Speech Recognition
This work provides a significant improvement in domain adaptation for end-to-end speech recognition systems, particularly beneficial for deploying ASR models in new domains with external language models.
This paper proposes an internal language model training (ILMT) method to improve domain-adaptive end-to-end (E2E) speech recognition. By training an internal language model within the E2E model, the method achieves up to 31.5% relative word error rate reduction on out-of-domain LibriSpeech and 11.4% on in-domain Microsoft production test sets, compared to standard E2E training with Shallow Fusion.
The efficacy of external language model (LM) integration with existing end-to-end (E2E) automatic speech recognition (ASR) systems can be improved significantly using the internal language model estimation (ILME) method. In this method, the internal LM score is subtracted from the score obtained by interpolating the E2E score with the external LM score, during inference. To improve the ILME-based inference, we propose an internal LM training (ILMT) method to minimize an additional internal LM loss by updating only the E2E model components that affect the internal LM estimation. ILMT encourages the E2E model to form a standalone LM inside its existing components, without sacrificing ASR accuracy. After ILMT, the more modular E2E model with matched training and inference criteria enables a more thorough elimination of the source-domain internal LM, and therefore leads to a more effective integration of the target-domain external LM. Experimented with 30K-hour trained recurrent neural network transducer and attention-based encoder-decoder models, ILMT with ILME-based inference achieves up to 31.5% and 11.4% relative word error rate reductions from standard E2E training with Shallow Fusion on out-of-domain LibriSpeech and in-domain Microsoft production test sets, respectively.