📄 DriftTTS:不用预训练、一次解码就合成语音的单阶段方法

英文题目:DriftTTS: One-stage Single Step Text-to-Speech via Generative Drifting

评分:6.0/10 | 创新 1.2/2 | 技术严谨 1/1.5 | 实验充分 1/1.5 | 清晰度 0.8/1 | 影响力 0.8/1.5 | 开源 0/1.5 | 可复现 0.3/0.5 | 工程/实践 0.9/1.5

👥 作者与机构

  • Jaehyun Ahn:机构信息未能在本次解析中与作者逐人可靠绑定
  • Dohyun Kim:机构信息未能在本次解析中与作者逐人可靠绑定
  • Heeseung Kim:机构信息未能在本次解析中与作者逐人可靠绑定

📌 核心摘要

文本到语音需要把文字变成时长可变的语音,基于对数密度对输入的梯度(score)或流匹配的方法要多次调用解码器才能保证自然度,推理慢,而已有少步方法依赖预训练的文本到语音教师模型或额外阶段。DriftTTS 把生成式漂移引入语音合成,直接学习从先验到数据分布的推送映射,训练时让生成样本向漂移场指示的目标移动,推理时一次解码得到梅尔语谱图,再用声码器变成波形。为处理变长语音,论文提出片段漂移,在训练中截取固定长度片段并以编码器输出为先验,使每个生成片段靠近其对应目标片段。在 LJSpeech 评测集上,单步 DriftTTS 的词错率为 1.80,主观平均意见分达到 4.43,与 10 步 Matcha-TTS 的 1.93 和 4.38 接近,推理速度约快 4.2 倍。结论适用于单说话人英语、固定声码器和单步解码条件,论文承认多说话人和零样本扩展以及对特征提取器的依赖尚未解决。训练在 4 张 NVIDIA A10 上进行 2000 轮,推理用单卡测量不含声码器的实时因子。

🔗 开源与复现资源

可达状态仅表示本次链接检查结果,不代表许可证、本文权重或运行复现已验证。

🧭 深度解读

文本转语音要解决什么输入输出问题

文本转语音的输入是一串字符或音素,输出是人耳听起来自然、发音正确的波形。中间论文用梅尔谱作为桥梁,梅尔谱是随时间变化的时频能量表示,横轴是帧,纵轴是频率通道,数值高低反映该时刻该频带的能量。合成系统先由声学模型根据文本预测梅尔谱,再由声码器把梅尔谱变成波形,本文统一用 HiFi-GAN 声码器,以便只比较声学模型本身的差异。

必须保留的信息有两类,一是说什么,即音素顺序和时长,二是怎么说,即音色、语调和频谱细节。如果只保证平均频谱接近,语音可能发音含糊或机械感重,所以既要测发音正确性,也要让人试听自然度。输出是变长的,因为不同句子长度不同,同一句话不同语速帧数也不同,这给直接套用固定尺寸图像生成方法带来困难。

从多步采样到少步加速走了哪条路

基于分数匹配和流匹配的语音模型把从先验到数据的传输拆成很多小步,每步去一点噪声或沿向量场走一小段,训练稳定、音质好,但推理时要反复调用解码器,常用十步以上,速度受限。图像领域出现了学习平均速度的 MeanFlow 和约束轨迹走直线的 Consistency Flow Matching,把步数压到很少。

语音领域跟进了这两条路线。IntMeanFlow 把 MeanFlow 用到零样本语音上,沿预训练流匹配模型产生的轨迹学习平均速度,一步可比多步。RapFlow-TTS 把一致性流匹配接到预训练好的 Matcha-TTS 上继续训练,让轨迹上不同起点指向同一数据样本,两步就能达到原来十步的水平。两者的共同代价是都依赖一个已经训好的语音模型来提供轨迹,训练分成预训练加蒸馏或微调两个阶段,复杂度和计算量都增加。本文要回答的是能否跳过预训练语音模型,1 阶段就训出一步可用的模型。

为什么单步且无预训练语音模型是难点

