arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2405.00966cs.CLcs.AIcs.SDeess.AS

多任务多语言语音模型的高效压缩

Efficient Compression of Multitask Multilingual Speech Models

PSL(巴黎文理研究大学);Meta AI;NAVER LABS Europe(NAVER欧洲实验室)

Thomas Palmeira Ferraz

AI总结:

本文提出 DistilWhisper,通过语言特定专家的轻量级模块化微调与 whisper-large-v2 知识蒸馏相结合,在保持多任务多语言能力的同时,高效提升低资源语言的 ASR 性能。

摘要:

Whisper 是一个覆盖 99 种语言的多任务多语言语音模型。它在其覆盖语言的一个子集上取得了值得称赞的自动语音识别(ASR)结果,但该模型在不可忽视数量的低资源语言上仍然表现不佳,这一问题在较小模型版本中更加严重。在这项工作中,我们考察了其局限性,证明了与说话人相关的(性别、年龄)和与模型相关的(资源丰富程度和模型规模)偏差的存在。尽管如此,我们表明只有与模型相关的偏差会被量化放大,从而对低资源语言和较小模型产生更大影响。为了寻找更好的压缩方法,我们提出了 DistilWhisper,该方法能够在保持多任务和多语言能力优势的同时,弥合这些语言在 ASR 上的性能差距。我们的方法涉及两个关键策略:使用语言特定专家对 whisper-small 进行轻量级模块化 ASR 微调,以及从 whisper-large-v2 进行知识蒸馏。这种双重方法使我们能够有效提升 ASR 性能,同时保持从多任务和多语言预训练中继承的鲁棒性。结果表明,我们的方法比标准微调或 LoRA 适配器更有效,在域内和域外测试集上均提升了目标语言的性能,同时在推理时仅引入可忽略不计的参数开销。

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