让语音识别模型按需压缩,高效又稳定。
MoME: Mixture of Matryoshka Experts for Audio-Visual Speech Recognition
- 用多尺度专家混合动态分配计算资源,支持灵活压缩。
- 在LRS2/LRS3上超越现有方法,参数少且抗噪强。
- 适合边缘设备部署,兼顾效率与可解释性。
大型语言模型(LLM)在音视频语音识别(AVSR)中表现强劲,但其高计算开销和对词元粒度的敏感性限制了其在资源受限场景的应用。现有的词元压缩方法需预先设定压缩率,输出固定长度,缺乏推理时的灵活性。马特里什卡表示学习(MRL)允许单个模型在不同粒度下运行,实现动态压缩。然而,现有MRL方法在训练时独立处理各尺度,导致跨尺度泛化能力弱、高压缩下的鲁棒性差、且可解释性不足。为此,我们提出MoME(马特里什卡专家混合),将稀疏的专家混合(MoE)引入基于MRL的LLM中。MoME通过top-k路由与共享专家,实现跨尺度和跨模态的动态容量分配;共享路由器确保不同粒度间专家激活的一致性,使高压缩序列也能受益于低压缩层的学习表示。在LRS2和LRS3上的实验表明,MoME在AVSR、ASR和VSR任务上均达到最先进性能,同时显著减少参数量,并保持噪声环境下的鲁棒性。该方法融合了MRL的适应性与MoE的高效性,为资源感知的语音识别提供可扩展、可解释的解决方案。
原文摘要 · Abstract (English)
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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。