如果解码器只调用一次就必须输出完整梅尔谱,它要一次完成从粗糙先验到精细谱结构的全部传输,映射跨度大,直接回归容易过平滑或丢失细节。而蒸馏类方法之所以可行,是因为老师已经给出了一条从噪声到数据的合理轨迹,学生只需模仿或拉直它。去掉老师后,学生既要发现数据分布在哪里,又要学会一步到达,优化难度上升。

另一个难点是变长。生成式漂移原本为固定形状图像设计,假设每次采样形状相同、类别来自固定集合。语音长度随文本变化,文本条件也不是固定类别编号,不能直接把整句拿来比较核相似度。若把整句补齐到同一长度再算距离,对齐错位会污染相似度。若逐帧独立比较,又会丢失帧间顺序,出现分布对了但排列错了的情况,这正是后文帧漂移实验要展示的失败模式。

DriftTTS 如何组织一次解码的训练流程

DriftTTS 沿用 Matcha-TTS 的文本编码器、时长预测器和解码器,但解码器不再预测随时间积分的速度场,而是直接预测目标梅尔谱段。因为生成式漂移不需要时间步输入,论文去掉了原解码器中的时间步嵌入,参数量有所减少。训练时文本编码器处理整句,解码器只处理截出的固定长度段,以解决变长问题。

下图是理解全文的关键,先看从文本到先验再到生成谱的主路径,再看生成谱与目标谱如何进入同一特征提取器做比较。读图时沿着输入到输出的箭头走一遍,确认先验、生成样本和目标样本分别从哪里来,又在哪里汇合计算损失。

看图路径: 1. 从底部文本经文本编码器到时长扩展,先找到先验的形成路径;2. 看解码器一次映射到生成谱,再与目标谱一起进入掩码自编码器的两条箭头;3. 对照右侧目标与生成分布轮廓的重合程度理解训练目标

原论文 Figure 1:Overview of DriftTTS during training.

论文图 1。原论文 Figure 1::“Overview of DriftTTS during training.”。

图中底部输入文本先经过文本编码器得到 token 级表示,对齐器给出每个 token 的时长,例如 hello 的各音素被扩展为不同帧数,堆叠成与目标等长的先验。解码器把该先验连同噪声条件一次映射为生成谱,生成谱与对应目标谱一起送入掩码自编码器,右侧等高线示意两者分布在特征空间逐渐贴近。推理时则根据预测时长扩展编码器输出,一次解码得到整句梅尔谱,再经声码器成波。

漂移场如何给出吸引与排斥的移动方向

生成式漂移把模型看作从先验分布到数据分布的推送映射,记生成样本为解码器输出,其分布随训练变化。相邻两次训练迭代间生成样本的改变量称为漂移,漂移为零意味着到达更新的不动点。为了得到可计算的更新方向,方法为每个生成样本构造一个漂移场向量,由两项组成,一项把它拉向附近真实样本,一项把它推离附近其他生成样本,前者保证覆盖数据,后者防止所有样本挤到同一众数上。

具体计算用核加权平均。对当前样本,分别与一批真实样本和一批生成样本算相似度,再用相似度加权的位移平均得到吸引向量和排斥向量,相减即得漂移场。相似度用拉普拉斯核,距离越近权重越大,温度控制邻域半径,温度小只看很近的邻居,温度大则看更远的范围。分母是归一化项,保证权重和为一。

\[\mathbf{V}_{p,q}({\mathbf{x}})=\frac{1}{Z_{p}Z_{q}}{\mathbb{E}}_{p,\,q}[k({\mathbf{x}},{\color[rgb]{0,0,1}\mathbf{y}^{+}})k({\mathbf{x}},{\color[rgb]{1,0.5352,0}\mathbf{y}^{-}})({\color[rgb]{0,0,1}\mathbf{y}^{+}}-{\color[rgb]{1,0.5352,0}\mathbf{y}^{-}})],\]

上式把吸引与排斥合并为对正负样本对的双重期望,权重是两个核的乘积,方向是正样本指向负样本的差向量形式,论文在小批量上估计该期望。

\[k_{\tau}({\mathbf{x}},\mathbf{y})=\exp\left(-\frac{1}{\tau}|{\mathbf{x}}-\mathbf{y}|_{2}\right).\]

