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

Preserving Long-Tailed Expert Information in Mixture-of-Experts Tuning

About

Despite MoE models leading many benchmarks, supervised fine-tuning (SFT) for the MoE architectures remains difficult because its router layers are fragile. Methods such as DenseMixer and ESFT mitigate router collapse with dense mixing or auxiliary load-balancing losses, but these introduce noisy gradients that often degrade performance. In preliminary experiments, we systematically pruned experts and observed that while certain super experts are activated far more frequently, discarding less used experts still leads to notable performance degradation. This suggests that even rarely activated experts encode non-trivial knowledge useful for downstream tasks. Motivated by this, we propose an auxiliary-loss-free MoE SFT framework that combines bias-driven sparsification with always-active gated condenser experts. Rather than enforcing balanced activation across all experts, our method encourages task-relevant experts to remain active while pushing long-tailed experts toward inactivity. The condenser experts provide a persistent, learnable pathway that alleviates gradient starvation and facilitates consolidation of information that would otherwise remain fragmented across sparsely activated experts. Analysis further suggest that this design better preserves long-tailed expert information under sparse routing. Experiments on large-scale MoE models demonstrate that our approach outperforms state-of-the-art SFT baselines such as DenseMixer and ESFT, achieving average gain of 2.5%+ on both mathematical reasoning and commonsenseQA benchmarks.

Haoze He, Xingyuan Ding, Xuan Jiang, Xinkai Zou, Alex Cheng, Yibo Zhao, Juncheng Billy Li, Heather Miller• 2026

Related benchmarks

TaskDatasetResultRank
Mathematical ReasoningMAWPS
Accuracy91.6
241
Math ReasoningAQUA
Accuracy38.6
188
Math ReasoningMATH 500
Pass@496.8
39
Mathematical ReasoningGSM8K v1.1 (test)
Accuracy81.7
28
Commonsense ReasoningCommonsense Reasoning
BoolQ Accuracy72.1
27
Math ReasoningGPQA Diamond
Pass@465.8
12
Math ReasoningAIME 2024
Pass@468.3
12
Math ReasoningAIME 2025
Pass@450
12
Mathematical ReasoningADDSUB
Accuracy85.6
7
Mathematical ReasoningSINGLEEQ
Accuracy (SingleEq)93.2
7
Showing 10 of 11 rows

Other info

Follow for update