Improving accuracy of rare words for RNN-Transducer through unigram shallow fusion

Published in arxiv (Submitted to ASRU 2021), 2020

End-to-end automatic speech recognition (ASR) systems, such as recurrent neural network transducer (RNN-T), have become popular, but rare word remains a challenge. In this paper, we propose a simple, yet effective method called unigram shallow fusion (USF) to improve rare words for RNN-T. In USF, we extract rare words from RNN-T training data based on unigram count, and apply a fixed reward when the word is encountered during decoding. We show that this simple method can improve performance on rare words by 3.7% WER relative without degradation on general test set, and the improvement from USF is additive to any additional language model based rescoring. Then, we show that the same USF does not work on conventional hybrid system. Finally, we reason that USF works by fixing errors in probability estimates of words due to Viterbi search used during decoding with subword-based RNN-T.I

Download paper here

Recommended citation: Ravi, V., Gu, Y., Gandhe, A., Rastrow, A., Liu, L., Filimonov, D., Novotney, S. and Bulyko, I., 2020. Improving accuracy of rare words for RNN-Transducer through unigram shallow fusion. arXiv preprint arXiv:2012.00133.