Latent Recurrent Transformer: Architecture Exploration, Training Strategies, and Scaling Behavior
Not on the map. Why?
Abstract (arXiv)
We study Latent Recurrent Transformer (LRT), a lightweight augmentation of autoregressive transformers that reuses a high-level source-layer hidden state from the previous token as recurrent memory for the next token. Because this state is already computed during ordinary decoding, LRT introduces a cross-token, cross-layer latent pathway while preserving the standard attention mechanism, KV-cache interface, and one model forward per generated token. To pretrain this recurrence without sequentially unrolling the full sequence, we introduce interleaved parallel training: one full-sequence initialization forward constructs a shared buffer, followed by sequential refinement of disjoint position subsets with parallel computation within each subset. This provides every token with recurrent-memory-aware supervision at approximately 2x ideal token compute. Across 1.3B- and 2.1B-parameter nanochat-style backbones and a wide range of training budgets, LRT improves both BPB and CORE under matched effective compute. Additionally, LRT outperforms two-forward PonderLM-2 and matches a three-loop Transformer in BPB, while retaining one-forward-per-token decoding with 9% latency overhead over the standard Transformer.
Key facts. Cited by 6 papers in the map (data through 14 Oct 2026). Builds on 39 of its 45 references that are in the map. First posted on arXiv 26 May 2026.
Builds on
39 of 45 references.
References · 39
- 2017Attention Is All You Needcited by 60,361
- 2020Language Models are Few-Shot Learnerscited by 32,419
- 2023Mamba: Linear-Time Sequence Modeling with Selective State Spacescited by 4,718
- 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,703
- 2021Efficiently Modeling Long Sequences with Structured State Spacescited by 2,465
- 2020Transformers are RNNs: Fast Autoregressive Transformers with Linear Attentioncited by 2,022
- 2024Gemma 2: Improving Open Language Models at a Practical Sizecited by 1,998
- 2019Transformer-XL: Attentive Language Models Beyond a Fixed-Length Contextcited by 1,850
- 2019Root Mean Square Layer Normalizationcited by 1,480
- 2023RWKV: Reinventing RNNs for the Transformer Eracited by 827
- 2024Training Large Language Models to Reason in a Continuous Latent Spacecited by 797
- 2022Simplified State Space Layers for Sequence Modelingcited by 788
27 more
- 2021Going deeper with Image Transformerscited by 764
- 2019Compressive Transformers for Long-Range Sequence Modellingcited by 726
- 2023Retentive Network: A Successor to Transformer for Large Language Modelscited by 613
- 2022Hungry Hungry Hippos: Towards Language Modeling with State Space Modelscited by 499
- 2025Scaling up Test-Time Compute with Latent Reasoning: A Recurrent Depth Approachcited by 415
- 2024Jamba: A Hybrid Transformer-Mamba Language Modelcited by 401
- 2024DataComp-LM: In search of the next generation of training sets for language modelscited by 398
- 2023Hyena Hierarchy: Towards Larger Convolutional Language Modelscited by 393
- 2024Learning to (Learn at Test Time): RNNs with Expressive Hidden Statescited by 355
- 2024Titans: Learning to Memorize at Test Timecited by 328
- 2024xLSTM: Extended Long Short-Term Memorycited by 327
- 2024Quiet-STaR: Language Models Can Teach Themselves to Think Before Speakingcited by 301
- 2025Gated Attention for Large Language Models: Non-linearity, Sparsity, and Attention-Sink-Freecited by 283
- 2023Think before you speak: Training Language Models With Pause Tokenscited by 264
- 2024Griffin: Mixing Gated Linear Recurrences with Local Attention for Efficient Language Modelscited by 242
- 2023Looped Transformers as Programmable Computerscited by 237
- 2022Memorizing Transformerscited by 226
- 2023Landmark Attention: Random-Access Infinite Context Length for Transformerscited by 213
- 2022Recurrent Memory Transformercited by 203
- 2024Leave No Context Behind: Efficient Infinite Context Transformers with Infini-attentioncited by 200
- 2024Let's Think Dot by Dot: Hidden Computation in Transformer Language Modelscited by 191
- 2022Block-Recurrent Transformerscited by 110
- 2025Parallel Continuous Chain-of-Thought with Jacobi Iterationcited by 25
- 2023Block-State Transformerscited by 24
- 2024Thinking Tokens for Language Modelingcited by 24
- 2024TransformerFAM: Feedback attention is working memorycited by 22
- 2025PonderLM-2: Pretraining LLM with Latent Thoughts in Continuous Spacecited by 16
Built on by
Citing papers per year: 2026: 6.
6 papers.
- 2026Full-bandwidth transformercited by 5
- 2026Maglev: Sliding Recurrent Memorycited by 3
- 2026WhiteMatter: All-to-All Cross-Layer Connections via KV Source Mixingcited by 0
- 2026Trading Depth for Time in Recurrent Transformerscited by 0
- 2026GradLev: Token-Parallel Test-Time Training Via Costate Predictioncited by 0
- 2026Recurrent Looped Transformercited by 0
- 2026Recurrent Looped Transformer6 Oct 2026
- 2026GradLev: Token-Parallel Test-Time Training Via Costate Prediction28 Sep 2026
- 2026Trading Depth for Time in Recurrent Transformers18 Sep 2026
- 2026WhiteMatter: All-to-All Cross-Layer Connections via KV Source Mixing19 Aug 2026
- 2026Full-bandwidth transformer9 Aug 2026
- 2026Maglev: Sliding Recurrent Memory3 Aug 2026
Similar, not cited
Close in wording, no citation either way
- 2026The Recurrent Transformer: Greater Effective Depth and Efficient Decoding
- 2026Looped Latent Attention: Cross-Loop KV Compression for Looped Transformers
- 2026T^2MLR: Transformer with Temporal Middle-Layer Recurrence
- 2026Hidden Decoding at Scale: Latent Computation Scaling for Large Language Models
- 2026SpotAttention: Plug-In Block-Sparse Routing for Pretrained Long-Context Transformers
- 2026Long-Context Aware Upcycling: A New Frontier for Hybrid LLM Scaling
- 2024RecurFormer: Not All Transformer Heads Need Self-Attention
- 2026The Context-Ready Transformer