论文阅读_音频生成_AudioLM

论文信息

name_en: AudioLM: a Language Modeling Approach to Audio Generation
name_ch: AudioLM：一种音频生成的语言建模方法
paper_addr: http://arxiv.org/abs/2209.03143
doi: https://doi.org/10.48550/arXiv.2209.03143
date_read: 2023-04-25
date_publish: 2022-09-07
tags: [‘语音合成’,‘深度学习’]
author: Zalán Borsos
citation: 36
demo：https://google-research.github.io/seanet/audiolm/examples

1 读后感

主要解决生成语音的两个问题：一致性和高质量。

2 摘要

这是一个利用长期一致性生成高质量音频的框架，它先将音频输入转成一系列离散的token，然后将生成音频作为表示空间的语言建模。提出了一种混合的分词方案来平衡重建质量和长依赖的结构。

使用Mask方法捕获长距离的关系，最终使用离散编码生成高品质的合成效果。它可以通过简短的提示，来生成自然连贯延续语音。利用大量无监督数据训练，在没有任何文字标注或注释的情况下，AudioLM 会生成句法和语义上合理的语音延续，同时还保持说话人身份和不可见的说话人的韵律。另外，还可以生成钢琴音乐。

3 介绍

在数据都是无监督的情况下，基于Transformer架构。具体使用的技术包括：对抗性神经音频压缩，自监督表示学习，语言建模。学习不同尺度的相互作用，保证语音的一致性。

贡献

• 提出AudioLM框架，分层方式结合语义和声学标记，以实现生成长期一致性和高质量的音频。
• 通过与w2v-BERT以及SoundStream的对比，证明了模型的可辨别性和重建质量优势的互补性。
• 模型可以不依赖文本标注，生成语音，句法和语义。只需要3s语音作为提示，即可生成训练期间未见过的语音，并保持说话人的声音，韵律，录音条件（混响、噪音）。
• 除合成人声外，还可以合成音乐声，其旋律、和声、音调和节奏都与提示一致。
• 为防御生成语音带来的潜在风险，还提出了一个分类器，用于识别合成音频和真实音频。

4 模型

声学token由 SoundStream处理，语义token由 w2v-BERT 的中间层产生。

4.1 组件

• 将输入音频x映射到离散的词表y：y=end(x)。
• 使用仅有decoder的Transformer模型，操作y，用时间t-1的预测t对应的词（预测阶段使用自回归）。
• 解码模型 ，将预测出的y^映射回音频格式。 x^=dec(y)

4.2 权衡离散音频表示

使用尽量少的数据同时需要保证生成的音质，这涉及比特率的下限和序列长度。这里引入了语义token和声学token。如图-1所示。它们的产生被解耦；语义token需要时序依赖，声学token需要保证高音质，且使用语义作为条件。

使用 SoundStream 计算声学token，它使用了RQV（残差向量量化）技术将嵌入降维和离散化，并映射到码表。

使用 w2v-BERT 计算语义标记。该模型可以自主学习音频表示，将输入的音频波形映射到一个富有语言特征的向量空间。通过使用两个自监督目标：掩码语言建模（MLM）损失和对比损失训练模型实现。选择w2v-BERT模型的MLM模块中的一个中间层并计算该层的嵌入，可以提取出语义标记。将这些标记进行聚类，并使用聚类中心索引作为语义标记。

实验证明，将二项解耦效果更好。

4.3 语义和声学标记的分层建模

先使用模型产生语义，然后再语义条件下生成高质量音频，有两个好处：

• 语义结果独立于音频结果。
• 减少了每个阶段的标记序列，训练和推理效率更高。

具体实现如图-2所示，包含三个场景：

• 长期结构一致性的语义建模：利用上文，使用自回归方法预测语义z。
• 以语义标记为条件的粗略声学建模：利用上文和语义，预测粗糙声的声学标记y。
• 精细声学建模：用粗糙声学标记y以及上文生成精细声学信息，生成高质量标记。
  SoundStream 嵌入的采样率是 w2v-BERT 嵌入的两倍。另外拆分两的场景的原因是可以限制序列长度。

4.4 预测

训练后，可以使用 AudioLM 生成音频，测试了以下三种情况：

4.4.1 无条件生成

无条件地对所有语义标记 ^z 进行采样，然后将其用作声学建模的条件。此实验证明了：模型可生成多种多样、句法和语义一致的语言内容，验证了语义与声学的无关性。

4.4.2 声学生成

使用从测试集 x 中提取的真实语义标记 z 作为条件来生成声学标记。生成的音频序列在说话人身份方面有所不同，但语义内容与 x 的真实内容匹配。这表明语义标记捕获了语义内容。

4.4.3 生成语音延续

从短提示 x 生成延续。首先将提示映射到相应的语义标记 z 和粗糙的声学标记 y。第一阶段生成语义标记的延续；第二阶段，将生成的语义与提示粗声学标记y连接起来，并将其作为条件提供给粗声学模型；在第三阶段，用精细的声学模型处理粗略的声学标记；最后，将提示和采样的声学标记都提供给 SoundStream 解码器以重建波形 x^。