人工智能

StyleTTS 2:具有大型语音语言模型的人级文本转语音合成

mm

由于自然和合成语音合成方法的增加,人工智能行业在过去几年中取得的一项重大成就是有效地合成具有潜在应用的文本转语音框架各种行业,包括有声书、虚拟助手、配音等,一些最先进的模型在广泛的语音相关任务中提供了人级的性能和效率。然而,尽管它们具有强大的性能,但仍有改进的余地,例如表达和多样化的语音、优化零拍文本转语音框架所需的大量训练数据以及对OOD或分布外文本的鲁棒性,这些因素促使开发人员致力于创建一个更强大和易用的文本转语音框架。

在本文中,我们将讨论StyleTTS-2,这是一个强大和创新的文本转语音框架,建立在StyleTTS框架的基础上,旨在推动最先进的文本转语音系统的发展。StyleTTS2框架将语音风格建模为潜在随机变量,并使用概率扩散模型来采样这些语音风格或随机变量,从而允许StyleTTS2框架在不使用参考音频输入的情况下有效地合成真实的语音。由于这种方法,StyleTTS2框架能够比当前最先进的文本转语音框架提供更好的结果,并且在广泛的语音任务中显示出高效率,同时也能够利用扩散模型框架提供的多样化语音合成。我们将更详细地讨论StyleTTS2框架,并探讨其架构和方法,同时也将查看框架的结果。让我们开始吧。

StyleTTS2用于文本转语音合成:介绍

StyleTTS2是一个创新性的文本转语音合成模型,它是基于StyleTTS的,StyleTTS是一个基于风格的文本转语音生成模型。StyleTTS2框架将语音风格建模为潜在随机变量,并使用概率扩散模型来采样这些语音风格或随机变量,从而允许StyleTTS2框架在不使用参考音频输入的情况下有效地合成真实的语音。将风格建模为潜在随机变量是StyleTTS2框架与其前身StyleTTS框架的区别之处,StyleTTS2框架旨在为输入文本生成最合适的语音风格,而无需参考音频输入,并且能够在利用扩散模型提供的多样化语音合成能力的同时实现有效的潜在扩散。此外,StyleTTS2框架还采用了预训练的大型语音语言模型(SLM)作为判别器,例如WavLM框架,并将其与自己的新型可微分持续时间建模方法相结合,以便对框架进行端到端训练,最后生成具有增强自然性的语音。由于其采用的方法,StyleTTS2框架在语音生成任务中超越了当前最先进的框架,并且是预训练大规模语音模型的最有效框架之一,在零拍设置中适用于扬声器适应任务。

继续讨论,为了实现人级文本转语音合成,StyleTTS2框架结合了现有工作的成果,包括用于语音合成的扩散模型和大型语音语言模型。扩散模型通常用于语音合成任务,因为它们具有细粒度的语音控制和多样化的语音采样能力。然而,扩散模型并不像基于GAN的非迭代框架那样高效,主要原因是需要迭代地采样潜在表示、波形和梅尔谱图以达到目标语音持续时间。

另一方面,最近关于大型语音语言模型的工作表明,它们能够增强文本转语音生成任务的质量,并且能够很好地适应扬声器。大型语音语言模型通常将文本输入转换为来自预训练语音语言框架的量化或连续表示,以用于语音重构任务。然而,这些语音语言模型的特征并不是直接针对语音合成进行优化的。相比之下,StyleTTS2框架利用大型SLM框架的知识,通过对抗训练来合成语音语言模型的特征,而无需使用潜在空间映射,因此能够直接学习语音合成优化的潜在空间。

StyleTTS2:架构和方法

在其核心,StyleTTS2建立在其前身StyleTTS框架的基础上,StyleTTS是一个非自回归的文本转语音框架,它使用风格编码器从参考音频中提取风格向量,从而允许表达和自然的语音生成。StyleTTS框架中使用的风格向量直接融入编码器、持续时间和预测器中,使用AdaIN或自适应实例归一化,从而允许StyleTTS模型生成具有不同语调、持续时间和情感的语音输出。StyleTTS框架由8个模型组成,分为三个类别

  1. 声学模型或语音生成系统,包括风格编码器、文本编码器和语音解码器。
  2. 文本转语音预测系统,使用语调和持续时间预测器。
  3. 实用系统,包括文本对齐器、音高提取器和用于训练的判别器。

由于其方法,StyleTTS框架在可控和多样化的语音合成方面提供了最先进的性能。然而,这种性能也有一些缺点,例如样本质量的下降、表达的局限性和对语音的干扰应用在实时系统中。

改进StyleTTS框架,StyleTTS2模型实现了增强的表达和多样化的文本转语音任务,具有改进的分布外性能和高人级质量。StyleTTS2框架使用端到端训练过程来优化不同组件,使用对抗训练和直接波形合成。与StyleTTS框架不同,StyleTTS2框架将语音风格建模为潜在变量,并通过扩散模型采样,从而在不使用参考音频的情况下生成多样化的语音样本。让我们更详细地探讨这些组件。

端到端训练用于干扰

在StyleTTS2框架中,使用端到端训练方法来优化各种文本转语音组件,用于干扰,而无需依赖固定组件。StyleTTS2框架通过修改解码器来实现这一点,直接从风格向量、音高和能量曲线以及对齐表示中生成波形。然后,框架删除解码器的最后一个投影层,并用波形解码器替换它。StyleTTS2框架使用两个编码器:HifiGAN基于的解码器直接生成波形,和iSTFT基于的解码器产生相位和幅度,然后转换为波形,以实现更快的干扰和训练。

