mlsanityData to 7 Oct 2026

Sparsity Moves Computation: How FFN Architecture Reshapes Attention in Small Transformers

arXiv 2605.09403 ↗First posted 10 May 2026cs.LG cs.AI cs.NEcited by 0

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

  1. 2025Decoding Knowledge Attribution in Mixture-of-Experts: A Framework of Basic-Refinement Collaboration and Efficiency Analysis30 May
  2. 2023Sparse MoE as the New Dropout: Scaling Dense and Self-Slimmable Transformers2 Mar
  3. 2021Taming Sparsely Activated Transformer with Stochastic Experts8 Oct
  4. 2021MoEfication: Transformer Feed-forward Layers are Mixtures of Experts5 Oct
  5. 2021Switch Transformers: Scaling to Trillion Parameter Models with Simple and Efficient Sparsity11 Jan

Other references · 18

  1. 2016Gaussian Error Linear Units (GELUs)cited by 3,431
  2. 2017Outrageously Large Neural Networks: The Sparsely-Gated Mixture-of-Experts Layercited by 3,413
  3. 2022Locating and Editing Factual Associations in GPTcited by 2,609
  4. 2020GLU Variants Improve Transformercited by 1,724
  5. 2017Sigmoid-Weighted Linear Units for Neural Network Function Approximation in Reinforcement Learningcited by 935
  6. 2022Toy Models of Superpositioncited by 928
  7. 2024DeepSeekMoE: Towards Ultimate Expert Specialization in Mixture-of-Experts Language Modelscited by 834
  8. 2023Progress measures for grokking via mechanistic interpretabilitycited by 822
  9. 2016Language Modeling with Gated Convolutional Networkscited by 688
  10. 2016Using the Output Embedding to Improve Language Modelscited by 266
  11. 2023The Clock and the Pizza: Two Stories in Mechanistic Explanation of Neural Networkscited by 153
  12. 2016Tying Word Vectors and Word Classifiers: A Loss Framework for Language Modelingcited by 87
6 more
  1. 2023Understanding Addition in Transformerscited by 34
  2. 2024Bilinear MLPs enable weight-based mechanistic interpretabilitycited by 28
  3. 2025Is Random Attention Sufficient for Sequence Modeling? Disentangling Trainable Components in the Transformercited by 12
  4. 2023Counting and Algorithmic Generalization with Transformerscited by 11
  5. 2024Counting in Small Transformers: The Delicate Interplay between Attention and Feed-Forward Layerscited by 5
  6. 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

  1. 2026Routing Absorption in Sparse Attention: Why Random Gates Are Hard to Beat
  2. 2026Mixture of Chapters: Scaling Learnt Memory in Transformers
  3. 2026Sparse Token Routing in Efficient Transformers
  4. 2026Adaptive Computation Depth via Learned Token Routing in Transformers
  5. 2026Sparse Mixture-of-Experts Routing in Visual Diffusion Transformers:Diagnosis, Boundary Calibration and Evolutionary Roadmap from Routing Collapse to Selective Deadlock
  6. 2025Learning to Skip the Middle Layers of Transformers
  7. 2026The Sparsity Ceiling: Where Spiking Networks Can and Cannot Trade Activity for Energy
  8. 2026Sparsely gated tiny linear experts