[控场AI]
· 5 分钟阅读· 2,506 字

WaveGen:真正端到端波形生成,告别中间表示

WaveGen:真正端到端波形生成,告别中间表示

WaveGen用单一扩散模型实现文本到波形直接生成,无需外部解码器,简化TTS管线。

WaveGen提出了一种端到端的文本到语音合成方法,使用单一扩散模型直接从文本生成波形,摒弃了传统多阶段管线中的梅尔频谱、神经音频编解码器或VAE潜变量等中间表示。其核心是Spectral Forcing技术,让模型内部同时学习频谱轨迹和波形轨迹,配合解耦扩散头和多尺度频谱v-loss稳定训练,并移除了时长预测器和外部语义模块。模型从0.1B扩展到0.6B参数,在Emilia数据集上训练,Seed-en基准达到WER 1.47和SIM 0.75,展示了端到端TTS的可行性和可扩展性。该工作开源了代码和checkpoint,为TTS架构演进提供了新思路。

WaveGen论文标题

语音合成领域长期依赖“文本→中间表示→波形”的多阶段管线,中间表示要么是梅尔频谱加声码器,要么是神经音频编解码器的离散token,要么是VAE潜变量。这些模块各自训练、各自调参,误差逐级累积,工程复杂度居高不下。一篇新论文提出的WaveGen试图彻底拆掉这条流水线:用一个扩散模型直接从文本生成波形,不依赖任何外部解码器、时长预测器或自监督语义表征。

为什么“端到端”说起来容易做起来难

真正端到端的TTS面临一个根本矛盾:文本和波形在时间尺度上相差悬殊——一句话对应的波形采样点动辄数万,而文本只有几十个token。扩散模型要在如此高的维度上直接建模,还要在生成过程中自行完成文本与语音的对齐,难度远高于先生成低帧率中间表示再上采样。

现实中大多数号称“端到端”的系统都做了妥协。常见做法包括:用神经音频编解码器把波形压缩成离散token,用梅尔频谱加声码器两段式合成,或者引入BERT文本嵌入、Wav2Vec 2.0语义特征以及REPA类语义蒸馏来稳定训练。这些手段确实有效,但每一个外部模块都意味着额外的训练阶段、额外的数据依赖和额外的部署负担。

WaveGen的核心主张是:这些妥协不是必需的。

扩散模型是一类生成模型,其核心思想是通过逐步去噪的过程来生成数据:先从纯噪声开始,模型学习如何一步步去除噪声,最终恢复出目标样本。在语音合成中,扩散模型可以建模波形或频谱的分布。与传统生成模型相比,扩散模型训练稳定且能生成高质量样本,但推理需要多次迭代,计算开销较大。WaveGen利用扩散模型直接在波形空间生成,挑战在于波形维度极高,需要精心设计训练目标来稳定高维扩散过程。

神经音频编解码器(neural audio codec)是一种将波形压缩为离散token的模型,通常由编码器、量化器和解码器组成,如SoundStream、EnCodec等。它能够大幅降低波形序列长度,使语言模型或扩散模型更容易建模。然而,编解码器本身需要单独训练,且量化过程会引入信息损失,可能影响合成音质。WaveGen刻意避免使用这类外部编解码器,以消除其对训练数据和部署的额外依赖。

REPA(Representation Alignment)是一种训练辅助技术,通过将模型内部表征与预训练自监督模型(如Wav2Vec 2.0)的特征对齐来加速扩散模型收敛。类似地,Wav2Vec 2.0是自监督语音表征模型,能提取丰富的语义和声学特征。这些方法虽能提升训练稳定性,但引入了对外部预训练模型的依赖,增加了训练复杂性和数据授权限制。WaveGen的目标是摒弃这些辅助手段,实现更简洁的一阶段训练。

Spectral Forcing:把频谱轨迹学进模型内部

WaveGen的关键设计是Spectral Forcing。它让单一模型同时学习两条轨迹:一条是模型内部的频谱轨迹,另一条是目标波形轨迹。频谱轨迹起到“内部脚手架”的作用,帮助扩散过程在高压缩比下保持稳定,但它完全在模型内部产生,不需要外部声码器或编解码器参与。

与之配套的是解耦扩散头(disentangled diffusion heads)架构和多尺度频谱v-loss训练目标。解耦头让不同扩散阶段各司其职,多尺度频谱损失则在多个频率分辨率上约束生成质量。文本-语音对齐同样在模型内部完成,时长预测器被彻底移除。

这套设计的直接收益是训练流程的简化:单一架构、开源数据、一阶段训练,不需要REPA之类的辅助优化手段。

规模扩展与实测表现

论文验证了端到端TTS的可扩展性,将模型从0.1B参数扩展到0.6B参数。在Emilia训练数据集上,WaveGen报告了当前最优的性能表现;在Seed-en基准上,词错误率(WER)为1.47,说话人相似度(SIM)为0.75。

这两个数字值得放在上下文里看。WER 1.47意味着合成语音的可懂度已经接近实用门槛之上;SIM 0.75说明音色克隆的保真度可观。更你可能没注意到达成这些指标的条件:没有时长预测器,没有外部语义模块,没有多阶段训练。这削弱了“端到端必然牺牲质量”的固有印象。

当然,单一基准上的数字不足以覆盖全部场景。跨语言、情感控制、长文本稳定性等维度是否同样成立,还需要社区在开源代码和checkpoint基础上进一步验证。

词错误率(WER) 是语音识别中衡量识别准确度的指标,计算为替换、插入和删除错误词数与总词数的比值,越低越好。在TTS评估中,通常用ASR系统转写合成语音,再计算WER以反映可懂度。说话人相似度(SIM) 则通过说话人验证模型提取嵌入,计算合成语音与目标说话人语音的余弦相似度,衡量音色克隆的保真度,越高越好。Seed-en基准是专门用于评估TTS系统性能的英文测试集,包含多种说话人和文本,WaveGen在此达到WER 1.47和SIM 0.75,表明其合成语音既清晰又接近目标音色。

对TTS工程实践意味着什么

如果WaveGen的路线可复现,TTS系统的部署形态可能发生变化。传统管线中,声码器、编解码器、对齐模块各自有版本管理和推理开销,端到端单模型把这些都收拢为一个推理图。对于需要低延迟或边缘部署的场景,这种简化有实际价值。

另一层影响在数据侧。摆脱对BERT嵌入和Wav2Vec 2.0特征的依赖,意味着训练不再受限于这些预训练模型的语种覆盖和授权条件,理论上可以用更纯粹的目标语种数据训练。

不过要清醒看待:内部频谱轨迹本质上仍是一种隐式中间表示,只是它不再被外部模块固化。它能否像梅尔频谱那样被直观调试和干预,目前还是开放问题。端到端带来的可解释性下降,是这条路线需要付出的代价。

项目已在GitHub开源(sh-lee-prml/WaveGen),代码与checkpoint均已放出,值得关注TTS架构演进的研究者和工程师跟进复现。

#WaveGen#端到端语音合成#文本到波形#Spectral Forcing#扩散模型TTS#TTS架构#语音合成
分享:

相关推荐