Massively Multilingual Shallow Fusion with Large Language Models

Ke Hu, Tara N. Sainath, Bo Li, Nan Du, Yanping Huang +5 more
2/17/2023
cs.CLcs.LG

No confident code implementation found for this paper. 3 possible matches can be shown below.

Abstract

While large language models (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 language model (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 language model (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.

View on arXivView PDF

Code Implementations

No confident code match yet

We couldn't find an author-owned or strongly-evidenced community implementation for this paper. 3 weaker matches are hidden by default — verify before relying on them.

No code implementations found yet.

Know of an implementation? Let us know in the comments below!

Similar Papers

Cite this paper

@article{hu2023massively,
  title  = {Massively Multilingual Shallow Fusion with Large Language Models},
  author = {Ke Hu and Tara N. Sainath and Bo Li and Nan Du and Yanping Huang and Andrew M. Dai and Yu Zhang and Rodrigo Cabrera and Zhifeng Chen and Trevor Strohman},
  year   = {2023},
  eprint = {2302.08917},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2302.08917v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2302.08917.svg)](https://codeofpaper.com/paper/2302.08917)

Discussion