Attention is Just Another Name for Coupling? A Fast-Slow ODE Perspective on Hierarchical Pretraining
Not on the map. Why?
Abstract (arXiv)
We re-interpret Transformer pretraining as a fast-slow, singularly perturbed flow along depth, with untied weights as its non-autonomous feature. The linearised dynamics is a depth-ordered product of layer maps. Along a token-homogeneous reference trajectory, the linearised layer factorises along the eigenbasis of a frozen attention kernel. Past a computable saturation depth, the flow factors through the block coarse-graining -- in other words, running the layers is running the coarse variable, dually. Weight perturbations supported on the decaying bundle move neither the persistent component of the distinguished trajectory nor the frozen kernel to first order, so the framework partitions parameter space into visible and invisible directions, with the cross-block coupling of the slow path sitting entirely on the visible side. How large a gate the slow path can carry is bounded by a stability margin. On the data side: if block emissions follow an exponential family, block-mean pooling captures all the information the slow path can use; but if neighbouring blocks carry no shared structure, no cross-block channel can help the prediction, and the gate amplitude is invisible in the prediction risk. Stability delimits what the architecture may do; the data decides what it will.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 20 of its 44 references that are in the map. First posted on arXiv 15 Jun 2026.
Builds on
20 of 44 references.
References · 20
- 2017Attention Is All You Needcited by 60,361
- 2020Scaling Laws for Neural Language Modelscited by 6,586
- 2016Layer Normalizationcited by 5,669
- 2016Gaussian Error Linear Units (GELUs)cited by 3,502
- 2022Training Compute-Optimal Large Language Modelscited by 3,019
- 2020Big Bird: Transformers for Longer Sequencescited by 1,684
- 2019Root Mean Square Layer Normalizationcited by 1,480
- 2018On the Spectral Bias of Neural Networkscited by 1,341
- 2020On Layer Normalization in the Transformer Architecturecited by 922
- 2019Deep Equilibrium Modelscited by 615
- 2023Gated Linear Attention Transformers with Hardware-Efficient Trainingcited by 514
- 2021Attention is Not All You Need: Pure Attention Loses Rank Doubly Exponentially with Depthcited by 419
8 more
- 2025Scaling up Test-Time Compute with Latent Reasoning: A Recurrent Depth Approachcited by 415
- 2018Generating Wikipedia by Summarizing Long Sequencescited by 335
- 2019SGD on Neural Networks Learns Functions of Increasing Complexitycited by 204
- 2025Hierarchical Reasoning Modelcited by 177
- 2023A mathematical perspective on Transformerscited by 164
- 2016Identity Matters in Deep Learningcited by 109
- 2026Parcae: Scaling Laws For Stable Looped Language Modelscited by 55
- 2026MiniMax Sparse Attentioncited by 44
Built on by
Not cited yet.