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

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.

Hongcheng Guo, Juntao Yao, Boyang Wang, Junjia Du, Shaosheng Cao, Donglin Di, Shun Zhang, Zhoujun Li• 2025

Related benchmarks

TaskDatasetResultRank
Commonsense ReasoningWinoGrande--
1581
Question AnsweringPIQA
Accuracy78.92
589
Sentence CompletionHellaSwag
Accuracy61.81
440
Math ReasoningGSM8K
Accuracy (GSM8K)39.4
190
commonsense inferenceHellaSwag
Accuracy63.06
171
Commonsense ReasoningWinoGrande--
94
Code ReasoningHumanEval
HumanEval Score32.9
70
Boolean Question AnsweringBoolQ
Accuracy77.12
56
Question AnsweringARC Challenge
Accuracy0.4
56
Multi-task Question AnsweringMMLU
Accuracy56.68
46
Showing 10 of 15 rows

Other info

Follow for update