atlas

Research map / Architectures and efficiency / Mixture of experts

Unchosen Experts Can Contribute Too: Unleashing MoE Models’ Power by Self-Contrast

NeurIPS 2024 · Architectures and efficiency › Mixture of experts · approach: Expert routing and specialization · 2 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes SCMoE, a training-free decoding method that contrasts top-2 (strong) and rank-k (weak) expert activations within the same MoE model to improve reasoning and code generation.

Problem

Unchosen experts in MoE models do not contribute to the output at inference, and increasing the number of activated experts does not reliably improve quality, leaving model capacity underutilized.

Approach

Use the MoE model's own strong activation (default top-2 routing) and weak activation (rank-k routing) and contrast their next-token logits, so unchosen experts contribute to the prediction without training or extra models.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Apply SCMoE self-contrast decodingMixtral 8x7B · GSM8Kaccuracy: 61.79 to 66.94 (+5.15)Helped
Apply SCMoE self-contrast decodingMixtral 8x7B · StrategyQAaccuracy: 72.83 to 76.29 (+3.46)Helped
Apply SCMoE self-contrast decodingMixtral 8x7B · MBPPpass@1 accuracy: 46.20 to 48.80 (+2.60)Helped
Apply SCMoE self-contrast decodingMixtral 8x7B · HumanEvalpass@1 accuracy: 33.54 to 41.46 (+7.92)Helped
Apply SCMoE self-contrast decodingDeepSeekMoE-16B · GSM8Kaccuracy: 18.95 to 20.77Helped
Apply SCMoE self-contrast decodingDeepSeekMoE-16B · StrategyQAaccuracy: 60.41 to 62.99Helped
Apply SCMoE self-contrast decodingDeepSeekMoE-16B · MBPPpass@1 accuracy: 35.20 to 37.20Helped
Apply SCMoE self-contrast decodingDeepSeekMoE-16B · HumanEvalpass@1 accuracy: 26.83 to 28.05Helped
Replace top-2 strong activation with task-specific top-k activation (ablation)SCMoE on Mixtral 8x7B · GSM8Kaccuracy: 66.94 to 68.92 (+1.98)Helped
Replace top-2 strong activation with task-specific top-k activation (ablation)SCMoE on Mixtral 8x7B · StrategyQAaccuracy: 76.29 to 76.42 (+0.13)Helped
Replace top-2 strong activation with task-specific top-k activation (ablation)SCMoE on Mixtral 8x7B · MBPPpass@1 accuracy: 48.80 to 50.60 (+1.80)Helped
Combine SCMoE with self-consistency (ablation)Mixtral 8x7B · GSM8K major@20accuracy: 75.59 to 78.31 (+2.72)Helped

Surprising findings

Limitations

Experimental setup

Datasets: GSM8K StrategyQA MBPP HumanEval

Benchmarks: GSM8K StrategyQA MBPP HumanEval

Metrics: accuracy pass@1 major@20 latency ratio KL divergence

Baselines: Greedy Dynamic Routing Ensemble Routing Contrastive Search Contrastive Decoding DoLa

Builds on: contrastive decoding DoLa top-k routing self-consistency

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy