HuBERT:通过隐藏单元的掩码预测进行自监督语音表示学习

HuBERT: Self-Supervised Speech Representation Learning by Masked Prediction of Hidden Units

阿卜杜勒拉赫曼·穆罕默德 Abdelrahman Mohamed · Meta AI · 2021-06-14 · arXiv:2106.07447 ↗ · 被引 4746

打开互动全文版(逐段中英对照 + 图/公式 + 论文问答)→

摘要 · Abstract

自监督语音表示学习方法面临三个独特挑战:(1) 每个输入话语包含多个声音单元,(2) 预训练阶段没有输入声音单元的词典,(3) 声音单元长度可变且没有明确分割。为了解决这些问题,我们提出了隐藏单元 BERT(HuBERT)方法,该方法利用离线聚类步骤为类似 BERT 的预测损失提供对齐的目标标签。我们方法的一个关键要素是仅在掩码区域应用预测损失,这迫使模型在连续输入上学习联合声学和语言模型。HuBERT 主要依赖于无监督聚类步骤的一致性,而不是分配聚类标签的内在质量。从简单的 100 个聚类的 k-means 教师模型开始,经过两次聚类迭代,HuBERT 模型在 Librispeech(960 小时)和 Libri-light(60,000 小时)基准测试中,在 10 分钟、1 小时、10 小时、100 小时和 960 小时微调子集上,匹配或改进了最先进的 wav2vec 2.0 性能。使用 1B 参数模型,HuBERT 在更具挑战性的 dev-other 和 test-other 评估子集上,相对词错误率分别降低了高达 19%和 13%。

Self-supervised approaches for speech representation learning are challenged by three unique problems: (1) there are multiple sound units in each input utterance, (2) there is no lexicon of input sound units during the pre-training phase, and (3) sound units have variable lengths with no explicit segmentation. To deal with these three problems, we propose the Hidden-Unit BERT (HuBERT) approach for self-supervised speech representation learning, which utilizes an offline clustering step to provide aligned target labels for a BERT-like prediction loss. A key ingredient of our approach is applying the prediction loss over the masked regions only, which forces the model to learn a combined acoustic and language model over the continuous inputs. HuBERT relies primarily on the consistency of the unsupervised clustering step rather than the intrinsic quality of the assigned cluster labels. Starting with a simple k-means teacher of 100 clusters, and using two iterations of clustering, the HuBERT model either matches or improves upon the state-of-the-art wav2vec 2.0 performance on the Librispeech (960h) and Libri-light (60,000h) benchmarks with 10min, 1h, 10h, 100h, and 960h fine-tuning subsets. Using a 1B parameter model, HuBERT shows up to 19% and 13% relative WER reduction on the more challenging dev-other and test-other evaluation subsets.

核心贡献 · Key contributions

局限 · Limitations

论文章节 · Sections(共 19)

阅读逐段中英对照全文 →