This episode examines "Alternating Updates for Efficient Transformers" (Baykal et al., Google Research and UC Berkeley, NeurIPS 2023), a technique that reports up to an 87% inference speedup on SQuAD at matched accuracy without changing attention mechanisms or reducing parameter count. The discussion contrasts this approach, called AltUp, with Mixture-of-Experts methods like Switch Transformer and GShard, clarifying that while MoE scales total parameter count and activates only a subset per token, AltUp instead widens the token embedding itself while keeping each transformer layer's compute cost fixed. The hosts unpack why widening representations normally taxes every layer quadratically through attention, and how AltUp sidesteps this by running only one slice of the widened vector through full computation per layer while cheaply predicting and correcting the rest. A key thread running through the conversation questions whether the paper's framing of AltUp as "conditional computation" alongside MoE is fully justified, teeing up a deeper look at whether its slice-selection mechanism is truly input-dependent like MoE routing or something simpler. Listeners interested in inference efficiency, transformer architecture tradeoffs, or the shift from parameter-count to FLOP-based cost thinking will find the mechanics of the predict-compute-correct cycle a genuinely novel angle on model scaling.
Sources:
1. Alternating Updates for Efficient Transformers — Cenk Baykal, Dylan Cutler, Nishanth Dikkala, Nikhil Ghosh, Rina Panigrahy, Xin Wang, 2023
http://arxiv.org/abs/2301.133102. Outrageously Large Neural Networks: The Sparsely-Gated Mixture-of-Experts Layer — Noam Shazeer, Azalia Mirhoseini, Krzysztof Maziarz, Andy Davis, Quoc Le, Geoffrey Hinton, Jeff Dean, 2017
https://scholar.google.com/scholar?q=Outrageously+Large+Neural+Networks%3A+The+Sparsely-Gated+Mixture-of-Experts+Layer3. GShard: Scaling Giant Models with Conditional Computation and Automatic Sharding — Dmitry Lepikhin, HyoukJoong Lee, Yuanzhong Xu, Dehao Chen, Orhan Firat, Yanping Huang, Maxim Krikun, Noam Shazeer, Zhifeng Chen, 2020
https://scholar.google.com/scholar?q=GShard%3A+Scaling+Giant+Models+with+Conditional+Computation+and+Automatic+Sharding4. Switch Transformers: Scaling to Trillion Parameter Models with Simple and Efficient Sparsity — William Fedus, Barret Zoph, Noam Shazeer, 2021
https://scholar.google.com/scholar?q=Switch+Transformers%3A+Scaling+to+Trillion+Parameter+Models+with+Simple+and+Efficient+Sparsity5. Mixture-of-Depths: Dynamically Allocating Compute in Transformer-Based Language Models — David Raposo, Sam Ritter, Blake Richards, Timothy Lillicrap, Peter Conway Humphreys, Adam Santoro, 2024
https://scholar.google.com/scholar?q=Mixture-of-Depths%3A+Dynamically+Allocating+Compute+in+Transformer-Based+Language+Models6. Reducing Activation Recomputation in Large Transformer Models — Vijay Korthikanti, Jared Casper, Sangkug Lym, et al., 2023
https://scholar.google.com/scholar?q=Reducing+Activation+Recomputation+in+Large+Transformer+Models7. Training Compute-Optimal Large Language Models (Chinchilla) — Jordan Hoffmann, Sebastian Borgeaud, Arthur Mensch, et al., 2022
https://scholar.google.com/scholar?q=Training+Compute-Optimal+Large+Language+Models+%28Chinchilla%298. Deja Vu: Contextual Sparsity for Efficient LLMs at Inference Time — Zichang Liu, Jue Wang, Tri Dao, et al., 2023
https://scholar.google.com/scholar?q=Deja+Vu%3A+Contextual+Sparsity+for+Efficient+LLMs+at+Inference+Time