atlas

Research map / Architectures and efficiency / Mixture of experts

MoE++: Accelerating Mixture-of-Experts Methods with Zero-Computation Experts

ICLR 2025 · Architectures and efficiency › Mixture of experts · approach: Expert routing and specialization · 0 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 MoE++, a heterogeneous MoE that adds zero, copy, and constant "zero-computation" experts alongside FFN experts plus gating residuals, trained from scratch on public data at 0.6B to 7B scale.

Problem

Vanilla MoE activates a fixed number of FFN experts for every token regardless of difficulty, causing training and inference inefficiency and sub-optimal performance, and expert communication/load imbalance across GPUs.

Approach

Add three near-parameter-free zero-computation experts (zero: discard; copy: skip via identity; constant: replace with a trainable vector mixed with input) so simple tokens use fewer FFN experts, and add gating residuals that carry router scores from the previous layer, plus a heterogeneous load-balance loss and capacity allocation.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
replace vanilla MoE with MoE++ using zero-computation experts and tau 0.757B/16E MoE · PIQAaccuracy: 72.6 -> 73.6Helped
replace vanilla MoE with MoE++ using zero-computation experts and tau 0.757B/16E MoE · WinoGrandeaccuracy: 58.8 -> 58.2Hurt
replace vanilla MoE with MoE++ using zero-computation experts and tau 0.757B/16E MoE · 7B model, 100B tokensthroughput increase: 27.8%Helped
add all three zero-computation expert types (zero, copy, and constant) (ablation)MoE++ 1B/(16+4)E · PIQAaccuracy: 68.4 -> 70.3Helped
add a constant expert (ablation)MoE++ 1B/(16+4)E · Averageaccuracy: 46.6 -> 47.4Helped
add gating residuals (ablation)MoE++ 1B/(16+4)E · Averageaccuracy: 47.5 -> 47.7Helped
lower tau from 0.75 to 0.10 (ablation)MoE++ 0.6B/(8+4)E · 0.6B model, 100B tokensthroughput increase: 25.2% -> 164.5%Helped
lower tau from 0.75 to 0.10 (ablation)MoE++ 0.6B/(8+4)E · Averageaccuracy: 45.6 -> 43.4Hurt

Surprising findings

Limitations

Experimental setup

Datasets: RedPajama Dolma Pile

Benchmarks: SciQ PIQA WinoGrande ARC-E ARC-C HellaSwag LogiQA BoolQ LAMBADA NQ MMLU

Metrics: accuracy exact match expert forward throughput average accuracy

Baselines: Vanilla MoE (0.6B/8E, 1B/16E, 2B/32E, 7B/16E) OpenMoE-8B/32E LLaMA2-7B OPT Pythia TinyLlama-1.1B Open-LLaMA-3B Sheared-LLaMA LLaMA-MoE-3.0B

Builds on: Switch Transformer GShard Expert Choice DeepSeekMoE Megatron

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