Enhancing Quantised End-to-End ASR Models via Personalisation

Qiuming Zhao, Guangzhi Sun, Chao Zhang, Mingxing Xu, Thomas Fang Zheng
9/17/2023
cs.SDcs.AIeess.AS

No official code found. Best implementation: qmgzhao/PQM (7 stars, MIT, last commit 2023-11-22).

Abstract

Recent end-to-end automatic speech recognition (ASR) models have become increasingly larger, making them particularly challenging to be deployed on resource-constrained devices. Model quantisation is an effective solution that sometimes causes the word error rate (WER) to increase. In this paper, a novel strategy of personalisation for a quantised model (PQM) is proposed, which combines speaker adaptive training (SAT) with model quantisation to improve the performance of heavily compressed models. Specifically, PQM uses a 4-bit NormalFloat Quantisation (NF4) approach for model quantisation and low-rank adaptation (LoRA) for SAT. Experiments have been performed on the LibriSpeech and the TED-LIUM 3 corpora. Remarkably, with a 7x reduction in model size and 1% additional speaker-specific parameters, 15.1% and 23.3% relative WER reductions were achieved on quantised Whisper and Conformer-based attention-based encoder-decoder ASR models respectively, comparing to the original full precision models.

View on arXivView PDF

Code Implementations(1)

qmgzhao/PQMCommunity100%
MIT
71PythonSep 9, 20232 years agoMIT

Similar Papers

Cite this paper

@article{zhao2023enhancing,
  title  = {Enhancing Quantised End-to-End ASR Models via Personalisation},
  author = {Qiuming Zhao and Guangzhi Sun and Chao Zhang and Mingxing Xu and Thomas Fang Zheng},
  year   = {2023},
  eprint = {2309.09136},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2309.09136v1}
}

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/2309.09136.svg)](https://codeofpaper.com/paper/2309.09136)

Discussion