arXiv:2409.19878cs.SDeess.AS2024-09中稿 · ICASSP 2025被引 22

用分层路由和动态阈值的LoRA专家混合,高效适配多口音语音识别模型。

HDMoLE: Mixture of LoRA Experts with Hierarchical Routing and Dynamic Thresholds for Fine-Tuning LLM-based ASR Models

  • 分层路由让不同口音对应专属LoRA专家,提升跨域协作。
  • 仅用9.6%参数量达全微调性能,通用域表现几乎无损。
  • 适合资源有限但需多口音适配的语音识别场景。

将大型语言模型(LLM)与自动语音识别(ASR)结合在通用领域表现优异。尽管全参数微调常用于将预训练的LLM-based ASR模型适配到特定领域,但其计算开销大且显著降低通用性能。本文提出一种新型参数高效的多领域微调方法HDMoLE,可避免灾难性遗忘,通过结合低秩适应(LoRA)与专家混合(MoE),实现对任意线性层的泛化。分层路由建立LoRA专家与口音领域的明确对应关系,促进专家间跨域协作;不同于静态的Top-K激活策略,动态阈值可自适应调整每层MoE激活的专家数量。在多口音及标准普通话数据集上的实验表明,将HDMoLE应用于LLM-based ASR模型的投影模块,在目标多口音领域达到接近全微调的性能,仅需全微调9.6%的可训练参数,且源通用域性能损失极小。

原文摘要 · Abstract (English)

Recent advancements in integrating Large Language Models (LLM) with automatic speech recognition (ASR) have performed remarkably in general domains. While supervised fine-tuning (SFT) of all model parameters is often employed to adapt pre-trained LLM-based ASR models to specific domains, it imposes high computational costs and notably reduces their performance in general domains. In this paper, we propose a novel parameter-efficient multi-domain fine-tuning method for adapting pre-trained LLM-based ASR models to multi-accent domains without catastrophic forgetting named \textit{HDMoLE}, which leverages hierarchical routing and dynamic thresholds based on combining low-rank adaptation (LoRA) with the mixer of experts (MoE) and can be generalized to any linear layer. Hierarchical routing establishes a clear correspondence between LoRA experts and accent domains, improving cross-domain collaboration among the LoRA experts. Unlike the static Top-K strategy for activating LoRA experts, dynamic thresholds can adaptively activate varying numbers of LoRA experts at each MoE layer. Experiments on the multi-accent and standard Mandarin datasets demonstrate the efficacy of HDMoLE. Applying HDMoLE to an LLM-based ASR model projector module achieves similar performance to full fine-tuning in the target multi-accent domains while using only 9.6% of the trainable parameters required for full fine-tuning and minimal degradation in the source general domain.

语音识别LoRA专家混合参数效率

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。