Cluster-Driven Expert Pruning for Mixture-of-Experts Large Language Models
About
Mixture-of-Experts (MoE) architectures have emerged as a promising paradigm for scaling large language models (LLMs) with sparse activation of task-specific experts. Despite their computational efficiency during inference, the massive overall parameter footprint of MoE models (e.g., GPT-4) introduces critical challenges for practical deployment. Current pruning approaches often fail to address two inherent characteristics of MoE systems: 1).intra-layer expert homogeneity where experts within the same MoE layer exhibit functional redundancy, and 2). inter-layer similarity patterns where deeper layers tend to contain progressively more homogeneous experts. To tackle these issues, we propose Cluster-driven Expert Pruning (C-Prune), a novel two-stage framework for adaptive task-specific compression of MoE LLMs. C-Prune operates through layer-wise expert clustering, which groups functionally similar experts within each MoE layer using parameter similarity metrics, followed by global cluster pruning, which eliminates redundant clusters across all layers through a unified importance scoring mechanism that accounts for cross-layer homogeneity. We validate C-Prune through extensive experiments on multiple MoE models and benchmarks. The results demonstrate that C-Prune effectively reduces model size while outperforming existing MoE pruning methods.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Commonsense Reasoning | WinoGrande | -- | 1581 | |
| Question Answering | PIQA | Accuracy78.92 | 589 | |
| Sentence Completion | HellaSwag | Accuracy61.81 | 440 | |
| Math Reasoning | GSM8K | Accuracy (GSM8K)39.4 | 190 | |
| commonsense inference | HellaSwag | Accuracy63.06 | 171 | |
| Commonsense Reasoning | WinoGrande | -- | 94 | |
| Code Reasoning | HumanEval | HumanEval Score32.9 | 70 | |
| Boolean Question Answering | BoolQ | Accuracy77.12 | 56 | |
| Question Answering | ARC Challenge | Accuracy0.4 | 56 | |
| Multi-task Question Answering | MMLU | Accuracy56.68 | 46 |