Switch Transformers: Scaling to Trillion Parameter Models with Simple and Efficient Sparsity
William Fedus, Barret Zoph, Noam Shazeer2022JMLR 2022
Read it on arxiv.org(opens in a new tab)Why this one
Read this after Outrageously Large Neural Networks, because Switch is the version of mixture of experts that made the idea feel buildable instead of exotic. The core move is brutal simplification: route each token to one expert, not several, then spend the saved complexity on making the system train and run. People often talk about MoE as if it is free capacity, but the paper is really about the tax: routing collapse, load balancing, communication between devices, precision trouble, and batches that stop looking like normal dense training. If you are used to thinking of a model as one fixed pile of weights, this changes the accounting. A trillion parameters can be mostly idle on any given token, which is why parameter count no longer maps cleanly to inference or training cost. Worth your evening because modern frontier model specs are unreadable without this distinction.
What to take away
- Switch routes each token to a single expert, trading some modeling richness for simpler, faster sparse computation.
- The auxiliary load-balancing loss is not decoration, it keeps routers from sending every token to the same experts.
- MoE scale changes capacity math: total parameters can rise much faster than per-token FLOPs.
Reads with
- Outrageously Large Neural Networks: The Sparsely-Gated Mixture-of-Experts Layer
switch is the simplified routing scheme that made sparse experts practical
- Training Compute-Optimal Large Language Models
dense scaling laws are the baseline that sparse active parameters complicate
- DeepSeek-V3 Technical Report
modern frontier moe engineering inherits switch's capacity and routing tradeoffs