A Single-Layer Model Can Do Language Modeling
On the map: State-space sequence models (S4, Mamba) → Core SSM architectures for language
also in 1 more
Abstract (arXiv)
Modern language models scale depth by stacking layers, each holding its own state - a per-layer KV cache in transformers, a per-layer matrix in Mamba, Gated DeltaNet (GDN), RWKV, and xLSTM. Biological systems lean heavily on recurrence rather than on stacking. We ask how far that shape can go on language modeling. We propose Grounded Prediction Networks (GPN): one state vector revisited at every step through a single recurrent block - one FFN, one shared matrix memory. At 130M parameters, a 1-layer GPN+M reaches FineWeb-Edu perplexity 18.06, within 13% of a 12-layer Transformer++ (16.05) and 18% of a 10-layer GDN (15.34); a 2-layer variant closes the gap to 6%/11%. We do not match the deep baselines. Because the working context is a single vector, we can directly inspect its geometry: a persistent default-token direction, a content-bearing horizon of tens of tokens, and memory heads that split spontaneously into fast and slow retention pools.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in Core SSM architectures for language, one of 10 directions in State-space sequence models (S4, Mamba). Builds on 16 of its 20 references that are in the map. First posted on arXiv 11 May 2026.
Builds on
16 of 20 references.
Closest ancestors · 5
Same direction, newest first
- 2024Learning to (Learn at Test Time): RNNs with Expressive Hidden States5 Jul
- 2024Transformers are SSMs: Generalized Models and Efficient Algorithms Through Structured State Space Duality31 May
- 2023Gated Linear Attention Transformers with Hardware-Efficient Training11 Dec
- 2023Mamba: Linear-Time Sequence Modeling with Selective State Spaces1 Dec
- 2021Efficiently Modeling Long Sequences with Structured State Spaces31 Oct
Other references · 11
- 2020Transformers are RNNs: Fast Autoregressive Transformers with Linear Attentioncited by 1,979
- 2024Gated Delta Networks: Improving Mamba2 with Delta Rulecited by 483
- 2021Linear Transformers Are Secretly Fast Weight Programmerscited by 471
- 2024Parallelizing Linear Transformers with the Delta Rule over Sequence Lengthcited by 373
- 2024xLSTM: Extended Long Short-Term Memorycited by 320
- 2024Titans: Learning to Memorize at Test Timecited by 303
- 2024Griffin: Mixing Gated Linear Recurrences with Local Attention for Efficient Language Modelscited by 232
- 2024Leave No Context Behind: Efficient Infinite Context Transformers with Infini-attentioncited by 194
- 2025RWKV-7 "Goose" with Expressive Dynamic State Evolutioncited by 134
- 2024Hymba: A Hybrid-head Architecture for Small Language Modelscited by 95
- 2025Nested Learning: The Illusion of Deep Learning Architecturescited by 69
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2022Transformer Feed-Forward Layers Build Predictions by Promoting Concepts in the Vocabulary Space
- 2026M2RNN: Non-Linear RNNs with Matrix-Valued States for Scalable Language Modeling
- 2026Hidden Decoding at Scale: Latent Computation Scaling for Large Language Models
- 2025Efficient Parallel Samplers for Recurrent-Depth Models and Their Connection to Diffusion Language Models
- 2026Harmonic: Hierarchical State Space Models for Efficient Long-Context Language Modeling
- 2024Memory Layers at Scale
- 2025ParaRNN: Unlocking Parallel Training of Nonlinear RNNs for Large Language Models
- 2026Trading Depth for Time in Recurrent Transformers