上式是所用核的定义,指数内是欧氏距离除以温度。实际是在掩码自编码器的特征空间计算该场,而不是直接在原始梅尔信号上计算,以更关注谱结构。

漂移场 × 拉普拉斯核: 漂移场指决定生成样本应向何处移动的向量,由靠近真实样本的吸引减去靠近其他生成样本的排斥构成,拉普拉斯核指按欧氏距离指数衰减的相似度权重,二者搭配使只有特征相近的正负样本才显著影响当前样本的移动方向,温度参数控制这种邻域的大小。

文本先验和对齐如何提供条件信息

原生成式漂移以标准正态噪声为起点,靠类别条件区分不同生成目标。语音没有固定类别,论文把起点重新定义为编码器输出,也就是按时长重复后的文本表示。这样每个先验本身就携带了该说什么的信息,且经单调对齐搜索后已在欧氏距离上接近其目标谱,吸引自然集中在正确目标附近,起到了隐式条件的作用。解码器还额外接收一个噪声样本作为随机条件,以保留一定的多样性。

这种先验带来一个约束。正样本不能随意取,因为每个生成样本只有一个正确去处,被拉向别的句子的谱段会给出错误信号,即使整体分布已经匹配。因此论文把同批的目标段整体作为正样本集合,确保其中包含每个生成样本的配对目标。负样本则用同批的生成段,且在计算排斥时屏蔽对角线,避免样本被自己排斥。

先验样本 × 单调对齐搜索: 先验样本指解码器输入的初始表示,单调对齐搜索指在训练时把文本编码器输出与目标梅尔谱按单调顺序对齐以估计每音素时长的方法,二者搭配的意义是把每个文本 token 按时长重复后得到与目标等长的先验,使生成样本从一开始就落在对应目标附近,从而为漂移提供有条件的吸引方向。

为什么用段做单位并借助掩码自编码器特征

为处理变长,训练时从先验和目标语音中截取固定长度段,论文沿用前人做法取 80 帧约 0.93 秒。文本编码器仍看整句以保证对齐准确,解码器只在段上学习一步映射。这样每次比较的形状固定,核距离有意义,又保留了段内相邻帧的相对位置。段漂移因此把相邻帧绑在一起优化,避免帧独立匹配时的错位。

特征提取器采用在图像上训练掩码重建的掩码自编码器,把梅尔谱看作单通道图像,用 4×4 掩码按 50% 概率遮挡训练。编码器呈瓶颈结构,比较时取对应编码层两两对比。论文假设高维原始梅尔段之间的距离易被粗粒度能量差异主导,导致漂移场只调整体响度而忽略局部谱纹理,而低维特征更关注结构。消融实验验证了这一点,去掉该提取器直接在梅尔空间漂移会出现能量发散。

段漂移 × 帧漂移: 段漂移指把固定长度的连续 80 帧作为一个整体计算特征和漂移,帧漂移指把每一帧独立看待分别计算漂移,二者对比的意义是说明语音帧之间有强时序依赖,只有把相邻帧绑在一起才能保留正确的排列,逐帧匹配即使总体分布对了也可能拼出不连贯的语音。

掩码自编码器 × 多温度损失: 掩码自编码器指先在梅尔谱上做遮挡重建预训练、再取其中间编码层作为特征提取器的网络,多温度损失指在多个核温度和多个编码层上分别计算漂移并取平均的目标,二者搭配是在更关注局部谱结构的低维特征空间里同时兼顾粗细不同的相似尺度,避免直接在高维梅尔空间计算距离被整体能量主导。

训练目标如何把迭代采样搬到训练中

训练目标的思想是让模型输出向其漂移后的位置靠拢。先算出当前生成样本在特征空间的位置,加上漂移场得到漂移后目标,再让模型输出的特征去拟合该目标的均方误差。数值上该损失等于漂移场模长的平方,但通过停止梯度把漂移后目标当作固定靶,梯度只经过模型输出一侧,不经过构造目标的路径。每次更新都使推送分布向数据分布挪动一步,多次训练迭代代替了推理时的多次采样迭代。

