Hubert: How Much Can a Bad Teacher Benefit ASR Pre-Training?

By Wei-Ning Hsu, Yao-Hung Hubert Tsai, Benjamin Bolte, Ruslan Salakhutdinov, Abdel-rahman Mohamed. 2021

184 citations recorded by the source corpus.

Compared to vision and language applications, self-supervised pre-training approaches for ASR are challenged by three unique problems: (1) There are multiple sound units in each input utterance, (2) With audio-only pre-training, there is no lexicon of sound units, and (3) Sound units have variable lengths with no explicit segmentation. In this paper, we propose the Hidden-Unit BERT (HUBERT) model which utilizes a cheap k-means clustering step to provide aligned target labels for pre-training of a BERT model. A key ingredient of our approach is applying the predictive loss over the masked regions only. This allows the pre-training stage to benefit from the consistency of the unsupervised teacher rather that its intrinsic quality. Starting with a simple k-means teacher of 100 cluster, and using two iterations of clustering, the HUBERT model matches the state-of-the-art wav2vec 2.0 performance on the ultra low-resource Libri-light 10h, 1h, 10min supervised subsets.

Subfields: Artificial Intelligence, Representation & Self-Supervised Learning, Speech & Audio, Signal Processing

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Artificial Intelligence

  1. Wei-Ning Hsu
  2. James R. Glass
  3. Abdel-rahman Mohamed
  4. Michael Auli
  5. Alexei Baevski
  6. Yossi Adi
  7. Hung-yi Lee
  8. Jinyu Li

In Representation & Self-Supervised Learning

  1. Wei-Ning Hsu
  2. James R. Glass
  3. Alexei Baevski
  4. Abdel-rahman Mohamed
  5. Yu-An Chung
  6. David F. Harwath
  7. Michael Auli
  8. Hung-yi Lee

In Speech & Audio

  1. Wei-Ning Hsu
  2. James R. Glass
  3. Abdel-rahman Mohamed
  4. Yossi Adi
  5. Alexei Baevski
  6. Jinyu Li
  7. Yu-An Chung
  8. Hung-yi Lee

In Signal Processing

  1. Wei-Ning Hsu
  2. James R. Glass
  3. Abdel-rahman Mohamed
  4. Hung-yi Lee
  5. Yu-An Chung
  6. Yossi Adi
  7. Jinyu Li
  8. David F. Harwath

Citations in the corpus