English

Prompt Tuning GPT-2 language model for parameter-efficient domain adaptation of ASR systems

Computation and Language 2022-07-25 v3 Machine Learning

Abstract

Automatic Speech Recognition (ASR) systems have found their use in numerous industrial applications in very diverse domains creating a need to adapt to new domains with small memory and deployment overhead. In this work, we introduce domain-prompts, a methodology that involves training a small number of domain embedding parameters to prime a Transformer-based Language Model (LM) to a particular domain. Using this domain-adapted LM for rescoring ASR hypotheses can achieve 7-13% WER reduction for a new domain with just 1000 unlabeled textual domain-specific sentences. This improvement is comparable or even better than fully fine-tuned models even though just 0.02% of the parameters of the base LM are updated. Additionally, our method is deployment-friendly as the learnt domain embeddings are prefixed to the input to the model rather than changing the base model architecture. Therefore, our method is an ideal choice for on-the-fly adaptation of LMs used in ASR systems to progressively scale it to new domains.

Keywords

Cite

@article{arxiv.2112.08718,
  title  = {Prompt Tuning GPT-2 language model for parameter-efficient domain adaptation of ASR systems},
  author = {Saket Dingliwal and Ashish Shenoy and Sravan Bodapati and Ankur Gandhe and Ravi Teja Gadde and Katrin Kirchhoff},
  journal= {arXiv preprint arXiv:2112.08718},
  year   = {2022}
}

Comments

Accepted at InterSpeech 2022