Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

DistilHuBERT: Speech Representation Learning by Layer-wise Distillation of Hidden-unit BERT

About

Self-supervised speech representation learning methods like wav2vec 2.0 and Hidden-unit BERT (HuBERT) leverage unlabeled speech data for pre-training and offer good representations for numerous speech processing tasks. Despite the success of these methods, they require large memory and high pre-training costs, making them inaccessible for researchers in academia and small companies. Therefore, this paper introduces DistilHuBERT, a novel multi-task learning framework to distill hidden representations from a HuBERT model directly. This method reduces HuBERT's size by 75% and 73% faster while retaining most performance in ten different tasks. Moreover, DistilHuBERT required little training time and data, opening the possibilities of pre-training personal and on-device SSL models for speech.

Heng-Jui Chang, Shu-wen Yang, Hung-yi Lee• 2021

Related benchmarks

TaskDatasetResultRank
Automatic Speech RecognitionLibrispeech (test-clean)
WER13.37
170
Emotion RecognitionER
Accuracy63.02
52
Slot FillingSUPERB SF
F1 Score82.57
33
Speaker IdentificationSUPERB SID
Accuracy73.54
14
Automatic Speech RecognitionSUPERB ASR
WER13.37
14
Phoneme RecognitionSUPERB PR
PER16.27
14
Showing 6 of 6 rows

Other info

Follow for update