End-to-End Training for Discrete Token LLM based TTS System
📄 End-to-End Training for Discrete Token LLM based TTS System #语音合成 #多任务学习 #强化学习 #流匹配 7.6/10 | 创新 1.4/2 | 严谨 1.1/1.5 | 实验 0.9/1.5 | 清晰 1/1 | 影响 1/1.5 | 开源 0.2/1.5 | 复现 0.5/0.5 | 工程 1.5/1.5 ✅ 7.6/10 | 前50% | #语音合成 | #多任务学习 | #强化学习 #流匹配 | arxiv 👥 作者与机构 论文作者:Changfeng Gao, Yong Ren, Jun Yuan, Ye Bai, Zhao You, ShiDong Shang。单位未在摘要或作者列表中明确列出,但论文标题和内容表明与小米AI实验室相关。根据作者信息,机构为小米AI实验室与南京大学。 💡 毒舌点评 SOTA声明的谨慎性:论文声称达到“new SOTA result”,但在主实验表格(Table 1)中,与JoyVoice(WER 0.97% zh, 1.69% en)和CosyVoice3-1.5B(WER 1.12% zh, 2.21% en)等模型相比,提升幅度有限(如中文WER从0.97%降至0.78%)。考虑到其使用了0.6B参数的LLM,性能增益是合理的,但“SOTA”的表述可能需要更谨慎地限定在特定模型规模下。 实验设计的局限:论文的核心实验主要基于内部大规模数据集训练的模型进行自比较(Table 1中的Stage1/2/3和w/o E2E),以及与外部模型的有限对比。然而,论文未提供在相同数据、相同基础模型(如Qwen3-0.6B)上与非E2E训练基线(w/o E2E-training)的直接、公平对比的具体实现细节(如是否使用了完全相同的训练数据子集和超参数),这削弱了“E2E训练关键”这一结论的绝对说服力。 组件贡献的归因模糊:虽然消融研究(Table 2, 3)移除了\(L_{LM}\)或\(L_{FM}\),证明了它们的重要性,但论文未深入分析三阶段训练中每个阶段(特别是Stage 2的独立微调)对最终性能的具体贡献比例。Stage 2允许为不同模块使用不同数据,这引入了额外变量,其效果与E2E优化本身的效果未被完全解耦。 理论分析的实用性存疑:Section 2.4的信息论分析将Tokenizer训练形式化为源编码问题,概念上有趣,但公式推导(如公式14)较为初步,且与实验结果的直接关联较弱(如Table 4中H和I的微小差异如何对应显著的性能提升?)。这部分更像理论点缀,而非深入的机理分析。 开源与可复现性短板:尽管论文在训练细节上描述详尽,但完全未提供代码、模型权重或内部数据集的获取途径。对于一个声称方法更简单、性能更优的框架,缺乏开源极大地限制了社区验证和直接应用其价值。 📌 核心摘要 本文提出了一种用于基于离散令牌的LLM的TTS系统的端到端(E2E)训练框架。该框架统一优化了语音分词器、自回归LLM、基于流匹配(FM)的声码器和一个辅助奖励模型(RM)。核心方法包括:1)通过一阶损失(\(L_1\))联合训练分词器,使其直接适应下游LLM预测、FM重构和RM识别任务;2)通过二阶损失(\(L_2\))在LLM生成的令牌分布上优化系统,缓解训练-测试不匹配;3)设计了三阶段训练流程以稳定优化。实验表明,该框架在Seed-TTS-Eval基准上,使用0.6B LLM和0.5B FM模型,取得了0.78%(中文)和1.56%(英文)的WER,达到了SOTA水平。消融研究验证了各损失项和端到端优化的重要性。理论分析从信息论角度探讨了高质量语音令牌应具备高熵和强时序依赖性。 ...