Efficient Compression of Multitask Multilingual Speech Models

Thomas Palmeira Ferraz
2024-05-02
Abstract:Whisper is a multitask and multilingual speech model covering 99 languages. It yields commendable automatic speech recognition (ASR) results in a subset of its covered languages, but the model still underperforms on a non-negligible number of under-represented languages, a problem exacerbated in smaller model versions. In this work, we examine its limitations, demonstrating the presence of speaker-related (gender, age) and model-related (resourcefulness and model size) bias. Despite that, we show that only model-related bias are amplified by quantization, impacting more low-resource languages and smaller models. Searching for a better compression approach, we propose DistilWhisper, an approach that is able to bridge the performance gap in ASR for these languages while retaining the advantages of multitask and multilingual capabilities. Our approach involves two key strategies: lightweight modular ASR fine-tuning of whisper-small using language-specific experts, and knowledge distillation from whisper-large-v2. This dual approach allows us to effectively boost ASR performance while keeping the robustness inherited from the multitask and multilingual pre-training. Results demonstrate that our approach is more effective than standard fine-tuning or LoRA adapters, boosting performance in the targeted languages for both in- and out-of-domain test sets, while introducing only a negligible parameter overhead at inference.
Computation and Language,Artificial Intelligence,Sound,Audio and Speech Processing
What problem does this paper attempt to address?
### Problems the paper attempts to solve This paper aims to solve the problem that multi - task and multi - language speech models (especially OpenAI's Whisper model) perform poorly on low - resource languages in small - scale model versions. Specifically, the paper focuses on the following aspects: 1. **Model performance degradation**: - **Low - resource languages**: The Whisper model has poor automatic speech recognition (ASR) performance on some low - resource languages, especially in smaller model versions. - **Quantization impact**: Although quantization techniques can compress the model, they will amplify the model - related biases and further affect the performance of low - resource languages. 2. **Bias analysis**: - **Speaker - related biases**: Including the influence of factors such as gender and age on model performance. - **Model - related biases**: Including the influence of resource richness and model size on performance. 3. **Improving compression methods**: - **DistilWhisper**: A new compression method is proposed. Through lightweight modular ASR fine - tuning and knowledge distillation, the ASR performance of low - resource languages is improved while retaining multi - task and multi - language capabilities. ### Specific problems - **How to effectively improve the ASR performance of low - resource languages?** - **How to reduce the number of model parameters while maintaining the multi - task and multi - language capabilities of the model?** - **How to mitigate the negative impact of quantization techniques on model performance?** ### Solutions The paper proposes the **DistilWhisper** method, which mainly includes the following two key strategies: 1. **Lightweight modular ASR fine - tuning**: - Use language - specific experts to fine - tune `whisper - small` to improve the ASR performance of specific languages. 2. **Knowledge distillation**: - Extract knowledge from `whisper - large - v2` and distill it into smaller models to improve performance. Through these strategies, DistilWhisper can effectively improve the ASR performance of low - resource languages without significantly increasing the parameter overhead during inference.