\[{\cal L}={\mathbb{E}}_{{\bm{\epsilon}}\sim p_{{\bm{\epsilon}}}}\left[\left|\phi({\mathbf{x}})-\texttt{sg}(\phi({\mathbf{x}})+\mathbf{V}(\phi({\mathbf{x}})))\right|_{2}^{2}\right],\]

上式中 phi 为特征提取器,x 为解码器输出,sg 为停止梯度,期望对先验噪声取平均。实际按温度与编码层的组合分别计算再取平均,而不是求和,因为求和会增大传给文本编码器的梯度,相对削弱先验损失的影响,不利于训练。

\[{\cal L}_{\textup{drift}}={\mathbb{E}}_{{\bm{\epsilon}},,i}\left[\left|\phi_{i}({\mathbf{x}})-\texttt{sg}(\phi_{i}({\mathbf{x}})+\mathbf{V}_{p,,q}(\phi_{i}({\mathbf{x}})))\right|_{2}^{2}\right].\]

上式是段漂移的多温度多层平均版本,下标 i 代表一种温度与层组合。最终总损失是文本编码器的先验损失、时长预测器的时长损失与解码器的漂移损失之和,各自负责对齐表示、时长和谱细节。

\[{\cal L}={\cal L}_{\textup{prior}}+{\cal L}_{\textup{duration}}+{\cal L}_{\textup{drift}}.\]

该求和意味着三部分联合单阶段训练,不需要先训好一个语音老师模型。推理时只用编码器预测时长并扩展,一次解码即得全句谱。

数据、训练配置和评测条件是什么

实验在 LJSpeech 上进行,沿用 Matcha-TTS 的划分,梅尔谱配置跟随 HiFi-GAN V1,以便声码器一致。掩码自编码器也在同一数据上训练。DriftTTS 训练 2000 轮,用四块 A10,优化器为 AdamW,学习率 1×10-4,权重衰减 0.01,梯度裁剪到 5.0,推理用指数滑动平均权重,衰减 0.999。每卡 64 对文本语音,批量大小在此比往常更重要,因为同批目标充当正样本、同批生成充当负样本,批量越大对漂移场的估计越好。解码器训练段长 80 帧,对齐在 GPU 上用 SuperMAS 实现。温度取 0.02、0.05 和 0.2 的组合。

评测分三方面。发音准确性用 Whisper large-v3-turbo 转写算词错率,数值越低越好。感知自然度做 5 分制平均意见分,听测来自 153 位听众的 290 条回复,分数越高越好。速度用不含声码器的实时率,即模型推理时间除以生成音频时长,在单卡批量为 1 下对 500 条评估语音取平均,并取五次交错运行的中位数,数值越小越快。对照模型都共享 Matcha-TTS 结构或在其上构建,包括 10 步 Matcha-TTS、2 步 RapFlow-TTS 及其 1 步版本、1 步 Matcha-TTS 及其加 MeanFlow 版本,以检验单步 DriftTTS 能否达到多步参考质量。消融则用词错率加 UTMOSv2 和整体 DNSMOS 快速比较,其中后两者为客观预测分,仅作相对参考。

单步 DriftTTS 能否达到多步基线的质量和速度

要回答的核心比较问题是,在相同数据、相同声码器和相同硬件下,单步 DriftTTS 的发音正确性和感知自然度是否与 10 步 Matcha-TTS 和 2 步 RapFlow-TTS 相当,速度又快多少。公平条件是各系统都不含声码器计时,听测在同一评估集上进行。词错率越低越好,平均意见分越高越好,实时率越小越快。下表整理原文主比较,包含真值、声码器重建真值、各步数基线与本方法。

ModelNFEWERMOSRTF
Ground truth–1.924.53–
Vocoded GT–1.994.48–
Matcha-TTS101.934.3818.7
RapFlow-TTS21.824.425.85
Matcha-TTS11.864.094.15
+ MeanFlow11.834.394.19
RapFlow-TTS11.794.384.19
DriftTTS11.804.434.41

