mlsanityData to 7 Oct 2026

Exploration of Fast-Slow Latent Recurrence for Train-Short, Test-Long Generalization

arXiv 2604.01577 ↗First posted 2 Apr 2026cs.LG cs.AIcited by 2

Not on the map. Why?

Abstract (arXiv)

We study out of distribution generalization in streaming tasks where models are trained on short sequences but must operate over much longer, unknown horizons under bounded memory. Our focus is on a persistent fast slow recurrent formulation in which a latent state is maintained across observations rather than reset at each stream step. For each incoming observation, the model performs multiple weight-shared latent updates with a recurrent core and then carries the resulting state forward to the next observation. This allows the model to maintain and refine a compact stream-level state without reprocessing a growing context. We evaluate this formulation across symbolic sequence prediction, supervised navigation, and partially observable reinforcement learning tasks. Across these settings, persistent latent recurrence improves OOD generalization over recurrent, state-space, and Transformer baselines. Through recurrent-core ablations, we identify architectural ingredients that are consistently associated with strong OOD performance, including state-dependent transitions and feature-wise nonlinear mixing. Together, these results highlight the value of revisiting persistent recurrence as an architectural bias for more generalizable sequence prediction.

Key facts. Cited by 2 papers in the map (data through 7 Oct 2026). Builds on 45 of its 58 references that are in the map. First posted on arXiv 2 Apr 2026.

Builds on

45 of 58 references.

References · 45

  1. 2017Attention Is All You Needcited by 59,932
  2. 2023Mamba: Linear-Time Sequence Modeling with Selective State Spacescited by 4,616
  3. 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,581
  4. 2019Transformer-XL: Attentive Language Models Beyond a Fixed-Length Contextcited by 1,830
  5. 2021MLP-Mixer: An all-MLP Architecture for Visioncited by 1,441
  6. 2024Transformers are SSMs: Generalized Models and Efficient Algorithms Through Structured State Space Dualitycited by 1,283
  7. 2021Train Short, Test Long: Attention with Linear Biases Enables Input Length Extrapolationcited by 1,085
  8. 2022Simplified State Space Layers for Sequence Modelingcited by 776
  9. 2024Training Large Language Models to Reason in a Continuous Latent Spacecited by 763
  10. 2023Extending Context Window of Large Language Models via Positional Interpolationcited by 721
  11. 2018Universal Transformerscited by 613
  12. 2016Adaptive Computation Time for Recurrent Neural Networkscited by 479
33 more
  1. 2025Scaling up Test-Time Compute with Latent Reasoning: A Recurrent Depth Approachcited by 372
  2. 2023Minigrid & Miniworld: Modular & Customizable Reinforcement Learning Environments for Goal-Oriented Taskscited by 320
  3. 2022Transformers Learn Shortcuts to Automatacited by 260
  4. 2023Looped Transformers as Programmable Computerscited by 224
  5. 2025Reasoning with Latent Thoughts: On the Power of Looped Transformerscited by 223
  6. 2022Exploring Length Generalization in Large Language Modelscited by 204
  7. 2022The Parallelism Tradeoff: Limitations of Log-Precision Transformerscited by 191
  8. 2024The Illusion of State in State-Space Modelscited by 181
  9. 2025Hierarchical Reasoning Modelcited by 167
  10. 2023A mathematical perspective on Transformerscited by 160
  11. 2025Scaling Latent Reasoning via Looped Language Modelscited by 145
  12. 2023Looped Transformers are Better at Learning Learning Algorithmscited by 135
  13. 2021Self-Attention Networks Can Process Bounded Hierarchical Languagescited by 105
  14. 2023Tensor Programs VI: Feature Learning in Infinite-Depth Neural Networkscited by 94
  15. 2024Looped Transformers for Length Generalizationcited by 91
  16. 2024Transformers Can Achieve Length Generalization But Not Robustlycited by 82
  17. 2023GTA: A Geometry-Aware Attention Mechanism for Multi-View Transformerscited by 75
  18. 2014A Clockwork RNNcited by 70
  19. 2025DeltaProduct: Improving State-Tracking in Linear RNNs via Householder Productscited by 69
  20. 2013How to Construct Deep Recurrent Neural Networkscited by 64
  21. 2020RNNs can generate bounded hierarchical languages with optimal memorycited by 59
  22. 2024Dynamic metastability in the self-attention modelcited by 42
  23. 2025Teaching Pretrained Language Models to Think Deeper with Retrofitted Recurrencecited by 41
  24. 2024Artificial Kuramoto Oscillatory Neuronscited by 37
  25. 2025Energy-Based Transformers are Scalable Learners and Thinkerscited by 30
  26. 2022DynaMixer: A Vision MLP Architecture with Dynamic Mixingcited by 28
  27. 2024On Expressive Power of Looped Transformers: Theoretical Analysis and Enhancement via Timestep Encodingcited by 28
  28. 2024Resonance RoPE: Improving Context Length Generalization of Large Language Modelscited by 25
  29. 2025Implicit Language Models are RNNs: Balancing Parallelization and Expressivitycited by 21
  30. 2024Language Models as Compilers: Simulating Pseudocode Execution Improves Algorithmic Reasoning in Language Modelscited by 20
  31. 2017Fast-Slow Recurrent Neural Networkscited by 12
  32. 2022Active Token Mixercited by 6
  33. 2025Long-Short Alignment for Effective Long-Context Modeling in LLMscited by 3

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 2.

2 papers.