上图表示用于预训练和联合训练的声学模型。为了减少训练时间,模块首先在预训练阶段优化,然后在联合训练中优化所有组件,除了音高提取器。联合训练不优化音高提取器的原因是它用于提供音高曲线的基准事实。

上图表示语音语言模型的对抗训练和干扰,使用预训练但未预调的WavLM框架。该过程与上述过程不同,因为它可以接受不同的输入文本,但会累积梯度来更新每个批次的参数。

风格扩散

StyleTTS2框架旨在通过潜在变量来建模语音,条件分布遵循该变量,这个变量被称为广义语音风格,代表语音样本中超出语音内容范围的任何特征,包括词汇重音、语调、语速和甚至形式转换。

语音语言模型判别器

语音语言模型以其能够编码广泛语义和音频方面的有价值信息而闻名,SLM表示传统上能够模仿人类对生成语音质量的感知。StyleTTS2框架使用对抗训练方法来利用SLM编码器执行生成任务的能力,并采用12层WavLM框架作为判别器。这种方法允许框架在OOD或分布外文本上进行训练,从而有助于提高性能。另外,为了防止过拟合问题,框架以相同的概率对OOD文本和分布内文本进行采样。

可微分持续时间建模

传统上,文本转语音框架使用持续时间预测器来生成音素持续时间,但这些持续时间预测器使用的上采样方法通常会在端到端训练过程中阻塞梯度流。NaturalSpeech框架使用基于注意力的上采样器来实现人级文本转语音转换。然而,StyleTTS2框架发现这种方法在对抗训练中不稳定,因为StyleTTS2使用可微分上采样和不同的对抗训练,而无需额外的损失项。虽然使用软动态时间扭曲方法可以帮助缓解这种不匹配,但使用它不仅计算代价高昂,而且在处理对抗目标或梅尔重构任务时,其稳定性也是一个问题。因此,为了实现人级性能并稳定训练过程,StyleTTC2框架使用非参数上采样方法。高斯上采样是一种流行的非参数上采样方法,用于将预测的持续时间转换为波形,尽管它具有固定的高斯核长度的限制,这限制了其准确建模不同长度的对齐的能力。

为了克服这一限制,StyleTTC2框架提出了一种新的非参数上采样方法,无需额外的训练,就能考虑对齐的不同长度。对于每个音素,StyleTTC2框架将对齐建模为随机变量,指示与音素对齐的语音帧的索引。

模型训练和评估

StyleTTC2框架在三个数据集上进行训练和实验:VCTK、LibriTTS和LJSpeech。StyleTTS2框架的单扬声器组件使用LJSpeech数据集进行训练,LJSpeech数据集包含大约13,000+个音频样本,分为12,500个训练样本、100个验证样本和500个测试样本,总运行时间约为24小时。StyleTTS2框架的多扬声器组件在VCTK数据集上进行训练,VCTK数据集包含超过44,000个音频片段,来自100多个不同口音的本土扬声器,分为43,500个训练样本、100个验证样本和500个测试样本。最后,为了使框架具有零拍适应能力,框架在LibriTTS数据集上进行训练,LibriTTS数据集包含大约250小时的音频,来自1,150多个扬声器。为了评估其性能,模型使用两个指标:MOS-N(自然度平均意见评分)和MOS-S(相似度平均意见评分)。

结果

StyleTTS2框架采用的方法和方法论在其性能中得到了体现,模型在NaturalSpeech数据集上超越了几种最先进的TTS框架,并在数据集上设定了新的标准。此外,StyleTTS2框架在VCTK数据集上超越了最先进的VITS框架,结果如下图所示。

StyleTTS2模型在LJSpeech数据集上也超越了之前的模型,并且在OOD或分布外文本上没有显示出质量下降,如同先前的框架在相同的指标上所示。此外,在零拍设置中,StyleTTC2模型在自然度方面超越了现有的Vall-E框架,尽管它在相似度方面落后。但是,值得注意的是,StyleTTS2框架能够实现具有竞争力的性能,尽管它只在245小时的音频样本上进行了训练,而Vall-E框架则在60,000多小时的音频样本上进行了训练,从而证明StyleTTC2是一个数据高效的替代方案,相比现有的大规模预训练方法,如Vall-E所使用的方法。

继续讨论,由于缺乏带有情感标签的音频文本数据,StyleTTC2框架使用GPT-4模型生成超过500个实例,跨越不同的情感,以可视化框架使用其扩散过程创建的风格向量。

在第一张图中,情感风格以LJSpeech模型的风格向量为响应,展示了StyleTTC2框架合成具有不同情感的表达性语音的能力。第二张图展示了五个个体扬声器的不同簇,从单个音频文件中源自广泛的多样性。最后一张图展示了扬声器1的情感簇的松散簇,揭示了尽管有一些重叠,但情感簇是突出的,这表明可以操纵扬声器的情感基调,无论参考音频样本及其输入语调如何。尽管使用了基于扩散的方法,StyleTTS2框架仍然超越了现有的最先进框架,包括VITS、ProDiff和FastDiff。

最后的思考

在本文中,我们讨论了StyleTTS2,这是一个新颖、强大和创新的文本转语音框架,建立在StyleTTS框架的基础上,旨在推动最先进的文本转语音系统的发展。StyleTTS2框架将语音风格建模为潜在随机变量,并使用概率扩散模型来采样这些语音风格或随机变量,从而允许StyleTTS2框架在不使用参考音频输入的情况下有效地合成真实的语音。StyleTTS2框架使用风格扩散和SLM判别器来实现人级文本转语音任务,并在广泛的语音任务中超越了现有的最先进框架。

专业为工程师,心为作家。 Kunal是一名技术作家,对AI和ML有着深厚的热爱和理解,致力于通过其引人入胜和信息丰富的文档来简化这些领域中的复杂概念。