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

PuzzleMoE: Efficient Compression of Large Mixture-of-Experts Models via Sparse Expert Merging and Bit-packed inference

About

Mixture-of-Experts (MoE) models have shown strong potential in scaling language models efficiently by activating only a small subset of experts per input. However, their widespread deployment remains limited due to the high memory overhead associated with storing all expert parameters, particularly as the number of experts increases. To address this challenge, prior works have explored expert dropping and merging strategies, yet they often suffer from performance drop at high compression ratios. In this paper, we introduce PuzzleMoE, a training-free MoE compression method that achieves both high accuracy and efficient inference through two key innovations: First, PuzzleMoE performs sparse expert merging by identifying element-wise weight redundancy and specialization. It uses a dual-mask to capture both shared and expert-specific parameters. Second, to avoid the overhead of storing binary masks and signs, PuzzleMoE introduces a bit-packed encoding scheme that reuses underutilized exponent bits, enabling efficient MoE inference on GPUs. Extensive experiments demonstrate that PuzzleMoE can compress MoE models by up to 50% while maintaining accuracy across various tasks. Specifically, it outperforms prior MoE compression methods by up to 16.7% on MMLU at 50% compression ratio, and achieves up to 1.28\times inference speedup.

Yushu Zhao, Zheng Wang, Minjia Zhang• 2025

Related benchmarks

TaskDatasetResultRank
Commonsense ReasoningWinoGrande--
1581
Code GenerationHumanEval (test)
Pass@145.1
701
Question AnsweringPIQA
Accuracy79.7
589
Sentence CompletionHellaSwag
Accuracy62.36
440
Commonsense ReasoningPIQA
Accuracy82.1
400
Common Sense ReasoningBoolQ
Accuracy88.2
280
Language ModelingWikiText-2
Perplexity4.1
205
commonsense inferenceHellaSwag
Accuracy58.3
171
Common Sense ReasoningARC Easy
ARC (easy) Accuracy83.2
149
Commonsense ReasoningWinoGrande--
94
Showing 10 of 29 rows

Other info

Follow for update