一文全面了解火山语音无监督预训练技术的落地实践

一直以来,火山语音团队都为时下风靡的视频平台提供基于语音识别技术的智能视频字幕解决方案,简单来说就是可以自动将视频中的语音和歌词转化成文字,来辅助视频创作的功能。但伴随平台用户的快速增长以及对语言种类更加丰富多样的要求,传统采用有监督学习技术来解决的办法日渐遭遇瓶颈,这让团队着实犯了难。

众所周知,传统的有监督学习会对人工标注的有监督数据产生严重依赖,尤其在大语种的持续优化以及小语种的冷启动方面。以中文普通话和英语这样的大语种为例,尽管视频平台提供了充足的业务场景语音数据,但有监督数据达到一定规模之后,继续标注的 ROI 将非常低,必然需要技术人员考虑如何有效利用百万小时级别的无标注数据,来进一步改善大语种语音识别的效果。

相对小众的语言或者方言,由于资源、人力等原因,数据的标注成本高昂。在标注数据极少的情况下(10 小时量级),有监督训练的效果非常差,甚至可能无法正常收敛;而采购的数据往往和目标场景不匹配,无法满足业务的需要。

为此火山语音团队迫切需要研究如何以尽可能低廉的标注成本充分利用大量的无标注数据,提升少量标注数据下的识别效果并落地到实际业务中,所以无监督预训练技术成为视频平台 ASR(Automatic Speech Recognition / 自动语音识别)能力向小语种推广的关键。

尽管近年来学术界在语音无监督预训练领域取得了许多重大进展,包括 Wav2vec2.0 [1]、HuBERT [2]等,但在工业界却鲜有落地案例可以参考借鉴。整体来看,火山语音团队认为,以下三方面主要原因,阻碍了无监督预训练技术的落地:

1. 模型参数量大、推理开销大。大量无标注数据需要用较大的模型做无监督预训练,才能得到高质量的语音表征,但这样的模型如果直接部署到线上,会带来高昂的推理成本。

2. 无监督预训练只关注语音表征的学习,需要结合大量纯文本训练的语言模型联合解码才能达到理想效果,和端到端 ASR 推理引擎不兼容。

3、无监督预训练开销大、周期长且不稳定。以 Wav2vec2.0 为例,300M 参数量的模型用 64 张 V100 GPU 预训练 60 万步,耗时长达半个月;此外由于数据分布的差异,在业务数据上训练容易发散。

对此火山语音团队在基于无监督预训练的语音识别技术落地过程中,针对以上三大痛点进行了算法改进和工程优化,形成一套完整易推广的落地方案。本文将针对方案,从落地流程、算法优化以及工程优化等环节展开详尽介绍。

落地流程

下图是基于无监督预训练的低资源语种 ASR 的落地流程,大致可以划分为数据收集、种子模型训练和模型迁移三个阶段。

基于无监督预训练的 ASR 落地流程

具体来说,第一阶段的数据收集,可以通过语种分流、采购等手段收集目标语言的无标注语音、标注语音和纯文本数据。

第二阶段的种子模型训练,也就是经典的“无监督预训练+有监督微调”过程。这一阶段将得到一个声学模型,通常基于连接时序分类(Connectionist Temporal Classification, CTC [3])损失函数微调。声学模型结合纯文本训练的语言模型,构成一个完整的语音识别系统,可以取得不错的识别效果。之所以称之为种子模型,是因为这个模型并不适合直接上线到业务,我们更倾向于使用 LAS(Listen, Attend and Spell [4])或 RNN-T(Recurrent Neural Network Transducer [5])这类端到端模型进行线上部署。

归其原因,主要是 LAS/RNN-T 具有出色的端到端建模能力,同时在近年来已经取得了优于传统 CTC 模型的效果,并在工业界得到越来越多的应用。火山语音团队针对端到端语音识别模型的推理和部署做了大量优化工作,已形成一套相对成熟的方案,并支持众多业务。在维持效果无损的前提下,如果可以沿用端到端推理引擎,就能大幅降低引擎的运维成本。

基于此团队设计了第三阶段,即模型迁移阶段。主要借鉴知识蒸馏的思想,用种子模型对无标注数据打伪标签,然后提供一个参数量较小的 LAS 模型做训练,同步实现了模型结构的迁移和推理计算量的压缩。整个流程的有效性在粤语 ASR 上得到验证,具体实验结果如下表所示:

