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.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Code Generation | HumanEval (test) | -- | 701 | |
| Code Generation | MBPP (test) | -- | 411 | |
| Mathematical Reasoning | GSM8K (test) | GSM8K EM66.26 | 12 | |
| Personalization | PrefEval (test) | Accuracy38.33 | 4 | |
| Mathematical Reasoning | MATH (test) | Exact Match (EM)21.4 | 4 | |
| Mathematical Reasoning | GSM8K | Exact Match (EM)64.22 | 3 | |
| Mathematical Reasoning | MATH | Exact Match (EM)37.32 | 3 | |
| Code Generation | HumanEval | pass@1074.39 | 3 | |
| Code Generation | MBPP | Pass@1060 | 3 |