about
Google breaks the trillion-parameter ceiling with the Switch Transformer (arxiv.org)
48 points by groar on Jan 18, 2021 | hide | past | pdf | 4 comments on HN

In plain words: Instead of using every parameter for each input, the model sends each input to just one small group of parameters, holding far more knowledge at the same computing cost. It pre-trained up to 7 times faster than the usual all-parameters approach.

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

In deep learning, models typically reuse the same parameters for all inputs. Mixture of Experts (MoE) defies this and instead selects different parameters for each incoming example. The result is a sparsely-activated model -- with outrageous numbers of parameters -- but a constant computational cost. However, despite several notable successes of MoE, widespread adoption has been hindered by complexity, communication costs and training instability -- we address these with the Switch Transformer. We simplify the MoE routing algorithm and design intuitive improved models with reduced communication and computational costs. Our proposed training techniques help wrangle the instabilities and we show large sparse models may be trained, for the first time, with lower precision (bfloat16) formats. We design models based off T5-Base and T5-Large to obtain up to 7x increases in pre-training speed with the same computational resources. These improvements extend into multilingual settings where we measure gains over the mT5-Base version across all 101 languages. Finally, we advance the current scale of language models by pre-training up to trillion parameter models on the "Colossal Clean Crawled Corpus" and achieve a 4x speedup over the T5-XXL model.

William Fedus, Barret Zoph, Noam Shazeer
arXiv:2101.03961 · cs.LG, cs.AI · submitted Jan 11, 2021 · updated Jun 16, 2022
abstract · pdf · html · JMLR

add comment on HN
Also discussed: Jan 2021 (2 points, 0 comments) · Jan 2021 (1 point, 0 comments) · Jan 2021 (1 point, 0 comments) · Jan 2021 (3 points, 0 comments)

It’s an interesting thought experiment to consider models that have more parameters than there are data points being analyzed.

What does that mean?

It basically means you need to learn about double descent:

https://twitter.com/hippopedoid/status/1243229024085835779

and regularize your models well, implicitly or explicitly. Your model might otherwise memorize the data instead of learning features.

All networks are like this. In a network of N nodes you have N^2 potential relationships. That’s just simple paired relationships. You can still go to higher order groups.
This is impressive but also requires a lot of power to train.

If this trend continues ML will soon surpass bitcoin as the worst polluter.