mlsanityData to 14 Oct 2026

How Linear Is a Transformer Feed-Forward Block? Per-Block Linear Recoverability Is Learned, Not Architectural

arXiv 2606.19379 ↗First posted 12 Jun 2026cs.LG cs.AI cs.CLcited by 0

Not on the map. Why?

Abstract (arXiv)

Transformer feed-forward networks (FFNs) are often treated as nonlinear stores of computation, yet how nonlinear a trained FFN block actually is has rarely been measured. We treat each FFN as a position-wise input-to-output map and split it into the exact least-squares linear approximation plus a residual. The held-out variance the closed-form linear map explains defines a block's linear recoverability (R^2_lin), an optimiser-free measure of its linearity. Across all twelve blocks of GPT-2, Pythia-160m, and llama-160m, R^2_lin is highly heterogeneous and non-monotone with depth, ranging from near-linear (>0.99) to strongly nonlinear (<0.3) between adjacent blocks, and is not set by the activation function: same-width GELU models GPT-2 and Pythia-160m have sharply different profiles, so recoverability is a learned property of individual trained blocks, not an architectural one. A low-rank bilinear probe of the residual recovers only a few points of R^2, with gain uncorrelated with residual nonlinearity: the unrecovered computation is not a single position-wise product but higher-order or distributed structure. The measurement also serves as a targeted compression signal: recoverable blocks admit large single-layer replacements (GPT-2's early FFN at 8x fewer parameters for +0.77 perplexity), while low-recoverability blocks flag where this is unsafe. It further exposes a methodological pitfall: trained linear baselines can badly under-converge on ill-conditioned transformer activations, so we report the exact closed-form least-squares ceiling throughout.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 11 of its 17 references that are in the map. First posted on arXiv 12 Jun 2026.

Builds on

11 of 17 references.

References · 11

  1. 2023LLaMA: Open and Efficient Foundation Language Modelscited by 14,132
  2. 2015Distilling the Knowledge in a Neural Networkcited by 10,324
  3. 2016Pointer Sentinel Mixture Modelscited by 3,096
  4. 2020GLU Variants Improve Transformercited by 1,804
  5. 2023Pythia: A Suite for Analyzing Large Language Models Across Training and Scalingcited by 1,786
  6. 2016Understanding intermediate layers using linear classifier probescited by 1,353
  7. 2020Transformer Feed-Forward Layers Are Key-Value Memoriescited by 1,336
  8. 2022LLM.int8(): 8-bit Matrix Multiplication for Transformers at Scalecited by 906
  9. 2021Revisiting Model Stitching to Compare Neural Representationscited by 209
  10. 2020Deep Polynomial Neural Networkscited by 63
  11. 2016Higher-Order Factorization Machinescited by 45

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2026Variance Is Not Importance: Structural Analysis of Transformer Compressibility Across Model Scales
  2. 2026Sparse Inter-Layer Dependencies of Transformer FFN Neurons
  3. 2025Activation Transport Operators
  4. 2026Sparsity Moves Computation: How FFN Architecture Reshapes Attention in Small Transformers
  5. 2026Finite-Time Analysis of Gradient Descent for Shallow Transformers
  6. 2026Transformer Field Theory: A Response-Theoretic Approach to Mechanistic Interpretability
  7. 2026Recoverable but Not Stationary:Local Linear Structures in Weights and Activations
  8. 2025BlockCert: Certified Blockwise Extraction of Transformer Mechanisms