分层优化提升多语言语音模型性能
Objective Soups: Multilingual Multi-Task Modeling for Speech Processing
- 将识别与翻译任务分层优化,缓解目标冲突
- 在CoVoST v2等数据集上表现优于传统方法
- 轻量层选择机制降低计算开销,适合实际部署
多语言多任务语音处理(MSP)训练常因语音识别与翻译等任务间目标冲突而受阻。尽管多目标优化(MOO)试图对齐梯度更新,但任务增多时其效果下降,难以找到共同下降方向。本文探究三种多目标MSP范式,称为‘目标汤’配方,分别在不同优化层级应用MOO以缓解任务间冲突。为提高效率,提出轻量级层选择机制,仅在最冲突层计算避冲突梯度,显著减少计算与内存开销。在CoVoST v2、LibriSpeech和AISHELL-1上的大量实验表明,将识别与翻译分层的双层配方始终优于标准扁平优化。结果表明,分层多目标优化是构建先进MSP模型更有效且可扩展的方法。代码已开源。
原文摘要 · Abstract (English)
Training a single model for multilingual, multi-task speech processing (MSP) is severely hampered by conflicting objectives between tasks like speech recognition and translation. While multi-objective optimization (MOO) aims to align gradient updates, its effectiveness diminishes as the number of tasks grows, making it difficult to find a common descent direction. This raises a fundamental question: should highly conflicting objectives be optimized jointly or separated into a hierarchical structure? To address this question, this paper investigates three multi-objective MSP formulations, which we refer to as \textbf{objective soup recipes}. These formulations apply multi-objective optimization at different optimization levels to mitigate potential conflicts among all objectives. To ensure efficiency, we introduce a lightweight layer-selection mechanism that computes the conflict-avoiding gradient using only the most problematic layers, minimizing computational and memory overhead. Extensive experiments on CoVoST v2, LibriSpeech, and AISHELL-1 reveal that a bi-level recipe separating recognition and translation tasks consistently outperforms standard flat optimization. Our work demonstrates that hierarchical MOO is a more effective and scalable approach for building state-of-the-art MSP models. Our code has been released at https://github.com/afmsaif/Objective_Soups.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。