Lifting the Curse of Capacity Gap in Distilling Language Models
On the map: Mixture-of-experts → MoE large language models
also in 1 more
Abstract (arXiv)
Pretrained language models (LMs) have shown compelling performance on various downstream tasks, but unfortunately they require a tremendous amount of inference compute. Knowledge distillation finds a path to compress LMs to small ones with a teacher-student paradigm. However, when the capacity gap between the teacher and the student is large, a curse of capacity gap appears, invoking a deficiency in distilling LMs. While a few studies have been carried out to fill the gap, the curse is not yet well tackled. In this paper, we aim at lifting the curse of capacity gap via enlarging the capacity of the student without notably increasing the inference compute. Largely motivated by sparse activation regime of mixture of experts (MoE), we propose a mixture of minimal experts (MiniMoE), which imposes extra parameters to the student but introduces almost no additional inference compute. Experimental results on GLUE and CoNLL demonstrate the curse of capacity gap is lifted by the magic of MiniMoE to a large extent. MiniMoE also achieves the state-of-the-art performance at small FLOPs compared with a range of competitive baselines. With a compression rate as much as ~50×, MiniMoE preserves ~95% GLUE score of the teacher.
Key facts. Cited by 25 papers in the map (data through 14 Oct 2026). Placed in MoE large language models, one of 12 directions in Mixture-of-experts. Builds on 55 of its 76 references that are in the map. First posted on arXiv 20 May 2023.
Builds on
55 of 76 references.
Closest ancestors · 4
Same direction, newest first
- 2022MoEBERT: from BERT to Mixture-of-Experts via Importance-Guided Adaptation15 Apr
- 2021GLaM: Efficient Scaling of Language Models with Mixture-of-Experts13 Dec
- 2020GShard: Scaling Giant Models with Conditional Computation and Automatic Sharding30 Jun
- 2017Outrageously Large Neural Networks: The Sparsely-Gated Mixture-of-Experts Layer23 Jan
Other references · 51
- 2017Attention Is All You Needcited by 60,361
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,399
- 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,599
- 2019RoBERTa: A Robustly Optimized BERT Pretraining Approachcited by 13,372
- 2015Distilling the Knowledge in a Neural Networkcited by 10,324
- 2022PaLM: Scaling Language Modeling with Pathwayscited by 5,326
- 2018GLUE: A Multi-Task Benchmark and Analysis Platform for Natural Language Understandingcited by 4,899
- 2019DistilBERT, a distilled version of BERT: smaller, faster, cheaper and lightercited by 4,307
- 2016SQuAD: 100,000+ Questions for Machine Comprehension of Textcited by 4,148
- 2021Switch Transformers: Scaling to Trillion Parameter Models with Simple and Efficient Sparsitycited by 3,073
- 2022Scaling Instruction-Finetuned Language Modelscited by 3,024
- 2017A Broad-Coverage Challenge Corpus for Sentence Understanding through Inferencecited by 2,682
38 more
- 2015Learning both Weights and Connections for Efficient Neural Networkscited by 2,031
- 2019Megatron-LM: Training Multi-Billion Parameter Language Models Using Model Parallelismcited by 1,868
- 2020MiniLM: Deep Self-Attention Distillation for Task-Agnostic Compression of Pre-Trained Transformerscited by 1,463
- 2019TinyBERT: Distilling BERT for Natural Language Understandingcited by 1,167
- 2017SemEval-2017 Task 1: Semantic Textual Similarity - Multilingual and Cross-lingual Focused Evaluationcited by 1,001
- 2018Neural Network Acceptability Judgmentscited by 958
- 2019Improved Knowledge Distillation via Teacher Assistantcited by 573
- 2019Patient Knowledge Distillation for BERT Model Compressioncited by 508
- 2020MobileBERT: a Compact Task-Agnostic BERT for Resource-Limited Devicescited by 482
- 2021Dynamic Neural Networks: A Surveycited by 408
- 2022DeepSpeed-MoE: Advancing Mixture-of-Experts Inference and Training to Power Next-Generation AI Scalecited by 404
- 2022Decoupled Knowledge Distillationcited by 388
- 2019On the Efficacy of Knowledge Distillationcited by 381
- 2020DeeBERT: Dynamic Early Exiting for Accelerating BERT Inferencecited by 332
- 2019Q8BERT: Quantized 8Bit BERTcited by 300
- 2020CLUE: A Chinese Language Understanding Evaluation Benchmarkcited by 278
- 2020BERT Loses Patience: Fast and Robust Inference with Early Exitcited by 273
- 2020DynaBERT: Dynamic BERT with Adaptive Width and Depthcited by 251
- 2020MiniLMv2: Multi-Head Self-Attention Relation Distillation for Compressing Pretrained Transformerscited by 245
- 2021Hash Layers For Large Sparse Modelscited by 237
- 2015Unifying distillation and privileged informationcited by 233
- 2020BinaryBERT: Pushing the Limit of BERT Quantizationcited by 178
- 2022Structured Pruning Learns Compact and Accurate Modelscited by 170
- 2018Mesh-TensorFlow: Deep Learning for Supercomputerscited by 169
- 2015Conditional Computation in Neural Networks for faster modelscited by 167
- 2021MoEfication: Transformer Feed-forward Layers are Mixtures of Expertscited by 152
- 2019Well-Read Students Learn Better: On the Importance of Pre-training Compact Modelscited by 120
- 2021FastMoE: A Fast Mixture-of-Expert Training Systemcited by 112
- 2020Length-Adaptive Transformer: Train Once with Length Drop, Use Anytime with Searchcited by 76
- 2021BERT Learns to Teach: Knowledge Distillation with Meta Learningcited by 61
- 2021Learning Student-Friendly Teacher Networks for Knowledge Distillationcited by 60
- 2020Contrastive Distillation on Intermediate Representations for Language Model Compressioncited by 57
- 2021ERNIE 3.0 Titan: Exploring Larger-scale Knowledge Enhanced Pre-training for Language Understanding and Generationcited by 46
- 2015Resiliency of Deep Neural Networks under Quantizationcited by 35
- 2021Distilling Linguistic Context for Language Model Compressioncited by 33
- 2020BERT-EMD: Many-to-Many Layer Mapping for BERT Compression with Earth Mover's Distancecited by 27
- 2022One Student Knows All Experts Know: From Sparse to Densecited by 25
- 2022Sparse Teachers Can Be Dense with Knowledgecited by 11
50 most cited of 51 All 51 (JSON)
Built on by
Citing papers per year: 2023: 7; 2024: 6; 2025: 10; 2026: 2.
25 papers.
Sort:
By area:
- 2024A Survey on Efficient Inference for Large Language Modelscited by 205
- 2023MobileVLM : A Fast, Strong and Open Vision Language Assistant for Mobile Devicescited by 122
- 2025Small Models Struggle to Learn from Strong Reasonerscited by 113
- 2023Towards the Law of Capacity Gap in Distilling Language Modelscited by 44
- 2024Revisiting Knowledge Distillation for Autoregressive Language Modelscited by 42
- 2023Norm Tweaking: High-performance Low-bit Quantization of Large Language Modelscited by 33
- 2024Speculative Decoding via Early-exiting for Faster LLM Inference with Thompson Sampling Control Mechanismcited by 32
- 2023The Efficiency Spectrum of Large Language Models: An Algorithmic Surveycited by 26
- 2025TAID: Temporally Adaptive Interpolated Distillation for Efficient Knowledge Transfer in Language Modelscited by 22
- 2023FPTQ: Fine-grained Post-Training Quantization for Large Language Modelscited by 16
- 2024PLaD: Preference-based Large Language Model Distillation with Pseudo-Preference Pairscited by 11
- 2024Don't Throw Away Data: Better Sequence Knowledge Distillationcited by 9
13 more
- 2025MiCoTA: Bridging the Learnability Gap with Intermediate CoT and Teacher Assistantscited by 9
- 2025Put Teacher in Student's Shoes: Cross-Distillation for Ultra-compact Model Compression Frameworkcited by 8
- 2025Distilled Pretraining: A modern lens of Data, In-Context Learning and Test-Time Scalingcited by 7
- 2025Masking Teacher and Reinforcing Student for Distilling Vision-Language Modelscited by 6
- 2025In Good GRACEs: Principled Teacher Selection for Knowledge Distillationcited by 5
- 2023Task-agnostic Distillation of Encoder-Decoder Language Modelscited by 4
- 2026Pedagogically-Inspired Data Synthesis for Language Model Knowledge Distillationcited by 3
- 2025Can Compact Language Models Search Like Agents? Distillation-Guided Policy Optimization for Preserving Agentic RAG Capabilitiescited by 2
- 2025AMiD: Knowledge Distillation for LLMs with α-mixture Assistant Distributioncited by 2
- 2024MoDification: Mixture of Depths Made Easycited by 1
- 2023On Elastic Language Modelscited by 0
- 2025LLMs in Mobile Apps: Practices, Challenges, and Opportunitiescited by 0
- 2026Evolving Knowledge Distillation for Lightweight Neural Machine Translationcited by 0
- 2026Evolving Knowledge Distillation for Lightweight Neural Machine Translation11 May 2026
- 2026Pedagogically-Inspired Data Synthesis for Language Model Knowledge Distillation12 Feb 2026
- 2025Masking Teacher and Reinforcing Student for Distilling Vision-Language Models23 Dec 2025
- 2025In Good GRACEs: Principled Teacher Selection for Knowledge Distillation4 Nov 2025
- 2025AMiD: Knowledge Distillation for LLMs with α-mixture Assistant Distribution13 Oct 2025
- 2025Distilled Pretraining: A modern lens of Data, In-Context Learning and Test-Time Scaling1 Sep 2025
- 2025Can Compact Language Models Search Like Agents? Distillation-Guided Policy Optimization for Preserving Agentic RAG Capabilities27 Aug 2025
- 2025Put Teacher in Student's Shoes: Cross-Distillation for Ultra-compact Model Compression Framework7 Jul 2025
- 2025MiCoTA: Bridging the Learnability Gap with Intermediate CoT and Teacher Assistants2 Jul 2025
- 2025LLMs in Mobile Apps: Practices, Challenges, and Opportunities21 Feb 2025
- 2025Small Models Struggle to Learn from Strong Reasoners17 Feb 2025
- 2025TAID: Temporally Adaptive Interpolated Distillation for Efficient Knowledge Transfer in Language Models28 Jan 2025
13 more
- 2024MoDification: Mixture of Depths Made Easy18 Oct 2024
- 2024Don't Throw Away Data: Better Sequence Knowledge Distillation15 Jul 2024
- 2024Speculative Decoding via Early-exiting for Faster LLM Inference with Thompson Sampling Control Mechanism6 Jun 2024
- 2024PLaD: Preference-based Large Language Model Distillation with Pseudo-Preference Pairs5 Jun 2024
- 2024A Survey on Efficient Inference for Large Language Models22 Apr 2024
- 2024Revisiting Knowledge Distillation for Autoregressive Language Models19 Feb 2024
- 2023MobileVLM : A Fast, Strong and Open Vision Language Assistant for Mobile Devices28 Dec 2023
- 2023The Efficiency Spectrum of Large Language Models: An Algorithmic Survey1 Dec 2023
- 2023On Elastic Language Models13 Nov 2023
- 2023Towards the Law of Capacity Gap in Distilling Language Models13 Nov 2023
- 2023Norm Tweaking: High-performance Low-bit Quantization of Large Language Models6 Sep 2023
- 2023FPTQ: Fine-grained Post-Training Quantization for Large Language Models30 Aug 2023
- 2023Task-agnostic Distillation of Encoder-Decoder Language Models21 May 2023
- 2023Towards the Law of Capacity Gap in Distilling Language Modelscited by 44
- 2024Revisiting Knowledge Distillation for Autoregressive Language Modelscited by 42
- 2025TAID: Temporally Adaptive Interpolated Distillation for Efficient Knowledge Transfer in Language Modelscited by 22
- 2024PLaD: Preference-based Large Language Model Distillation with Pseudo-Preference Pairscited by 11
- 2024Don't Throw Away Data: Better Sequence Knowledge Distillationcited by 9
- 2025In Good GRACEs: Principled Teacher Selection for Knowledge Distillationcited by 5
- 2025AMiD: Knowledge Distillation for LLMs with α-mixture Assistant Distributioncited by 2
- 2026Evolving Knowledge Distillation for Lightweight Neural Machine Translationcited by 0
- 2026Evolving Knowledge Distillation for Lightweight Neural Machine Translation11 May 2026
- 2025In Good GRACEs: Principled Teacher Selection for Knowledge Distillation4 Nov 2025
- 2025AMiD: Knowledge Distillation for LLMs with α-mixture Assistant Distribution13 Oct 2025
- 2025TAID: Temporally Adaptive Interpolated Distillation for Efficient Knowledge Transfer in Language Models28 Jan 2025
- 2024Don't Throw Away Data: Better Sequence Knowledge Distillation15 Jul 2024
- 2024PLaD: Preference-based Large Language Model Distillation with Pseudo-Preference Pairs5 Jun 2024
- 2024Revisiting Knowledge Distillation for Autoregressive Language Models19 Feb 2024
- 2023Towards the Law of Capacity Gap in Distilling Language Models13 Nov 2023
- 2023Norm Tweaking: High-performance Low-bit Quantization of Large Language Modelscited by 33
- 2023The Efficiency Spectrum of Large Language Models: An Algorithmic Surveycited by 26
- 2023FPTQ: Fine-grained Post-Training Quantization for Large Language Modelscited by 16
- 2025Put Teacher in Student's Shoes: Cross-Distillation for Ultra-compact Model Compression Frameworkcited by 8
- 2023Task-agnostic Distillation of Encoder-Decoder Language Modelscited by 4
- 2024MoDification: Mixture of Depths Made Easycited by 1
- 2023On Elastic Language Modelscited by 0
- 2025LLMs in Mobile Apps: Practices, Challenges, and Opportunitiescited by 0
- 2025Put Teacher in Student's Shoes: Cross-Distillation for Ultra-compact Model Compression Framework7 Jul 2025
- 2025LLMs in Mobile Apps: Practices, Challenges, and Opportunities21 Feb 2025
- 2024MoDification: Mixture of Depths Made Easy18 Oct 2024
- 2023The Efficiency Spectrum of Large Language Models: An Algorithmic Survey1 Dec 2023
- 2023On Elastic Language Models13 Nov 2023
- 2023Norm Tweaking: High-performance Low-bit Quantization of Large Language Models6 Sep 2023
- 2023FPTQ: Fine-grained Post-Training Quantization for Large Language Models30 Aug 2023
- 2023Task-agnostic Distillation of Encoder-Decoder Language Models21 May 2023
Similar, not cited
Close in wording, no citation either way
- 2024DistiLLM: Towards Streamlined Distillation for Large Language Models
- 2026Benchmarking Distilled Language Models: Performance and Efficiency in Resource-Constrained Settings
- 2026Distilling What Matters: Confidence-Aware Selective Distillation for Large Language Models
- 2024MiniPLM: Knowledge Distillation for Pre-Training Language Models
- 2024Survey on Knowledge Distillation for Large Language Models: Methods, Evaluation, and Application
- 2026Less is MoE: Trimming Experts in Domain-Specialist Language Models
- 2026Strong Teacher Not Needed? On Distillation in LLM Pretraining
- 2025Every Expert Matters: Towards Effective Knowledge Distillation for Mixture-of-Experts Language Models