表中 DriftTTS 在一次函数求值下词错率与平均意见分与 10 步 Matcha-TTS 和 2 步 RapFlow-TTS 处于相近水平,而直接把 Matcha-TTS 压到 1 步时自然度明显掉落。加 MeanFlow 或 RapFlow 的 1 步版本能恢复大部分自然度,但它们都需要先训好 Matcha-TTS 再继续训练。速度上论文报告 DriftTTS 比 10 步快约 4.2 倍、比 2 步快约 1.3 倍,但比同为单步的基线略慢,作者归因于解码器实现而非训练目标本身。该结论只在 LJSpeech 单说话人、所用划分与听众组下得到,换数据集、换声码器或换硬件,相对差距会变化。

函数求值次数 × 实时率: 函数求值次数指合成一句语音时调用解码器网络的次数,实时率指模型推理耗时除以生成音频时长,二者搭配的意义是把计算代价与感知速度联系起来,函数求值次数决定了主要计算量,实时率则是在同一硬件和批量大小下可比的速度体现,数值越小都表示越快。

训练过程的分布变化支持了什么判断

除了表格数字,论文用降维可视化展示训练过程中生成分布向目标分布的靠拢。每幅图把评估集中每条语音的固定 80 帧对数梅尔段展平,再用只在目标上拟合的一套主成分基投影,生成与目标用不同颜色画等高线,因此可比的是分布轮廓的重合度,而不是某条曲线的走向。从早期到后期的面板演变是判断漂移是否持续起作用的依据。

看图路径: 1. 按训练轮次从左到右比较生成等高线与目标等高线的贴合变化;2. 注意每幅图都是固定 80 帧展平后用同一主成分基投影的结果;3. 区分颜色代表的生成与目标,不要把轮廓重叠误读为单样本轨迹

原论文 Figure 2:Evolution of the generated distribution toward the target distribution during training.

论文图 2。原论文 Figure 2::“Evolution of the generated distribution toward the target distribution during training.”。

图中从第 1 轮到第 15 轮再到第 100 轮,生成轮廓从偏离逐渐套住目标轮廓,说明在特征空间的吸引与排斥确实把整体分布推向数据。但这只是分布层面的证据,不能证明每句话的时序细节都正确,还需结合听测和下文帧级别消融来确认排列是否正确。单一一套主成分也会压缩信息,投影重合不等于高维完全一致。

特征提取器、漂移单元和温度各自起什么作用

消融要回答 3 个问题。去掉掩码自编码器直接在梅尔空间做段漂移会怎样,把段拆成帧独立漂移会怎样,只用单一温度与多温度平均有何差别。比较时以默认多温度段漂移为基线,指标仍看词错率、UTMOSv2 和整体 DNSMOS,方向与前文一致。下表整理原文消融,基线之后分别是去掩码自编码器、帧粒度的无掩码自编码器变体和 3 个单温度设置。

TypeWERUTMOSv2DNSMOS
Baseline1.803.9313.263
w/o MAE20.030.5171.394
frame, w/o MAE2.722.1371.724
single temp 0.022.003.8293.149
single temp 0.052.003.9823.244
single temp 0.21.903.9363.273

结果显示特征提取器和漂移单元影响很大。去掉掩码自编码器后各项指标崩溃,论文描述为能量发散,高能量梅尔 bin 被推向 0 的上界,低能量 bin 持续走低,作者推测高维原始段距离被粗能量差异主导,漂移场只调响度不调结构,且低维帧级变体未出现该发散,佐证不稳定来自高维核计算,但严格分析留待未来工作。帧漂移保留了尚可的词错率,但自然度预测分明显下降,符合错误收敛的解释,即先验保留了语言内容所以能读对字,但帧可能匹配到错误的目标帧。单温度与多温度在预测分上接近,没有一致最优,论文沿用前人组合并因其词错率最低而采用。

帧漂移的错误收敛在谱图上直观可见,下图对比同一验证句的目标与帧模型预测,两面板具有相同帧数。观察时上下对比谐波纹理的连续性,重点看竖向断裂和帧间不自然衔接,而非整体颜色深浅。

