Sparsity Moves Computation: How FFN Architecture Reshapes Attention in Small Transformers
On the map: Mixture-of-experts → Expert routing strategies
Abstract (arXiv)
Architectural choices inside the Transformer feedforward network (FFN) block do not merely affect the block itself; they reshape the computations learned by the rest of the model. We study this effect in one-layer Transformers trained on digit addition with carry, modular arithmetic, and histogram counting. Comparing dense FFNs, gated linear units (GLUs), mixture-of-experts (MoE), and MoE-GLUs, we find that sparse MoE routing can shift computation from FFN to attention, with the strongest ablation-visible effect on carry-based addition. We decompose this redistribution into reduced per-token FFN capacity and sparse partitioning across experts. Critically, frozen random routing nearly matches learned routing, suggesting that redistribution is driven largely by architectural sparsity rather than router-learned specialization. As a secondary finding, GLU-style multiplicative gating rotates task-relevant Fourier structure out of the per-neuron basis and into distributed subspaces, making neuron-level interpretability less informative while preserving structured computation. We validate these conclusions with random-routing, narrow-FFN, and top-2 MoE controls, plus parameter-matching, activation-function, and width-scaling analyses. Together, these results show that local FFN design choices can have nonlocal consequences for Transformer computation.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in Expert routing strategies, one of 12 directions in Mixture-of-experts. Builds on 23 of its 27 references that are in the map. First posted on arXiv 10 May 2026.
Builds on
23 of 27 references.
Closest ancestors · 5
Same direction, newest first
- 2025Decoding Knowledge Attribution in Mixture-of-Experts: A Framework of Basic-Refinement Collaboration and Efficiency Analysis30 May
- 2023Sparse MoE as the New Dropout: Scaling Dense and Self-Slimmable Transformers2 Mar
- 2021Taming Sparsely Activated Transformer with Stochastic Experts8 Oct
- 2021MoEfication: Transformer Feed-forward Layers are Mixtures of Experts5 Oct
- 2021Switch Transformers: Scaling to Trillion Parameter Models with Simple and Efficient Sparsity11 Jan
Other references · 18
- 2016Gaussian Error Linear Units (GELUs)cited by 3,431
- 2017Outrageously Large Neural Networks: The Sparsely-Gated Mixture-of-Experts Layercited by 3,413
- 2022Locating and Editing Factual Associations in GPTcited by 2,609
- 2020GLU Variants Improve Transformercited by 1,724
- 2017Sigmoid-Weighted Linear Units for Neural Network Function Approximation in Reinforcement Learningcited by 935
- 2022Toy Models of Superpositioncited by 928
- 2024DeepSeekMoE: Towards Ultimate Expert Specialization in Mixture-of-Experts Language Modelscited by 834
- 2023Progress measures for grokking via mechanistic interpretabilitycited by 822
- 2016Language Modeling with Gated Convolutional Networkscited by 688
- 2016Using the Output Embedding to Improve Language Modelscited by 266
- 2023The Clock and the Pizza: Two Stories in Mechanistic Explanation of Neural Networkscited by 153
- 2016Tying Word Vectors and Word Classifiers: A Loss Framework for Language Modelingcited by 87
6 more
- 2023Understanding Addition in Transformerscited by 34
- 2024Bilinear MLPs enable weight-based mechanistic interpretabilitycited by 28
- 2025Is Random Attention Sufficient for Sequence Modeling? Disentangling Trainable Components in the Transformercited by 12
- 2023Counting and Algorithmic Generalization with Transformerscited by 11
- 2024Counting in Small Transformers: The Delicate Interplay between Attention and Feed-Forward Layerscited by 5
- 2026Deconstructing Pre-training: Knowledge Attribution Analysis in MoE and Dense Modelscited by 4
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026Routing Absorption in Sparse Attention: Why Random Gates Are Hard to Beat
- 2026Mixture of Chapters: Scaling Learnt Memory in Transformers
- 2026Sparse Token Routing in Efficient Transformers
- 2026Adaptive Computation Depth via Learned Token Routing in Transformers
- 2026Sparse Mixture-of-Experts Routing in Visual Diffusion Transformers:Diagnosis, Boundary Calibration and Evolutionary Roadmap from Routing Collapse to Selective Deadlock
- 2025Learning to Skip the Middle Layers of Transformers
- 2026The Sparsity Ceiling: Where Spiking Networks Can and Cannot Trade Activity for Energy
- 2026Sparsely gated tiny linear experts