vix.ing · top · new · best · stats · spec

Outrageously Large Neural Networks: The Sparsely-Gated Mixture-of-Experts Layer

Sparsely-Gated MoE layers increase model capacity by over 1000x with minimal compute loss, improving language modeling and translation.

2017/01/23 by Noam Shazeer, Shazeer, Noam, Azalia Mirhoseini +13 · 11 voices · 732 citations
Computer Science · #Advanced Neural Network Applications #Domain Adaptation and Few-Shot Learning #Topic Modeling #cs.CL #cs.LG #cs.NE #stat.ML

paper · pdf · doi:10.48550/arxiv.1701.06538

openalex publication_date 2017/01/23 · openalex created_date 2025/10/10 · openalex updated_date 2026/07/28

Abstract

The capacity of a neural network to absorb information is limited by its number of parameters. Conditional computation, where parts of the network are active on a per-example basis, has been proposed in theory as a way of dramatically increasing model capacity without a proportional increase in computation. In practice, however, there are significant algorithmic and performance challenges. In this work, we address these challenges and finally realize the promise of conditional computation, achieving greater than 1000x improvements in model capacity with only minor losses in computational efficiency on modern GPU clusters. We introduce a Sparsely-Gated Mixture-of-Experts layer (MoE), consisting of up to thousands of feed-forward sub-networks. A trainable gating network determines a sparse combination of these experts to use for each example. We apply the MoE to the tasks of language modeling and machine translation, where model capacity is critical for absorbing the vast quantities of knowledge available in the training corpora. We present model architectures in which a MoE with up to 137 billion parameters is applied convolutionally between stacked LSTM layers. On large language modeling and machine translation benchmarks, these models achieve significantly better results than state-of-the-art at lower computational cost.

Summary

The paper presents a Sparsely-Gated Mixture-of-Experts (MoE) layer that enables conditional computation, where only a small subset of the network is active for any given input. By addressing GPU hardware constraints and load balancing through specific gating mechanisms and loss functions, the authors scale models up to 137 billion parameters. They demonstrate significant improvements in perplexity and BLEU scores across large-scale language modeling and machine translation tasks.

machine-generated · gemma4:31b

In simple words

Imagine a giant book of answers where each page is written by a different expert. Instead of reading the whole book for every question, a smart guide picks just two or three pages that best fit the question. This lets the book be huge without taking forever to read. The authors made this work on fast computers by making sure the experts share the work evenly.

machine-generated · gemma4:31b

Outline

machine-generated · gemma4:31b

Argument

  1. Increasing model capacity generally improves accuracy but leads to quadratic growth in compute costs.
    Citations of previous work in text, images, and audio domains.
  2. Conditional computation can increase capacity without proportional compute increases by activating only parts of the network per example.
    Theoretical proposals cited in related work.
  3. Practical implementation on GPUs is hindered by branching overhead, shrinking batch sizes, and network bandwidth.
    Analysis of GPU architecture and distributed computing bottlenecks.
  4. A Sparsely-Gated MoE layer with Noisy Top-K gating can effectively route inputs to a small number of experts.
    Proposed architectural design and mathematical formulation of the gating function.
  5. Combining data parallelism for standard layers and model parallelism for experts maintains large batch sizes per expert.
    Proposed system architecture for distributed training.
  6. Adding importance and load balancing losses prevents the 'rich-get-richer' effect where a few experts are over-trained.
    Experimental results in Appendix showing CV of Importance/Load with different loss weights.
  7. Massive capacity increases lead to better performance on very large datasets.
    Empirical results on 1B and 100B word corpora and WMT'14 translation benchmarks.

machine-generated · gemma4:31b

Assumptions

machine-generated · gemma4:31b

Claims

machine-generated · gemma4:31b

Methods

machine-generated · gemma4:31b

Limitations

admitted by authors:

noticed by the model, not admitted:

machine-generated · gemma4:31b

Benchmarks

1 Billion Word Language Modeling Benchmark

methodmetricvalue
Best Published ResultsTest Perplexity (10 epochs)34.7
Low-Budget MoE ModelTest Perplexity (10 epochs)34.1
Medium-Budget MoE ModelTest Perplexity (10 epochs)31.3
High-Budget MoE ModelTest Perplexity (10 epochs)28.0

WMT'14 En→Fr newstest2014

methodmetricvalue
MoE with 2048 ExpertsBLEU40.35
MoE with 2048 Experts (longer training)BLEU40.56
GNMTBLEU39.22
GNMT+RLBLEU39.92

WMT'14 En→De newstest2014

methodmetricvalue
MoE with 2048 ExpertsBLEU26.03
GNMTBLEU24.91
GNMT+RLBLEU24.66

Google Production En→Fr dataset

methodmetricvalue
MoE with 2048 ExpertsTest BLEU36.57
GNMTTest BLEU35.56

Multilingual Machine Translation

methodmetricvalue
GNMT-MultiPerplexity (dev)4.14
MoE-MultiPerplexity (dev)3.35

machine-generated · gemma4:31b

Key equations

machine-generated · gemma4:31b

Proof sketch

  1. The authors propose a Sparsely-Gated Mixture-of-Experts (MoE) layer to increase model capacity without proportional increases in computation.
  2. They introduce a 'Noisy Top-K' gating mechanism that selects only a few experts per input, reducing computational cost.
  3. To prevent expert collapse (where a few experts dominate), they implement auxiliary loss functions based on the coefficient of variation to balance importance and load across experts.
  4. Performance challenges are addressed by combining data and model parallelism to maintain large batch sizes for each expert.
  5. The approach is validated through language modeling and machine translation tasks, demonstrating that increasing capacity via MoE significantly improves results on large datasets.

machine-generated · gemma4:31b

Open questions

machine-generated · gemma4:31b

Cited by

Discussions

Related