用最优传输正则化提升语音与文本对齐,增强语音大模型泛化能力
Optimal Transport Regularization for Speech Text Alignment in Spoken Language Models
- 将语音-文本对齐建模为最优传输问题,通过运输计划生成正则化损失
- 在多语言语音识别中显著改善跨数据集泛化性能,提升对齐效果
- 无需额外标签或参数,可无缝融入现有语音大模型训练流程
语音语言模型(SLMs)作为将大语言模型扩展至语音输入的新兴方向,虽展现出巨大潜力,但现有研究显示其在跨数据集泛化上仍表现不佳,即便针对相同语言和任务。这反映出语音与文本表示间存在显著模态差距。语音嵌入的高度可变性可能导致模型依赖非预期的语音特征,从而损害泛化能力。为此,本文提出最优传输正则化(OTReg),将语音-文本对齐建模为最优传输问题,每轮训练中通过计算最优传输计划建立语音与转录本嵌入间的结构化对应关系,并基于此设计正则化损失以优化模型生成更对齐的语音嵌入。该方法轻量且不引入额外标签或可学习参数,可无缝集成至现有训练流程。大量多语言自动语音识别实验表明,OTReg有效缓解了模态差距,提升了语音-文本对齐度,进而显著增强了模型在多样数据集上的泛化性能。
原文摘要 · Abstract (English)
Spoken Language Models (SLMs), which extend Large Language Models (LLMs) to perceive speech inputs, have gained increasing attention for their potential to advance speech understanding tasks. However, despite recent progress, studies show that SLMs often struggle to generalize across datasets, even for trained languages and tasks, raising concerns about whether they process speech in a text-like manner as intended. A key challenge underlying this limitation is the modality gap between speech and text representations. The high variability in speech embeddings may allow SLMs to achieve strong in-domain performance by exploiting unintended speech variations, ultimately hindering generalization. To mitigate this modality gap, we introduce Optimal Transport Regularization (OTReg), a method that formulates speech-text alignment as an optimal transport problem and derives a regularization loss to improve SLM training. In each training iteration, OTReg first establishes a structured correspondence between speech and transcript embeddings by determining the optimal transport plan, then incorporates the regularization loss based on this transport plan to optimize SLMs in generating speech embeddings that align more effectively with transcript embeddings. OTReg is lightweight, requiring no additional labels or learnable parameters, and integrates seamlessly into existing SLM training procedures. Extensive multilingual ASR experiments demonstrate that OTReg enhances speech-text alignment, mitigates the modality gap, and consequently improves SLM generalization across diverse datasets.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。