LightHuBERT: Lightweight and Configurable Speech Representation Learning with Once-for-All Hidden-Unit BERT
Not on the map. Why?
Abstract (arXiv)
Self-supervised speech representation learning has shown promising results in various speech processing tasks. However, the pre-trained models, e.g., HuBERT, are storage-intensive Transformers, limiting their scope of applications under low-resource settings. To this end, we propose LightHuBERT, a once-for-all Transformer compression framework, to find the desired architectures automatically by pruning structured parameters. More precisely, we create a Transformer-based supernet that is nested with thousands of weight-sharing subnets and design a two-stage distillation strategy to leverage the contextualized latent representations from HuBERT. Experiments on automatic speech recognition (ASR) and the SUPERB benchmark show the proposed LightHuBERT enables over 109 architectures concerning the embedding dimension, attention dimension, head number, feed-forward network ratio, and network depth. LightHuBERT outperforms the original HuBERT on ASR and five SUPERB tasks with the HuBERT size, achieves comparable performance to the teacher model in most tasks with a reduction of 29% parameters, and obtains a 3.5× compression ratio in three SUPERB tasks, e.g., automatic speaker verification, keyword spotting, and intent classification, with a slight accuracy loss. The code and pre-trained models are available at https://github.com/mechanicalsea/lighthubert.
Key facts. Cited by 50 papers in the map (data through 14 Oct 2026). Builds on 28 of its 33 references that are in the map. First posted on arXiv 29 Mar 2022.
Builds on
28 of 33 references.
References · 28
- 2019DistilBERT, a distilled version of BERT: smaller, faster, cheaper and lightercited by 4,307
- 2020wav2vec 2.0: A Framework for Self-Supervised Learning of Speech Representationscited by 3,905
- 2021HuBERT: Self-Supervised Speech Representation Learning by Masked Prediction of Hidden Unitscited by 2,342
- 2021WavLM: Large-Scale Self-Supervised Pre-Training for Full Stack Speech Processingcited by 1,657
- 2019fairseq: A Fast, Extensible Toolkit for Sequence Modelingcited by 1,488
- 2019Once-for-All: Train One Network and Specialize it for Efficient Deploymentcited by 787
- 2022data2vec: A General Framework for Self-supervised Learning in Speech, Vision and Languagecited by 727
- 2021SUPERB: Speech processing Universal PERformance Benchmarkcited by 666
- 2019The State of Sparsity in Deep Neural Networkscited by 476
- 2019Libri-Light: A Benchmark for ASR with Limited or No Supervisioncited by 467
- 2019AutoML: A Survey of the State-of-the-Artcited by 390
- 2021W2v-BERT: Combining Contrastive Learning and Masked Language Modeling for Self-Supervised Speech Pre-Trainingcited by 347
16 more
- 2020DynaBERT: Dynamic BERT with Adaptive Width and Depthcited by 251
- 2021AutoFormer: Searching Transformers for Visual Recognitioncited by 188
- 2021SpeechT5: Unified-Modal Encoder-Decoder Pre-Training for Spoken Language Processingcited by 180
- 2021DistilHuBERT: Speech Representation Learning by Layer-wise Distillation of Hidden-unit BERTcited by 144
- 2019A Survey of Deep Learning Applications to Autonomous Vehicle Controlcited by 106
- 2019Knowledge Distillation from Internal Representationscited by 87
- 2022Unified Visual Transformer Compressioncited by 79
- 2021UniSpeech: Unified Speech Representation Learning with Labeled and Unlabeled Datacited by 77
- 2021Prune Once for All: Sparse Pre-Trained Language Modelscited by 66
- 2020Neural Architecture Transfercited by 60
- 2021PARP: Prune, Adjust and Re-Prune for Self-Supervised Speech Recognitioncited by 53
- 2018Speech2Vec: A Sequence-to-Sequence Framework for Learning Word Embeddings from Speechcited by 52
- 2021Shrinking Bigfoot: Reducing wav2vec 2.0 footprintcited by 21
- 2019Speech-XLNet: Unsupervised Acoustic Model Pretraining For Self-Attention Networkscited by 17
- 2021You Only Compress Once: Towards Effective and Elastic BERT Compression via Exploit-Explore Stochastic Nature Gradientcited by 14
- 2021EfficientTDNN: Efficient Architecture Search for Speaker Recognitioncited by 3
Built on by
Citing papers per year: 2022: 11; 2023: 21; 2024: 9; 2025: 7; 2026: 2.
50 papers.
Sort:
By area:
- 2024A Large-Scale Evaluation of Speech Foundation Modelscited by 40
- 2023DinoSR: Self-Distillation and Online Clustering for Self-supervised Speech Representation Learningcited by 37
- 2023DPHuBERT: Joint Distillation and Pruning of Self-Supervised Speech Modelscited by 36
- 2023A Survey of Techniques for Optimizing Transformer Inferencecited by 35
- 2023Structured Pruning of Self-Supervised Pre-trained Models for Speech Recognition and Understandingcited by 31
- 2022SpeechCLIP: Integrating Speech with Pre-Trained Vision and Language Modelcited by 29
- 2022SUPERB @ SLT 2022: Challenge on Generalization and Efficiency of Self-Supervised Speech Representation Learningcited by 26
- 2022FitHuBERT: Going Thinner and Deeper for Knowledge Distillation of Speech Self-Supervised Learningcited by 25
- 2023Reducing Barriers to Self-Supervised Learning: HuBERT Pre-training with Academic Computecited by 21
- 2023Fine-tuning Strategies for Faster Inference using Speech Self-Supervised Models: A Comparative Studycited by 15
- 2023Task-Agnostic Structured Pruning of Speech Representation Modelscited by 15
- 2023DQ-Whisper: Joint Distillation and Quantization for Efficient Multilingual Speech Recognitioncited by 11
38 more
- 2023I3D: Transformer architectures with input-dependent dynamic depth for speech recognitioncited by 10
- 2023CoLLD: Contrastive Layer-to-layer Distillation for Compressing Multilingual Pre-trained Speech Encoderscited by 9
- 2024Automatic Speech Recognition with BERT and CTC Transformers: A Reviewcited by 8
- 2023DistillW2V2: A Small and Streaming Wav2vec 2.0 Based ASR Modelcited by 7
- 2024Dynamic Encoder Size Based on Data-Driven Layer-wise Pruning for Speech Recognitioncited by 7
- 2023Recycle-and-Distill: Universal Compression Strategy for Transformer-based Speech SSL Models with Attention Map Reusing and Masking Distillationcited by 6
- 2022Is Smaller Always Faster? Tradeoffs in Compressing Self-Supervised Speech Transformerscited by 5
- 2024Efficiently Distilling LLMs for Edge Applicationscited by 5
- 2025Effective and Efficient Mixed Precision Quantization of Speech Foundation Modelscited by 5
- 2022Continual Learning for On-Device Speech Recognition using Disentangled Conformerscited by 4
- 2023Distilling HuBERT with LSTMs via Decoupled Knowledge Distillationcited by 4
- 2024Domain Adaptation in Intent Classification Systems: A Reviewcited by 4
- 2024DAISY: Data Adaptive Self-Supervised Early Exit for Speech Representation Modelscited by 4
- 2024One-pass Multiple Conformer and Foundation Speech Systems Compression and Quantization Using An All-in-one Neural Modelcited by 4
- 2022Exploring Effective Distillation of Self-Supervised Speech Models for Automatic Speech Recognitioncited by 3
- 2023One-Step Knowledge Distillation and Fine-Tuning in Using Large Pre-Trained Self-Supervised Learning Models for Speaker Verificationcited by 3
- 2023STaR: Distilling Speech Temporal Relation for Lightweight Speech Self-Supervised Learning Modelscited by 3
- 2025StableQuant: Layer Adaptive Post-Training Quantization for Speech Foundation Modelscited by 3
- 2025DiceHuBERT: Distilling HuBERT with a Self-Supervised Learning Objectivecited by 3
- 2022Adversarial Speaker Distillation for Countermeasure Model on Automatic Speaker Verificationcited by 2
- 2022Once-for-All Sequence Compression for Self-Supervised Speech Modelscited by 2
- 2023On-Device Constrained Self-Supervised Speech Representation Learning for Keyword Spotting via Knowledge Distillationcited by 2
- 2022Improving the Robustness of DistilHuBERT to Unseen Noisy Conditions via Data Augmentation, Curriculum Learning, and Multi-Task Enhancementcited by 1
- 2023Small-footprint slimmable networks for keyword spottingcited by 1
- 2023Efficiency-oriented approaches for self-supervised speech representation learningcited by 1
- 2025Efficient Supernet Training with Orthogonal Softmax for Scalable ASR Model Compressioncited by 1
- 2025Effective and Efficient One-pass Compression of Speech Foundation Models Using Sparsity-aware Self-pinching Gatescited by 1
- 2025HARNESS: Lightweight Distilled Arabic Speech Foundation Modelscited by 1
- 2022Real-time Speech Interruption Analysis: From Cloud to Client Deploymentcited by 0
- 2022Application of Knowledge Distillation to Multi-task Speech Representation Learningcited by 0
- 2023Mispronunciation detection using self-supervised speech representationscited by 0
- 2023GenDistiller: Distilling Pre-trained Language Models based on Generative Modelscited by 0
- 2023Noise robust distillation of self-supervised speech models via correlation metricscited by 0
- 2024GenDistiller: Distilling Pre-trained Language Models based on an Autoregressive Generative Modelcited by 0
- 2024Speech Representation Learning Revisited: The Necessity of Separate Learnable Parameters and Robust Data Augmentationcited by 0
- 2025Synergistic Effects of Knowledge Distillation and Structured Pruning for Self-Supervised Speech Modelscited by 0
- 2026HARNESS: Lightweight Distilled Arabic Speech Foundation Modelscited by 0
- 2026Towards Data-free and Training-free Compression for Speech Foundation Models Using Parameter Clusteringcited by 0
- 2026Towards Data-free and Training-free Compression for Speech Foundation Models Using Parameter Clustering10 Jun 2026
- 2026HARNESS: Lightweight Distilled Arabic Speech Foundation Models31 Mar 2026
- 2025HARNESS: Lightweight Distilled Arabic Speech Foundation Models18 Sep 2025
- 2025DiceHuBERT: Distilling HuBERT with a Self-Supervised Learning Objective25 Jun 2025
- 2025Effective and Efficient One-pass Compression of Speech Foundation Models Using Sparsity-aware Self-pinching Gates28 May 2025
- 2025StableQuant: Layer Adaptive Post-Training Quantization for Speech Foundation Models21 Apr 2025
- 2025Synergistic Effects of Knowledge Distillation and Structured Pruning for Self-Supervised Speech Models9 Feb 2025
- 2025Efficient Supernet Training with Orthogonal Softmax for Scalable ASR Model Compression31 Jan 2025
- 2025Effective and Efficient Mixed Precision Quantization of Speech Foundation Models7 Jan 2025
- 2024Automatic Speech Recognition with BERT and CTC Transformers: A Review12 Oct 2024
- 2024Speech Representation Learning Revisited: The Necessity of Separate Learnable Parameters and Robust Data Augmentation20 Aug 2024
- 2024Dynamic Encoder Size Based on Data-Driven Layer-wise Pruning for Speech Recognition10 Jul 2024
38 more
- 2024One-pass Multiple Conformer and Foundation Speech Systems Compression and Quantization Using An All-in-one Neural Model14 Jun 2024
- 2024GenDistiller: Distilling Pre-trained Language Models based on an Autoregressive Generative Model12 Jun 2024
- 2024DAISY: Data Adaptive Self-Supervised Early Exit for Speech Representation Models8 Jun 2024
- 2024A Large-Scale Evaluation of Speech Foundation Models15 Apr 2024
- 2024Efficiently Distilling LLMs for Edge Applications1 Apr 2024
- 2024Domain Adaptation in Intent Classification Systems: A Review26 Mar 2024
- 2023Noise robust distillation of self-supervised speech models via correlation metrics19 Dec 2023
- 2023Efficiency-oriented approaches for self-supervised speech representation learning18 Dec 2023
- 2023STaR: Distilling Speech Temporal Relation for Lightweight Speech Self-Supervised Learning Models14 Dec 2023
- 2023GenDistiller: Distilling Pre-trained Language Models based on Generative Models20 Oct 2023
- 2023Distilling HuBERT with LSTMs via Decoupled Knowledge Distillation18 Sep 2023
- 2023CoLLD: Contrastive Layer-to-layer Distillation for Compressing Multilingual Pre-trained Speech Encoders14 Sep 2023
- 2023Mispronunciation detection using self-supervised speech representations30 Jul 2023
- 2023A Survey of Techniques for Optimizing Transformer Inference16 Jul 2023
- 2023On-Device Constrained Self-Supervised Speech Representation Learning for Keyword Spotting via Knowledge Distillation6 Jul 2023
- 2023Reducing Barriers to Self-Supervised Learning: HuBERT Pre-training with Academic Compute11 Jun 2023
- 2023Task-Agnostic Structured Pruning of Speech Representation Models2 Jun 2023
- 2023DPHuBERT: Joint Distillation and Pruning of Self-Supervised Speech Models28 May 2023
- 2023One-Step Knowledge Distillation and Fine-Tuning in Using Large Pre-Trained Self-Supervised Learning Models for Speaker Verification27 May 2023
- 2023Recycle-and-Distill: Universal Compression Strategy for Transformer-based Speech SSL Models with Attention Map Reusing and Masking Distillation19 May 2023
- 2023DQ-Whisper: Joint Distillation and Quantization for Efficient Multilingual Speech Recognition18 May 2023
- 2023DinoSR: Self-Distillation and Online Clustering for Self-supervised Speech Representation Learning17 May 2023
- 2023Small-footprint slimmable networks for keyword spotting21 Apr 2023
- 2023DistillW2V2: A Small and Streaming Wav2vec 2.0 Based ASR Model16 Mar 2023
- 2023I3D: Transformer architectures with input-dependent dynamic depth for speech recognition14 Mar 2023
- 2023Fine-tuning Strategies for Faster Inference using Speech Self-Supervised Models: A Comparative Study12 Mar 2023
- 2023Structured Pruning of Self-Supervised Pre-trained Models for Speech Recognition and Understanding27 Feb 2023
- 2022Continual Learning for On-Device Speech Recognition using Disentangled Conformers2 Dec 2022
- 2022Is Smaller Always Faster? Tradeoffs in Compressing Self-Supervised Speech Transformers17 Nov 2022
- 2022Improving the Robustness of DistilHuBERT to Unseen Noisy Conditions via Data Augmentation, Curriculum Learning, and Multi-Task Enhancement12 Nov 2022
- 2022Once-for-All Sequence Compression for Self-Supervised Speech Models4 Nov 2022
- 2022Application of Knowledge Distillation to Multi-task Speech Representation Learning29 Oct 2022
- 2022Exploring Effective Distillation of Self-Supervised Speech Models for Automatic Speech Recognition27 Oct 2022
- 2022Real-time Speech Interruption Analysis: From Cloud to Client Deployment24 Oct 2022
- 2022SUPERB @ SLT 2022: Challenge on Generalization and Efficiency of Self-Supervised Speech Representation Learning16 Oct 2022
- 2022SpeechCLIP: Integrating Speech with Pre-Trained Vision and Language Model3 Oct 2022
- 2022FitHuBERT: Going Thinner and Deeper for Knowledge Distillation of Speech Self-Supervised Learning1 Jul 2022
- 2022Adversarial Speaker Distillation for Countermeasure Model on Automatic Speaker Verification31 Mar 2022
- 2023DinoSR: Self-Distillation and Online Clustering for Self-supervised Speech Representation Learningcited by 37
- 2022FitHuBERT: Going Thinner and Deeper for Knowledge Distillation of Speech Self-Supervised Learningcited by 25
- 2023DistillW2V2: A Small and Streaming Wav2vec 2.0 Based ASR Modelcited by 7
- 2024Efficiently Distilling LLMs for Edge Applicationscited by 5
- 2023Distilling HuBERT with LSTMs via Decoupled Knowledge Distillationcited by 4
- 2023One-Step Knowledge Distillation and Fine-Tuning in Using Large Pre-Trained Self-Supervised Learning Models for Speaker Verificationcited by 3
- 2025DiceHuBERT: Distilling HuBERT with a Self-Supervised Learning Objectivecited by 3
- 2023On-Device Constrained Self-Supervised Speech Representation Learning for Keyword Spotting via Knowledge Distillationcited by 2
- 2025HARNESS: Lightweight Distilled Arabic Speech Foundation Modelscited by 1
- 2022Application of Knowledge Distillation to Multi-task Speech Representation Learningcited by 0
- 2023GenDistiller: Distilling Pre-trained Language Models based on Generative Modelscited by 0
- 2023Noise robust distillation of self-supervised speech models via correlation metricscited by 0
12 most cited of 13 All 13 (JSON)
- 2025HARNESS: Lightweight Distilled Arabic Speech Foundation Models18 Sep 2025
- 2025DiceHuBERT: Distilling HuBERT with a Self-Supervised Learning Objective25 Jun 2025
- 2025Synergistic Effects of Knowledge Distillation and Structured Pruning for Self-Supervised Speech Models9 Feb 2025
- 2024Efficiently Distilling LLMs for Edge Applications1 Apr 2024
- 2023Noise robust distillation of self-supervised speech models via correlation metrics19 Dec 2023
- 2023GenDistiller: Distilling Pre-trained Language Models based on Generative Models20 Oct 2023
- 2023Distilling HuBERT with LSTMs via Decoupled Knowledge Distillation18 Sep 2023
- 2023On-Device Constrained Self-Supervised Speech Representation Learning for Keyword Spotting via Knowledge Distillation6 Jul 2023
- 2023One-Step Knowledge Distillation and Fine-Tuning in Using Large Pre-Trained Self-Supervised Learning Models for Speaker Verification27 May 2023
- 2023DinoSR: Self-Distillation and Online Clustering for Self-supervised Speech Representation Learning17 May 2023
- 2023DistillW2V2: A Small and Streaming Wav2vec 2.0 Based ASR Model16 Mar 2023
- 2022Application of Knowledge Distillation to Multi-task Speech Representation Learning29 Oct 2022
12 newest of 13 All 13 (JSON)
- 2024A Large-Scale Evaluation of Speech Foundation Modelscited by 40
- 2023DPHuBERT: Joint Distillation and Pruning of Self-Supervised Speech Modelscited by 36
- 2023A Survey of Techniques for Optimizing Transformer Inferencecited by 35
- 2023Structured Pruning of Self-Supervised Pre-trained Models for Speech Recognition and Understandingcited by 31
- 2022SpeechCLIP: Integrating Speech with Pre-Trained Vision and Language Modelcited by 29
- 2022SUPERB @ SLT 2022: Challenge on Generalization and Efficiency of Self-Supervised Speech Representation Learningcited by 26
- 2023Reducing Barriers to Self-Supervised Learning: HuBERT Pre-training with Academic Computecited by 21
- 2023Fine-tuning Strategies for Faster Inference using Speech Self-Supervised Models: A Comparative Studycited by 15
- 2023Task-Agnostic Structured Pruning of Speech Representation Modelscited by 15
- 2023DQ-Whisper: Joint Distillation and Quantization for Efficient Multilingual Speech Recognitioncited by 11
- 2023I3D: Transformer architectures with input-dependent dynamic depth for speech recognitioncited by 10
- 2023CoLLD: Contrastive Layer-to-layer Distillation for Compressing Multilingual Pre-trained Speech Encoderscited by 9
12 most cited of 36 All 36 (JSON)
- 2026Towards Data-free and Training-free Compression for Speech Foundation Models Using Parameter Clustering10 Jun 2026
- 2026HARNESS: Lightweight Distilled Arabic Speech Foundation Models31 Mar 2026
- 2025Effective and Efficient One-pass Compression of Speech Foundation Models Using Sparsity-aware Self-pinching Gates28 May 2025
- 2025StableQuant: Layer Adaptive Post-Training Quantization for Speech Foundation Models21 Apr 2025
- 2025Efficient Supernet Training with Orthogonal Softmax for Scalable ASR Model Compression31 Jan 2025
- 2025Effective and Efficient Mixed Precision Quantization of Speech Foundation Models7 Jan 2025
- 2024Automatic Speech Recognition with BERT and CTC Transformers: A Review12 Oct 2024
- 2024Speech Representation Learning Revisited: The Necessity of Separate Learnable Parameters and Robust Data Augmentation20 Aug 2024
- 2024Dynamic Encoder Size Based on Data-Driven Layer-wise Pruning for Speech Recognition10 Jul 2024
- 2024One-pass Multiple Conformer and Foundation Speech Systems Compression and Quantization Using An All-in-one Neural Model14 Jun 2024
- 2024GenDistiller: Distilling Pre-trained Language Models based on an Autoregressive Generative Model12 Jun 2024
- 2024A Large-Scale Evaluation of Speech Foundation Models15 Apr 2024
12 newest of 36 All 36 (JSON)