AI RESEARCH

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

arXiv CS.AI

ArXi:2605.09403v1 Announce Type: cross 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.