mlsanityData to 14 Oct 2026

Persistent Recurrent Memory Between Transformer Layers - Improves Language Model Generalization

arXiv 2609.17251 ↗First posted 15 Sep 2026cs.CLcited by 0

Not on the map. Why?

Abstract (arXiv)

We introduce a simple architectural modification to decoder-only transformers: a persistent recurrent state that observes hidden representations via cross-attention, updates itself through a GRU, and modulates subsequent processing via gated addition. Inserted between the lower and upper halves of a 6-layer transformer, this module adds only 3.7% additional parameters while reducing evaluation loss from 2.438 ± 0.004 to 1.743 ± 0.018, corresponding to a 28.5% reduction on held-out language modeling data. The improvement is statistically significant across 5 random seeds (p < 0.01) and corresponds to reduced overfitting (generalization gap 0.12 vs 0.26). Through controlled ablations, we demonstrate that the improvement stems entirely from the persistent memory topology, not from auxiliary self-prediction objectives. A model with identical topology but no auxiliary loss performs equivalently, while a random auxiliary loss provides no benefit. Representation probing reveals that the persistent state encodes narrative position (52% vs 33% chance level)---information that standard attention maintains less efficiently. Our results suggest that bridging transformer layers with a lightweight recurrent memory is a simple, effective approach to improving generalization in small-scale language models.

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

Builds on

7 of 12 references.

References · 7

  1. 2023Mamba: Linear-Time Sequence Modeling with Selective State Spacescited by 4,718
  2. 2019Transformer-XL: Attentive Language Models Beyond a Fixed-Length Contextcited by 1,850
  3. 2014Neural Turing Machinescited by 644
  4. 2023TinyStories: How Small Can Language Models Be and Still Speak Coherent English?cited by 449
  5. 2014Memory Networkscited by 363
  6. 2022UL2: Unifying Language Learning Paradigmscited by 295
  7. 2022Block-Recurrent Transformerscited by 110

Built on by

Not cited yet.