首先,团队采购了 1kh 成品数据用于实验对比,直接训练 LAS 模型性能不佳,字错率(Character Error Rate, CER)高达 44.2%。经过分析,火山语音认为主要原因是采购数据(对话)和业务测试集(视频)领域不匹配,在 wav2vec2.0 上的初步实验也发现了类似的现象。

相比用采购数据做预训练,采用和目标领域一致的数据做预训练,在业务测试集上的 CER 可以从 42.0%下降到 29.4%,于是团队将业务场景的无标注数据积累到 50kh,模型参数量从 100M 增加到 300M,CER 进一步下降到 23.1%。

最后团队验证了模型迁移的效果,结合粤语语言模型对 50kh 无标注数据解码得到伪标签,训练 LAS 模型。可以看到,基于伪标签训练的 LAS 模型基本可以保持 CTC 种子模型的识别效果且模型参数量减少了三分之一,可以直接基于成熟的端到端推理引擎部署上线。

模型参数量和 CER 对比

最终在模型结构和参数量不变的前提下,团队用 50kh 无标注业务数据和 10h 有标注业务数据取得了 23.0%的 CER,相对基线模型下降 48%。解决了线上计算量和兼容性的问题之后,聚焦到整个流程中最为核心的无监督预训练技术,针对 wav2vec2.0,火山语音团队分别从算法和工程两个维度进行了优化。

算法优化

wav2vec2.0 作为 Meta AI 在 2020 年提出来的自监督预训练模型,开启了语音无监督表征学习的新篇章。其核心思想在于用量化模块将输入特征离散化,并通过对比学习优化,模型主体与 BERT 类似,随机 mask 部分输入特征。

wav2vec2.0 模型结构示意图(来源:wav2vec 2.0 Figure 1 [1])

对此在业务数据上训练 wav2vec 2.0 模型遇到了两个棘手的问题:一个是训练效率低,300M 的大模型 64 卡需要十几天才能训完;另一个是训练不稳定,容易发散。为此团队提出 Efficient wav2vec 以缓解上述两个问题:

对于训练效率低的问题,团队通过降低模型的帧率来加快训练速度,将输入特征从 waveform 替换成 filterbanks,帧率由原来的 20ms 变成 40ms。这样既大幅降低了特征提取卷积的计算量,同时也大幅降低了 Transformer 内部编码的长度,从而提高训练效率。对于训练不稳定的问题,则是通过分析无监督预训练的学习方式并结合业务数据实际情况综合判断解决,对比学习损失可以用下式表达:

为了解决上述两个问题,火山语音对应提出了两点改进:

等长数据流:预训练过程中将整个训练集视为由每句话首尾拼接而成的一段音频,每个训练样本从中截取固定长度得到,这样做是为了保证负样本数量足够多,且上下文编码网络内部的长度在不同帧率下一致,进而保证训练的稳健性。

自适应连续 mask:为缓解数据噪音对训练的影响,选取较小的 mask 长度且强制每个 mask 区域连续,并且 mask 区域对应的音频长度在不同帧率下相当。这样既减轻了噪音数据下对比学习的难度,同时也做到了适配不同的帧率。

在业务数据上对比了 wav2vec2.0(w2v)与 Efficient wav2vec (w2v-e)的效果,如下表所示(所有模型均采用 64 V100 GPUs 训练):

可以看到改进过的 Efficient wav2vec 相对原始 wav2vec 2.0 有稳定 5%的性能提升,并且训练效率接近翻倍。

工程优化

尽管团队提出的 Efficient wav2vec 已经从算法层面将训练效率提升近两倍,但由于 300M 模型通信量大,训练通信依然存在波动且多机扩展效率低。对此火山语音团队总结道:“为了提高模型预训练在同步梯度场景下的通信效率,我们基于 BytePS 的分布式训练框架,在通信后端完成了 Bucket 分组通信优化技术,数据并行效率能取得 10%的提升;同时针对模型参数定义顺序与梯度更新顺序不同造成的等待问题,还实现了自适应的参数重排(Parameter Reorder)策略。”在这些优化基础上,进一步结合梯度累加等技术,300M 模型的单卡扩展效率由 55.42%提升至 81.83%,多机扩展效率由 60.54%提升至 91.13%,原来需要 6.5 天训完的模型现在只需要 4 天就可以训完,耗时缩短 40%。

