Papers · Switch Transformers
AI PAPER

Switch Transformers: Scaling to Trillion Parameter Models with Simple and Efficient Sparsity

Simplified mixture-of-experts routing to scale Transformers to a trillion parameters while keeping computation costs manageable by activating only a fraction of the model for each input.

Authors
William Fedus, Barret Zoph, Noam Shazeer
Institution
Google Brain
Published
January NaN, 2021
Citations
4,000
Category
Architecture
Impact
major
PAPER EXPLAINED

Plain-English Summary

Normal neural networks use every parameter for every input. A 100-billion parameter model does 100 billion operations for each token. Mixture-of-experts (MoE) models break this pattern by dividing the model into many “expert” sub-networks and routing each input to only a few of them. This means you can have a model with a trillion parameters but only use a fraction of them for any given input, keeping computation manageable.

Switch Transformers simplified previous MoE approaches by routing each token to just one expert (instead of multiple), making the system simpler and more efficient. The result was a model with 1.6 trillion parameters that trained 7x faster than a comparable dense model while achieving the same quality.

Think of it like a hospital with many specialist doctors. Each patient sees only the relevant specialist, not every doctor. The hospital has enormous total expertise, but each visit is efficient because only the right expert is consulted.

Key Innovation

The key simplification was the “switch” routing: each token is sent to exactly one expert based on a learned routing function, rather than the weighted combination of multiple experts used in prior work. This reduced communication costs and complexity while maintaining quality. The paper also introduced techniques for stable training of sparse models, which had previously been unreliable.

The architecture allows scaling model capacity (total parameters) independently of computational cost (activated parameters per input). This decoupling is powerful because it means you can build much larger models without proportionally increasing inference cost.

Impact on the Field

Switch Transformers revived interest in sparse models after years of dense model dominance. The paper demonstrated that the MoE approach could be made practical and reliable, opening the path for models that are much larger in total parameters than what dense architectures could achieve with the same compute budget.

The architecture became particularly influential for inference efficiency. Since only a fraction of parameters are active for each input, MoE models can be faster and cheaper to run than dense models of equivalent quality.

Models That Built on This

GPT-4 is widely reported to use a mixture-of-experts architecture. Mixtral (Mistral) openly adopted MoE, demonstrating that an open-source MoE model could rival much larger dense models. Google’s Gemini models use MoE techniques. DeepSeek’s models use fine-grained expert routing inspired by this work. The MoE approach has become the dominant architecture for frontier models seeking to maximize capability while managing compute costs.