pytorch / pytorch/audio

Add training recipes for HuBERT model pre-training and ASR fine-tuning

Open
#1,908 0 comments 9 reactions 1 assignee View on GitHub

@nateanl is already working on this.

Since Oct 20, 2021.

Dominant language
Python
Stars
2.9k
Forks
798
Avg merge
58m
Merged PRs (30d)
3

Description

🚀 The feature

Hidden-Unit BERT (HuBERT), a self-supervised model for speech representations was proposed and wildly used in down-stream tasks, such as speech recognition, speech diarization, speaker identification, etc. It can achieve impressive Word Error Rate by fine-tuning on only 10 minutes of supervised data.

To fine-tune the HuBERT model for customized down-stream task, people need to install and adopt their training pipeline to fairseq. It will be great to add a training recipe to torchaudio that loads the torchaudio's HuBERT model and simply the training process.

Motivation, pitch
  • Add preprocessing scripts (MFCC feature extraction, KMeans model training, pseudo-label prediction).
  • Add a PyTorch-Lightning trainer for HuBERT Base model pre-training using MFCC features.
  • Add a PyTorch-Lightning trainer for HuBERT Large model pre-training using HuBERT Base model representations.
  • Add a PyTorch-Lightning trainer for HuBERT Large model fine-tuning on LibriSpeech ASR task.
Alternatives

No response

Additional context

No response

Contributor guide

Open the contributing guide

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.