此外,为了支持未来探索的大模型大数据场景,火山语音工程团队进一步完成了一系列超大规模模型的原子能力建设。首先实现了 local OSS 技术,在去除优化器大部分的冗余内存占用的同时,解决了机间扩展效率问题;之后在同步梯度通信上支持了 bucket lazy init,减少了一倍参数量的显存占用,能大幅降低显存峰值并适配显存资源紧张的超大模型场景;最后在数据并行的基础上,还支持了模型并行和流水线并行,并在 1B 和 10B 模型上完成了验证和定制化支持。这一系列优化为大模型大数据的训练打下坚实基础。

目前,通过采用低资源 ASR 落地流程,已有两个低资源语言成功落地视频字幕和内容安全业务。除语音识别外,基于 wav2vec2.0 的预训练模型在其他多个下游任务上也已取得显著收益,涉及音频事件检测、语种识别、情感检测等,未来将陆续落地到视频内容安全、推荐、分析、音频分流、电商客服情感分析等相关业务中。无监督预训练技术的落地将显著降低各类音频数据的标注成本,缩短标注周期,实现对业务需求的快速响应。

总结与展望

火山语音团队在实践中摸索出一套基于 wav2vec2.0 的低资源语种 ASR 落地方案,解决了推理开销大的问题,实现了与端到端引擎的无缝衔接。针对其中最核心的 wav2vec2.0 训练效率低和不稳定的问题,提出了 Efficient wav2vec。相比 wav2vec2.0,在下游任务上效果提升 5%,预训练耗时缩短一半,结合工程上的优化,最终预训练耗时相比原始版本缩短 70%。未来,火山语音团队将在以下三个方向持续挖掘探索:

无监督算法升级:在 wav2vec 2.0 之后语音无监督预训练的研究工作如雨后春笋,团队将持续跟进最新的研究,并内化到业务场景。现阶段主要尝试了 HuBERT[2]、MAE[6] 和 data2vec[7]等无监督模型,并探索了它们各自在不同下游任务下的表现。未来将从两个方面提升无监督模型性能:根据不同业务场景,设计高效适配的无监督方案;设计通用的无监督模型,提升在各类下游任务的性能表现。

多语言多模态:目前无监督与多语言结合的研究工作有许多,比如 XLSR[8]。团队在此基础上提出了 S3Net[9],其通过在预训练模型中划分出多个稀疏子网络来分别对不同语言进行建模,有效缓解了不同语言之间的相互干扰(Language Interference)问题,对大语料语言有明显的性能提升效果。现有的研究工作主要集中在音频编码器端进行,而目前主流的端到端模型均采用了编码器-解码器结构,即音频文本多模态建模。团队判断单纯的音频端预训练已经不能满足端到端模型的需要,未来将在音频文本多模态预训练上进行探索工作,分别是海量非对齐音频文本与端到端模型联合建模以及纯无监督的多模态预训练。

1. 大数据大模型:现有的模型在 10 万小时规模时其性能就接近饱和,团队在中文 10 万小时标注数据训练的模型基础上,利用 100 万小时无标注数据做 NST[10]训练,在通用测试集上取得相对 7%的 CER 下降,同时模型的泛化能力得到明显的改善,在 20 个领域测试集上平均 CER 相对下降 15%。要充分吸收百万小时量级的海量数据就需要更大的模型,目前团队已经在 1B 参数量级的模型上取得初步进展。大模型的性能上限高,随之而来的问题是落地难。为了将大模型落地到实际业务中,未来将尝试各种模型压缩方法,如矩阵分解、权重裁剪和知识蒸馏等,尽可能做到无损压缩效果。

火山语音,长期以来面向字节跳动各大业务线以及火山引擎 ToB 行业与创新场景,提供全球领先的 AI 语音技术能力以及卓越的全栈语音产品解决方案,包括音频理解、音频合成、虚拟数字人、对话交互、音乐检索、智能硬件等。目前团队的语音识别和语音合成覆盖了多种语言和方言,多篇技术论文入选各类 AI 顶级会议,为抖音、剪映、飞书、番茄小说、Pico 等业务提供了领先的语音能力,并适用于短视频、直播、视频创作、办公以及穿戴设备等多样化场景,通过火山引擎开放给外部企业。