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.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Automatic Speech Recognition | Librispeech (test-clean) | WER13.37 | 170 | |
| Emotion Recognition | ER | Accuracy63.02 | 52 | |
| Slot Filling | SUPERB SF | F1 Score82.57 | 33 | |
| Speaker Identification | SUPERB SID | Accuracy73.54 | 14 | |
| Automatic Speech Recognition | SUPERB ASR | WER13.37 | 14 | |
| Phoneme Recognition | SUPERB PR | PER16.27 | 14 |