MoEBERT: from BERT to Mixture-of-Experts via Importance-Guided Adaptation
On the map: Mixture-of-experts → MoE large language models
Abstract (arXiv)
Pre-trained language models have demonstrated superior performance in various natural language processing tasks. However, these models usually contain hundreds of millions of parameters, which limits their practicality because of latency requirements in real-world applications. Existing methods train small compressed models via knowledge distillation. However, performance of these small models drops significantly compared with the pre-trained models due to their reduced model capacity. We propose MoEBERT, which uses a Mixture-of-Experts structure to increase model capacity and inference speed. We initialize MoEBERT by adapting the feed-forward neural networks in a pre-trained model into multiple experts. As such, representation power of the pre-trained model is largely retained. During inference, only one of the experts is activated, such that speed can be improved. We also propose a layer-wise distillation method to train MoEBERT. We validate the efficiency and effectiveness of MoEBERT on natural language understanding and question answering tasks. Results show that the proposed method outperforms existing task-specific distillation algorithms. For example, our method outperforms previous approaches by over 2% on the MNLI (mismatched) dataset. Our code is publicly available at https://github.com/SimiaoZuo/MoEBERT.
Key facts. Cited by 37 papers in the map (data through 7 Oct 2026). Placed in MoE large language models, one of 12 directions in Mixture-of-experts. Builds on 36 of its 53 references that are in the map. First posted on arXiv 15 Apr 2022.
Builds on
36 of 53 references.
Closest ancestors · 2
Same direction, newest first
- 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 · 34
- 2017Attention Is All You Needcited by 59,932
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
- 2014Adam: A Method for Stochastic Optimizationcited by 36,960
- 2020Language Models are Few-Shot Learnerscited by 32,199
- 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,491
- 2019RoBERTa: A Robustly Optimized BERT Pretraining Approachcited by 13,266
- 2015Distilling the Knowledge in a Neural Networkcited by 10,076
- 2018GLUE: A Multi-Task Benchmark and Analysis Platform for Natural Language Understandingcited by 4,854
- 2019DistilBERT, a distilled version of BERT: smaller, faster, cheaper and lightercited by 4,243
- 2016SQuAD: 100,000+ Questions for Machine Comprehension of Textcited by 4,115
- 2021Switch Transformers: Scaling to Trillion Parameter Models with Simple and Efficient Sparsitycited by 3,014
- 2017A Broad-Coverage Challenge Corpus for Sentence Understanding through Inferencecited by 2,665
22 more
- 2018Deep contextualized word representationscited by 2,285
- 2020DeBERTa: Decoding-enhanced BERT with Disentangled Attentioncited by 2,157
- 2014FitNets: Hints for Thin Deep Netscited by 1,619
- 2018Know What You Don't Know: Unanswerable Questions for SQuADcited by 1,538
- 2020MiniLM: Deep Self-Attention Distillation for Task-Agnostic Compression of Pre-Trained Transformerscited by 1,414
- 2021DeBERTaV3: Improving DeBERTa using ELECTRA-Style Pre-Training with Gradient-Disentangled Embedding Sharingcited by 1,219
- 2019TinyBERT: Distilling BERT for Natural Language Understandingcited by 1,153
- 2017SemEval-2017 Task 1: Semantic Textual Similarity - Multilingual and Cross-lingual Focused Evaluationcited by 998
- 2018Neural Network Acceptability Judgmentscited by 952
- 2019Importance Estimation for Neural Network Pruningcited by 531
- 2019Patient Knowledge Distillation for BERT Model Compressioncited by 502
- 2020MobileBERT: a Compact Task-Agnostic BERT for Resource-Limited Devicescited by 476
- 2021BASE Layers: Simplifying Training of Large, Sparse Modelscited by 315
- 2020DynaBERT: Dynamic BERT with Adaptive Width and Depthcited by 246
- 2020MiniLMv2: Multi-Head Self-Attention Relation Distillation for Compressing Pretrained Transformerscited by 240
- 2021Hash Layers For Large Sparse Modelscited by 236
- 2020BERT-of-Theseus: Compressing BERT by Progressive Module Replacingcited by 112
- 2021Taming Sparsely Activated Transformer with Stochastic Expertscited by 112
- 2020Contrastive Distillation on Intermediate Representations for Language Model Compressioncited by 57
- 2021Super Tickets in Pre-Trained Language Models: From Model Compression to Improving Generalizationcited by 56
- 2020BERT-EMD: Many-to-Many Layer Mapping for BERT Compression with Earth Mover's Distancecited by 27
- 2021M6-T: Exploring Sparse Expert Models and Beyondcited by 22
Built on by
Citing papers per year: 2022: 3; 2023: 11; 2024: 13; 2025: 6; 2026: 4.
37 papers.
Sort:
By area:
- 2024A Survey on Knowledge Distillation of Large Language Modelscited by 276
- 2022Sparse Upcycling: Training Mixture-of-Experts from Dense Checkpointscited by 207
- 2024LLaMA-MoE: Building Mixture-of-Experts from LLaMA with Continual Pre-trainingcited by 122
- 2022Less is More: Task-aware Layer-wise Distillation for Language Model Compressioncited by 91
- 2024Model Compression and Efficient Inference for Large Language Models: A Surveycited by 62
- 2023Towards the Law of Capacity Gap in Distilling Language Modelscited by 41
- 2023Language-driven All-in-one Adverse Weather Removalcited by 36
- 2023Lifting the Curse of Capacity Gap in Distilling Language Modelscited by 25
- 2024LLAVADI: What Matters For Multimodal Large Language Models Distillationcited by 25
- 2023Exploiting Activation Sparsity with Dense to Dynamic-k Mixture-of-Experts Conversioncited by 15
- 2022Multi-CLS BERT: An Efficient Alternative to Traditional Ensemblingcited by 14
- 2023Blockwise Parallel Transformer for Large Context Modelscited by 12
25 more
- 2024CFSP: An Efficient Structured Pruning Framework for LLMs with Coarse-to-Fine Activation Informationcited by 11
- 2024Multi-student Diffusion Distillation for Better One-step Generatorscited by 9
- 2024Flexible and Adaptable Summarization via Expertise Separationcited by 7
- 2025Analytical FFN-to-MoE Restructuring via Activation Pattern Analysiscited by 7
- 2024Exploring the Benefit of Activation Sparsity in Pre-trainingcited by 5
- 20253D Gaussian Splatting Data Compression with Mixture of Priorscited by 5
- 2024LaDiMo: Layer-wise Distillation Inspired MoEfiercited by 4
- 2023Partition-and-Debias: Agnostic Biases Mitigation via A Mixture of Biases-Specific Expertscited by 3
- 2023Task-oriented Memory-efficient Pruning-Adaptercited by 2
- 2023COMET: Learning Cardinality Constrained Mixture of Experts with Trees and Local Searchcited by 2
- 2024On Parameter Estimation in Deviated Gaussian Mixture of Expertscited by 2
- 2025Mixture of Routerscited by 2
- 2025WINA: Weight Informed Neuron Activation for Accelerating Large Language Model Inferencecited by 2
- 2023A Stability Analysis of Fine-Tuning a Pre-Trained Modelcited by 1
- 2024Contrastive Learning and Mixture of Experts Enables Precise Vector Embeddingscited by 1
- 2024ElastiFormer: Learned Redundancy Reduction in Transformer via Self-Distillationcited by 1
- 2023G-SPEED: General SParse Efficient Editing MoDelcited by 0
- 2023On Elastic Language Modelscited by 0
- 2024Exploiting Student Parallelism for Efficient GPU Inference of BERT-like Models in Online Servicescited by 0
- 2025NEU-ESC: A Comprehensive Vietnamese dataset for Educational Sentiment analysis and topic Classification toward multitask learningcited by 0
- 2025ExpertSim: Fast Particle Detector Simulation Using Mixture-of-Generative-Expertscited by 0
- 2026ExFusion: Efficient Transformer Training via Multi-Experts Fusioncited by 0
- 2026Mixture-of-Experts in Remote Sensing: A Surveycited by 0
- 2026A Study on Hidden Layer Distillation for Large Language Model Pre-Trainingcited by 0
- 2026KDFP: A first-principles approach to knowledge distillation in large language modelscited by 0
- 2026KDFP: A first-principles approach to knowledge distillation in large language models7 Oct 2026
- 2026A Study on Hidden Layer Distillation for Large Language Model Pre-Training12 May 2026
- 2026Mixture-of-Experts in Remote Sensing: A Survey3 Apr 2026
- 2026ExFusion: Efficient Transformer Training via Multi-Experts Fusion30 Mar 2026
- 2025ExpertSim: Fast Particle Detector Simulation Using Mixture-of-Generative-Experts28 Aug 2025
- 2025NEU-ESC: A Comprehensive Vietnamese dataset for Educational Sentiment analysis and topic Classification toward multitask learning30 Jun 2025
- 2025WINA: Weight Informed Neuron Activation for Accelerating Large Language Model Inference26 May 2025
- 20253D Gaussian Splatting Data Compression with Mixture of Priors6 May 2025
- 2025Mixture of Routers30 Mar 2025
- 2025Analytical FFN-to-MoE Restructuring via Activation Pattern Analysis6 Feb 2025
- 2024ElastiFormer: Learned Redundancy Reduction in Transformer via Self-Distillation22 Nov 2024
- 2024Multi-student Diffusion Distillation for Better One-step Generators30 Oct 2024
25 more
- 2024Exploring the Benefit of Activation Sparsity in Pre-training4 Oct 2024
- 2024CFSP: An Efficient Structured Pruning Framework for LLMs with Coarse-to-Fine Activation Information20 Sep 2024
- 2024Exploiting Student Parallelism for Efficient GPU Inference of BERT-like Models in Online Services22 Aug 2024
- 2024LaDiMo: Layer-wise Distillation Inspired MoEfier8 Aug 2024
- 2024LLAVADI: What Matters For Multimodal Large Language Models Distillation28 Jul 2024
- 2024LLaMA-MoE: Building Mixture-of-Experts from LLaMA with Continual Pre-training24 Jun 2024
- 2024Flexible and Adaptable Summarization via Expertise Separation8 Jun 2024
- 2024A Survey on Knowledge Distillation of Large Language Models20 Feb 2024
- 2024Model Compression and Efficient Inference for Large Language Models: A Survey15 Feb 2024
- 2024On Parameter Estimation in Deviated Gaussian Mixture of Experts7 Feb 2024
- 2024Contrastive Learning and Mixture of Experts Enables Precise Vector Embeddings28 Jan 2024
- 2023Language-driven All-in-one Adverse Weather Removal3 Dec 2023
- 2023On Elastic Language Models13 Nov 2023
- 2023Towards the Law of Capacity Gap in Distilling Language Models13 Nov 2023
- 2023G-SPEED: General SParse Efficient Editing MoDel16 Oct 2023
- 2023Exploiting Activation Sparsity with Dense to Dynamic-k Mixture-of-Experts Conversion6 Oct 2023
- 2023Partition-and-Debias: Agnostic Biases Mitigation via A Mixture of Biases-Specific Experts19 Aug 2023
- 2023COMET: Learning Cardinality Constrained Mixture of Experts with Trees and Local Search5 Jun 2023
- 2023Blockwise Parallel Transformer for Large Context Models30 May 2023
- 2023Lifting the Curse of Capacity Gap in Distilling Language Models20 May 2023
- 2023Task-oriented Memory-efficient Pruning-Adapter26 Mar 2023
- 2023A Stability Analysis of Fine-Tuning a Pre-Trained Model24 Jan 2023
- 2022Sparse Upcycling: Training Mixture-of-Experts from Dense Checkpoints9 Dec 2022
- 2022Multi-CLS BERT: An Efficient Alternative to Traditional Ensembling10 Oct 2022
- 2022Less is More: Task-aware Layer-wise Distillation for Language Model Compression4 Oct 2022
- 2022Sparse Upcycling: Training Mixture-of-Experts from Dense Checkpointscited by 207
- 2024LLaMA-MoE: Building Mixture-of-Experts from LLaMA with Continual Pre-trainingcited by 122
- 2023Language-driven All-in-one Adverse Weather Removalcited by 36
- 2023Lifting the Curse of Capacity Gap in Distilling Language Modelscited by 25
- 2023Exploiting Activation Sparsity with Dense to Dynamic-k Mixture-of-Experts Conversioncited by 15
- 2025Analytical FFN-to-MoE Restructuring via Activation Pattern Analysiscited by 7
- 2024Exploring the Benefit of Activation Sparsity in Pre-trainingcited by 5
- 20253D Gaussian Splatting Data Compression with Mixture of Priorscited by 5
- 2024LaDiMo: Layer-wise Distillation Inspired MoEfiercited by 4
- 2023COMET: Learning Cardinality Constrained Mixture of Experts with Trees and Local Searchcited by 2
- 2024On Parameter Estimation in Deviated Gaussian Mixture of Expertscited by 2
- 2025WINA: Weight Informed Neuron Activation for Accelerating Large Language Model Inferencecited by 2
12 most cited of 15 All 15 (JSON)
- 2026Mixture-of-Experts in Remote Sensing: A Survey3 Apr 2026
- 2026ExFusion: Efficient Transformer Training via Multi-Experts Fusion30 Mar 2026
- 2025WINA: Weight Informed Neuron Activation for Accelerating Large Language Model Inference26 May 2025
- 20253D Gaussian Splatting Data Compression with Mixture of Priors6 May 2025
- 2025Analytical FFN-to-MoE Restructuring via Activation Pattern Analysis6 Feb 2025
- 2024Exploring the Benefit of Activation Sparsity in Pre-training4 Oct 2024
- 2024LaDiMo: Layer-wise Distillation Inspired MoEfier8 Aug 2024
- 2024LLaMA-MoE: Building Mixture-of-Experts from LLaMA with Continual Pre-training24 Jun 2024
- 2024On Parameter Estimation in Deviated Gaussian Mixture of Experts7 Feb 2024
- 2024Contrastive Learning and Mixture of Experts Enables Precise Vector Embeddings28 Jan 2024
- 2023Language-driven All-in-one Adverse Weather Removal3 Dec 2023
- 2023Exploiting Activation Sparsity with Dense to Dynamic-k Mixture-of-Experts Conversion6 Oct 2023
12 newest of 15 All 15 (JSON)
- 2025Mixture of Routerscited by 2
- 2025Mixture of Routers30 Mar 2025
- 2024A Survey on Knowledge Distillation of Large Language Modelscited by 276
- 2022Less is More: Task-aware Layer-wise Distillation for Language Model Compressioncited by 91
- 2024Model Compression and Efficient Inference for Large Language Models: A Surveycited by 62
- 2023Towards the Law of Capacity Gap in Distilling Language Modelscited by 41
- 2022Multi-CLS BERT: An Efficient Alternative to Traditional Ensemblingcited by 14
- 2023Blockwise Parallel Transformer for Large Context Modelscited by 12
- 2024CFSP: An Efficient Structured Pruning Framework for LLMs with Coarse-to-Fine Activation Informationcited by 11
- 2024Flexible and Adaptable Summarization via Expertise Separationcited by 7
- 2023Partition-and-Debias: Agnostic Biases Mitigation via A Mixture of Biases-Specific Expertscited by 3
- 2023Task-oriented Memory-efficient Pruning-Adaptercited by 2
- 2023A Stability Analysis of Fine-Tuning a Pre-Trained Modelcited by 1
- 2023G-SPEED: General SParse Efficient Editing MoDelcited by 0
12 most cited of 18 All 18 (JSON)
- 2026KDFP: A first-principles approach to knowledge distillation in large language models7 Oct 2026
- 2026A Study on Hidden Layer Distillation for Large Language Model Pre-Training12 May 2026
- 2025ExpertSim: Fast Particle Detector Simulation Using Mixture-of-Generative-Experts28 Aug 2025
- 2025NEU-ESC: A Comprehensive Vietnamese dataset for Educational Sentiment analysis and topic Classification toward multitask learning30 Jun 2025
- 2024CFSP: An Efficient Structured Pruning Framework for LLMs with Coarse-to-Fine Activation Information20 Sep 2024
- 2024Exploiting Student Parallelism for Efficient GPU Inference of BERT-like Models in Online Services22 Aug 2024
- 2024Flexible and Adaptable Summarization via Expertise Separation8 Jun 2024
- 2024A Survey on Knowledge Distillation of Large Language Models20 Feb 2024
- 2024Model Compression and Efficient Inference for Large Language Models: A Survey15 Feb 2024
- 2023On Elastic Language Models13 Nov 2023
- 2023Towards the Law of Capacity Gap in Distilling Language Models13 Nov 2023
- 2023G-SPEED: General SParse Efficient Editing MoDel16 Oct 2023
12 newest of 18 All 18 (JSON)
Similar, not cited
Close in wording, no citation either way
- 2025Mixture of Experts in Large Language Models
- 2024A Closer Look into Mixture-of-Experts in Large Language Models
- 2024Diversifying the Expert Knowledge for Task-Agnostic Pruning in Sparse Mixture-of-Experts
- 2022ST-MoE: Designing Stable and Transferable Sparse Expert Models
- 2025Efficiently Editing Mixture-of-Experts Models with Compressed Experts
- 2024A Survey on Mixture of Experts in Large Language Models
- 2025Training Matryoshka Mixture-of-Experts for Elastic Inference-Time Expert Utilization
- 2024MoE-I2: Compressing Mixture of Experts Models through Inter-Expert Pruning and Intra-Expert Low-Rank Decomposition