Massively Multilingual Shallow Fusion with Large Language Models (2302.08917v1)
Abstract: While LLMs (LLM) have made impressive progress in natural language processing, it remains unclear how to utilize them in improving automatic speech recognition (ASR). In this work, we propose to train a single multilingual LLM (LM) for shallow fusion in multiple languages. We push the limits of the multilingual LM to cover up to 84 languages by scaling up using a mixture-of-experts LLM, i.e., generalist LLM (GLaM). When the number of experts increases, GLaM dynamically selects only two at each decoding step to keep the inference computation roughly constant. We then apply GLaM to a multilingual shallow fusion task based on a state-of-the-art end-to-end model. Compared to a dense LM of similar computation during inference, GLaM reduces the WER of an English long-tail test set by 4.4% relative. In a multilingual shallow fusion task, GLaM improves 41 out of 50 languages with an average relative WER reduction of 3.85%, and a maximum reduction of 10%. Compared to the baseline model, GLaM achieves an average WER reduction of 5.53% over 43 languages.
- Ke Hu (57 papers)
- Tara N. Sainath (79 papers)
- Bo Li (1107 papers)
- Nan Du (66 papers)
- Yanping Huang (40 papers)
- Andrew M. Dai (40 papers)
- Yu Zhang (1399 papers)
- Rodrigo Cabrera (3 papers)
- Zhifeng Chen (65 papers)
- Trevor Strohman (38 papers)