English

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

Machine Learning 2026-05-18 v2 Artificial Intelligence Neural and Evolutionary Computing

Abstract

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.

Cite

@article{arxiv.2605.09403,
  title  = {Sparsity Moves Computation: How FFN Architecture Reshapes Attention in Small Transformers},
  author = {Gabriel Smithline and Chris Mascioli},
  journal= {arXiv preprint arXiv:2605.09403},
  year   = {2026}
}

Comments

Preprint

R2 v1 2026-07-01T13:01:26.102Z