From Text to Talk: Audio-Language Model Needs Non-Autoregressive Joint Training
📄 From Text to Talk: Audio-Language Model Needs Non-Autoregressive Joint Training #语音对话系统 #扩散模型 #端到端 #多模态模型 #大语言模型 🔥 8.5/10 | 前25% | #语音对话系统 | #扩散模型 | #端到端 #多模态模型 学术质量 6.5/7 | 选题价值 1.5/2 | 复现加成 0.5 | 置信度 高 👥 作者与机构 第一作者:Tianqiao Liu(好未来教育集团 TAL Education Group,暨南大学 Guangdong Institute of Smart Education) 通讯作者:Xueyi Li(暨南大学 Guangdong Institute of Smart Education) 作者列表:Tianqiao Liu(好未来教育集团,暨南大学)、Xueyi Li(暨南大学)、Hao Wang(北京大学)、Haoxuan Li(北京大学)、Zhichao Chen(北京大学)、Weiqi Luo(暨南大学)、Zitao Liu(暨南大学) 💡 毒舌点评 论文对端到端语音模型中文本与音频生成范式错配问题的洞察一针见血,并给出了一个理论上优雅、实验上有效的混合训练框架,是当前S2S建模思路的一次重要升级。但论文对模型推理时块级扩散的计算开销分析着墨不多,且训练数据依赖大量合成语音(如CosyVoice2生成),其在真实复杂声学环境下的泛化能力仍是潜在挑战。 🔗 开源详情 代码:论文明确提供了代码仓库链接:https://github.com/ai4ed/TtT。 模型权重:论文中未提及预训练模型权重(如Pretrain+TtT的检查点)的公开下载链接。 数据集:论文中使用的训练数据大多为公开数据集(如AISHELL, LibriSpeech, VoiceAssistant-400K等),但具体的数据处理脚本和混合配方未完全开源。评估数据集如URO-Bench、Audio-QA集等为公开基准。 Demo:论文中未提及提供在线演示(Demo)。 复现材料:论文提供了极其详细的训练细节(超参数、优化器设置、训练策略概率)、模型配置(基于Qwen2.5)、以及架构和注意力机制的示意图(图2, 3),并附有详尽的附录。这些构成了坚实的复现基础。 引用的开源项目:论文依赖并引用了多个开源项目作为基础组件,主要包括: 音频分词器/解码器:GLM-4-Voice (Zeng et al., 2024)。 主干LLM:Qwen2.5系列。 ASR评估工具:Whisper (Radford et al., 2023)、Paraformer。 TTS数据生成:CosyVoice2。 训练框架:DeepSpeed。 论文中提及的开源计划:论文中未提及额外的开源计划(如未来发布模型权重或扩展数据)。 📌 核心摘要 本文针对现有端到端语音到语音(S2S)模型用统一自回归(AR)方法建模文本和音频所存在的范式错配问题,提出了“Text-to-Talk”(TtT)框架。核心问题在于,文本生成是强序列依赖的(目标-目标依赖),而音频生成更依赖输入源(源-目标依赖),强行用AR约束音频会引入不必要的误差传播。方法核心是设计一个混合生成框架,在同一个Transformer中,对文本使用标准AR建模,对音频段使用吸收离散扩散(一种NAR范式)建模,并证明了这种联合训练目标是目标联合分布的上界。与已有方法相比,新在两点:1)首次识别并形式化了文本与音频在依赖结构上的不对称性;2)提出了一个统一的架构和训练框架来适配这种不对称性,而非强行统一生成范式。主要实验结果显示,TtT在Audio-QA、ASR、AAC和URO-Bench等多个基准上,一致超越了纯AR和纯NAR的基线模型。例如,在3B参数规模下,TtT在多个ASR数据集上的WER大幅优于Qwen2.5-3B (AR),在Audio-QA任务上也显著提升。实际意义在于,为构建更自然、高效、符合生成特性的端到端语音交互系统提供了新思路。主要局限性包括:1)块级扩散推理的效率需要进一步评估;2)模型性能对大规模多模态预训练数据(约200B tokens)有一定依赖;3)尽管在轻量级模型中表现优异,但与某些超大参数量模型(如GLM-4-Voice)在综合基准上仍有差距。 ...