MoNE: Replacing Redundant Experts with Lightweight Novices for Structured Pruning of MoE

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Zhang, Geng, Han, Yuxuan, Lou, Yuxuan, Zhang, Yiqi, Zhao, Wangbo, You, Yang
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866918348829229056
author Zhang, Geng
Han, Yuxuan
Lou, Yuxuan
Zhang, Yiqi
Zhao, Wangbo
You, Yang
author_facet Zhang, Geng
Han, Yuxuan
Lou, Yuxuan
Zhang, Yiqi
Zhao, Wangbo
You, Yang
contents Mixture-of-Experts (MoE) enables efficient scaling of large language models by activating only a subset of experts per input token. However, deploying MoE-based models incurs significant memory overhead due to the need to retain all experts in memory. While structured pruning is promising to reduce memory costs, existing methods often show suboptimal performance and unstable degradation in three dimensions: model architectures, calibration data sources, and calibration sample sizes. This paper proposes Mixture-of-Novices-and-Experts (MoNE), a novel expert pruning method that replaces redundant experts with lightweight novices to achieve effective and robust model compression. MoNE evaluates expert redundancy based on two metrics: access frequency and output variance. Experts exhibiting low usage and stable outputs are pruned and replaced with lightweight novices-unbiased estimations of their original outputs-minimizing performance degradation. Extensive experiments demonstrate that MoNE consistently outperforms baseline methods with minimal accuracy degradation across the three dimensions, confirming its effectiveness and robustness. Notably, it outperforms baselines by up to 2.72 for the average zero shot accuracy across nine downstream tasks under 25% pruning ratio, with only 0.14 performance drop for Qwen2-57B-A14B. The code is available at https://github.com/zxgx/mode-pd.
format Preprint
id arxiv_https___arxiv_org_abs_2507_00390
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle MoNE: Replacing Redundant Experts with Lightweight Novices for Structured Pruning of MoE
Zhang, Geng
Han, Yuxuan
Lou, Yuxuan
Zhang, Yiqi
Zhao, Wangbo
You, Yang
Machine Learning
Mixture-of-Experts (MoE) enables efficient scaling of large language models by activating only a subset of experts per input token. However, deploying MoE-based models incurs significant memory overhead due to the need to retain all experts in memory. While structured pruning is promising to reduce memory costs, existing methods often show suboptimal performance and unstable degradation in three dimensions: model architectures, calibration data sources, and calibration sample sizes. This paper proposes Mixture-of-Novices-and-Experts (MoNE), a novel expert pruning method that replaces redundant experts with lightweight novices to achieve effective and robust model compression. MoNE evaluates expert redundancy based on two metrics: access frequency and output variance. Experts exhibiting low usage and stable outputs are pruned and replaced with lightweight novices-unbiased estimations of their original outputs-minimizing performance degradation. Extensive experiments demonstrate that MoNE consistently outperforms baseline methods with minimal accuracy degradation across the three dimensions, confirming its effectiveness and robustness. Notably, it outperforms baselines by up to 2.72 for the average zero shot accuracy across nine downstream tasks under 25% pruning ratio, with only 0.14 performance drop for Qwen2-57B-A14B. The code is available at https://github.com/zxgx/mode-pd.
title MoNE: Replacing Redundant Experts with Lightweight Novices for Structured Pruning of MoE
topic Machine Learning
url https://arxiv.org/abs/2507.00390