看图路径: 1. 上下对比同一验证句的目标谱与帧漂移预测谱的纹理连续性;2. 观察下图竖条状断裂和谐波不衔接的位置;3. 结合发音内容仍可辨认来理解可懂度尚可但自然度下降的现象

原论文 Figure 3:False convergence of frame drifting.

论文图 3。原论文 Figure 3::“False convergence of frame drifting. The target is the mel-spectrogram of a validation utterance, and the frame-wise model predicts it using the alignment found by MAS, so the…”。

上为目标谱,下为帧漂移预测谱,后者大体跟随目标结构但许多帧与邻居衔接不自然,出现细碎竖条,这正是段漂移要通过绑定邻帧来避免的问题。该图也提醒分布匹配为零漂移不等于语音正确,排列信息必须显式保留。

结论在什么边界内成立还有哪些未验证

当前证据只支持单说话人英语朗读场景。LJSpeech 数据单一,多说话人、零样本克隆、噪声或跨语言泛化都未评测,论文结论部分也把多说话人和零样本扩展列为未来工作。对掩码自编码器的依赖尚未摆脱,仍需先训好该特征提取器,所谓无预训练仅指无需预训练语音模型,不能理解为零预训练。

评测本身也有边界。主观分来自特定听众组和 290 次回复,换一组听众或不同文本,绝对分会浮动。客观分 UTMOSv2 和 DNSMOS 只是预测分,不能替代人工听感,消融用它们只能看相对趋势。速度结论限定在单卡 A10、批量为 1、不含声码器的条件下,端到端延迟还需计入声码器和文本前端。单步略慢于同类单步基线的现象只归因于实现,未做剖析,换实现可能变化。能量发散的解释仍是假设,没有严格证明。

复现第一步做什么需要哪些关键设置

复现应先准备与原文一致的数据划分和 HiFi-GAN V1 梅尔配置,并训好把梅尔谱当单通道图像的掩码自编码器,掩码 4×4、遮挡概率 50%,取编码器多层特征。声学模型部分按 Matcha-TTS 结构搭建文本编码器、时长预测器和去掉时间步嵌入的解码器,对齐用 GPU 上的 SuperMAS。训练用整句训编码器、用 80 帧段训解码器,同批目标作正样本、同批生成作负样本并屏蔽自交互,批量取显存允许的最大值,原文每卡 64 对。损失按先验加时长加多温度多层漂移平均组织,温度取 0.02、0.05 和 0.2 的组合,优化器、学习率、梯度裁剪和滑动平均按原文设置训足轮次。

验证时先看分布是否靠拢,再听长句连贯性,最后才对比词错率与主观分。作者演示页提供了合成示例,可用于定性对照,但第三方页面只能证明论文提到了该链接,不能据此断言代码或权重已开源,复现仍应以论文描述和自训模型为准。

什么时候值得尝试以及还缺哪项验证

当已有 Matcha-TTS 式结构、希望一步合成且不愿维护预训练老师模型的 2 阶段流程时,DriftTTS 的段漂移值得尝试。它的价值在于用训练时的分布漂移替代推理时的多步积分,以一次解码达到接近多步的自然度,代价是需要较大的批量来估计正负期望、依赖掩码自编码器特征,以及对高维直接漂移敏感。

初学者复述关键步骤可记为文本编码加对齐得到等长先验,解码器一次映射为谱段,在掩码自编码器多层多温度特征上算吸引减排斥的漂移场,用停止梯度的拟合损失逐步把分布推向数据。还缺的验证包括多说话人下的条件充分性、端到端含声码器的延迟对比,以及能量发散机制的理论分析。在这些补齐之前,不宜把单数据集的相近分数夸大为普遍的一步最优。

📎 论文与评分元数据

排名:前50% | 文档类型:方法研究 | arXiv 原文

⚖️ 评分明细

评分属于系统判断,不是论文实验结果;八维数值与总分见页首,原始审计记录保留在后端。

  • 评分规则:type-aware-v1

  • 评分模型:muse-spark-1.3-contributor

  • 评分请求协议:openai_responses


← 返回 2026-10-19 语音/音乐/音频论文速递