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

EPnG: Adaptive Expert Prune-and-Grow for Parameter-Efficient MoE Fine-tuning

About

Mixture-of-Experts (MoE) models scale efficiently but remain costly to adapt due to redundant experts and uniform parameter allocation. Existing parameter-efficient fine-tuning (PEFT) methods such as LoRA ignore MoE routing dynamics, leading to suboptimal resource use. We propose EPnG, an adaptive prune-and-grow framework that reallocates LoRA capacity based on expert importance derived from router gate probabilities. EPnG prunes under-utilized experts and expands high-importance experts via rank growth with orthogonal initialization, while maintaining a fixed parameter budget. Across OLMoE and Qwen1.5-MoE, EPnG consistently outperforms LoRA under the same budget and achieves performance comparable to full fine-tuning while updating only 0.55%-0.72% of parameters (up to 140x-180x fewer). These results demonstrate that aligning PEFT with MoE routing yields a more effective and scalable fine-tuning strategy.

Ahin Lee, Sehyun Yun, Taesik Gong• 2026

Related benchmarks

TaskDatasetResultRank
Code GenerationHumanEval (test)--
701
Code GenerationMBPP (test)--
411
Mathematical ReasoningGSM8K (test)
GSM8K EM66.26
12
PersonalizationPrefEval (test)
Accuracy38.33
4
Mathematical ReasoningMATH (test)
Exact Match (EM)21.4
4
Mathematical ReasoningGSM8K
Exact Match (EM)64.22
3
Mathematical ReasoningMATH
Exact Match (EM)37.32
3
Code GenerationHumanEval
pass@1074.39
3
Code GenerationMBPP
Pass@1060
3
Showing 9 of 9 rows

Other info

Follow for update