arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2510.04136eess.AScs.CVcs.SD

MoME: 用于视听语音识别的套娃专家混合模型

MoME: Mixture of Matryoshka Experts for Audio-Visual Speech Recognition

Imperial College London(伦敦帝国学院);Meta AI;NatWest AI Research(NatWest人工智能研究)

Umberto Cappellazzo, Minsu Kim, Pingchuan Ma, Honglie Chen, Xubo Liu, Stavros Petridis, Maja Pantic

AI总结:

针对大语言模型在视听语音识别中计算成本高、现有MRL方法跨尺度泛化弱的问题,提出融合稀疏MoE与MRL的MoME框架,在多任务上达SOTA且参数量更少、噪声鲁棒性强。

摘要:

大语言模型(LLM)近期在视听语音识别(AVSR)领域展现出强劲潜力,但其高计算需求以及对token粒度的敏感性限制了其在资源受限场景下的实用性。Token压缩方法能够降低推理成本,但这类方法需要预先固定压缩率,且仅能输出单一固定长度的结果,无法在推理时灵活平衡信息密度与效率。套娃表示学习(Matryoshka representation learning, MRL)通过让单个模型支持多种token粒度运行解决了这一问题,可动态调整压缩率。然而,当前基于MRL的方法在训练时独立处理每个尺度,限制了跨尺度泛化能力、高压缩比下的鲁棒性以及可解释性。为克服这些局限,我们提出MoME(Mixture of Matryoshka Experts,套娃专家混合模型)——一种将稀疏专家混合(Mixture-of-Experts, MoE)集成到基于MRL的AVSR大语言模型中的新型框架。MoME为冻结的LLM增设top-k路由专家与共享专家,可实现跨尺度与跨模态的动态容量分配;共享路由器可促进不同粒度下专家激活的一致性,让压缩序列能够从更低压缩率下学习到的表示中获益。在LRS2和LRS3数据集上的实验表明,MoME在AVSR、自动语音识别(ASR)和视觉语音识别(VSR)任务上均达到了当前最优性能,同时所需参数量显著更少,且在噪声环境下仍能保持鲁棒性。MoME融合了MRL的自适应性与MoE的高效性,为资源感知型语音识别提供了一种可扩展且具备可解释性的解决方案。

英文摘要:

Large language models (LLMs) have recently shown strong potential in audio-visual speech recognition (AVSR), but their high computational demands and sensitivity to token granularity limit their practicality in resource-constrained settings. Token compression methods can reduce inference cost, but they require fixing a compression rate in advance and produce a single fixed-length output, offering no flexibility to balance information density and efficiency at inference time. Matryoshka representation learning (MRL) addresses this by enabling a single model to operate across multiple token granularities, allowing compression rates to be adjusted dynamically. However, current MRL-based methods treat each scale independently during training, limiting cross-scale generalization, robustness at high compression, and interpretability. To overcome these limitations, we propose MoME (Mixture of Matryoshka Experts), a novel framework that integrates sparse Mixture-of-Experts (MoE) into MRL-based LLMs for AVSR. MoME augments a frozen LLM with top-k routed and shared experts, allowing dynamic capacity allocation across scales and modalities. A shared router promotes consistent expert activation across granularities, enabling compressed sequences to benefit from representations learned at lower compression. Experiments on LRS2 and LRS3 demonstrate that MoME achieves state-of-the-art performance across AVSR, ASR, and VSR tasks, while requiring significantly fewer parameters and maintaining robustness under noise. MoME unifies the adaptability of MRL with the efficiency of MoE, offering a scalable and interpretable solution for resource-aware speech recognition.

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

↑