Learning Length-Extrapolatable Recurrent Models
Not on the map. Why?
Abstract (arXiv)
Recurrent models provide a natural path to long-context modeling, yet models trained with backpropagation through time (BPTT) often fail beyond their training horizon. Classical analyses emphasize gradients that vanish or explode along temporal paths. However, dense per-token losses can still train a shared recurrent rule despite severe decay, showing that decay alone does not determine whether learning fails. We instead study state credit: the signal through which future losses reach earlier recurrent states before contributing to parameter updates. Accordingly, we intervene directly on state credit and propose Credit Stabilization through Time (CST). During backward propagation, CST locally rescales the state-credit signal to stabilize its norm without rotating the component being corrected, while leaving the forward computation unchanged. Because controlled synthetic tasks and real data exhibit different credit dynamics, we specialize CST to each regime. In both settings, CST improves performance beyond the training horizon, with gains observed at up to 128x the training length.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 41 of its 58 references that are in the map. First posted on arXiv 8 Sep 2026.
Builds on
41 of 58 references.
References · 41
- 2017Attention Is All You Needcited by 60,361
- 2021Emerging Properties in Self-Supervised Vision Transformerscited by 6,968
- 2024Qwen2.5 Technical Reportcited by 4,942
- 2016Overcoming catastrophic forgetting in neural networkscited by 4,787
- 2023Mamba: Linear-Time Sequence Modeling with Selective State Spacescited by 4,718
- 2014Learning Phrase Representations using RNN Encoder-Decoder for Statistical Machine Translationcited by 4,057
- 2024Gemini 1.5: Unlocking multimodal understanding across millions of tokens of contextcited by 3,224
- 2021Efficiently Modeling Long Sequences with Structured State Spacescited by 2,465
- 2020The Pile: An 800GB Dataset of Diverse Text for Language Modelingcited by 2,411
- 2020Transformers are RNNs: Fast Autoregressive Transformers with Linear Attentioncited by 2,022
- 2017Gradient Episodic Memory for Continual Learningcited by 1,898
- 2021Train Short, Test Long: Attention with Linear Biases Enables Input Length Extrapolationcited by 1,103
29 more
- 2024Chameleon: Mixed-Modal Early-Fusion Foundation Modelscited by 927
- 2023Extending Context Window of Large Language Models via Positional Interpolationcited by 740
- 2023YaRN: Efficient Context Window Extension of Large Language Modelscited by 601
- 2024Gated Delta Networks: Improving Mamba2 with Delta Rulecited by 532
- 2016RL2: Fast Reinforcement Learning via Slow Reinforcement Learningcited by 503
- 2021Linear Transformers Are Secretly Fast Weight Programmerscited by 502
- 2018Fundamentals of Recurrent Neural Network (RNN) and Long Short-Term Memory (LSTM) Networkcited by 420
- 2024Parallelizing Linear Transformers with the Delta Rule over Sequence Lengthcited by 396
- 2016Learning to reinforcement learncited by 377
- 2024Learning to (Learn at Test Time): RNNs with Expressive Hidden Statescited by 355
- 2021Efficient Attentions for Long Document Summarizationcited by 345
- 2024LongRoPE: Extending LLM Context Window Beyond 2 Million Tokenscited by 313
- 2015Unitary Evolution Recurrent Neural Networkscited by 294
- 2020Query-Key Normalization for Transformerscited by 280
- 2023Physics of Language Models: Part 3.1, Knowledge Storage and Extractioncited by 270
- 2026Self-Distillation Enables Continual Learningcited by 247
- 2025Kimi Linear: An Expressive, Efficient Attention Architecturecited by 196
- 2022In-context Reinforcement Learning with Algorithm Distillationcited by 190
- 2025Test-Time Training Done Rightcited by 130
- 2022A Length-Extrapolatable Transformercited by 127
- 2024What is Wrong with Perplexity for Long-context Language Modeling?cited by 69
- 2017Unbiased Online Recurrent Optimizationcited by 47
- 2018Dynamical Isometry and a Mean Field Theory of RNNs: Gating Enables Signal Propagation in Recurrent Neural Networkscited by 45
- 2017Unbiasing Truncated Backpropagation Through Timecited by 37
- 2018Approximating Real-Time Recurrent Learning with Random Kronecker Factorscited by 29
- 2026M2RNN: Non-Linear RNNs with Matrix-Valued States for Scalable Language Modelingcited by 8
- 2025Spherical Leech Quantization for Visual Tokenization and Generationcited by 6
- 2026Pretraining Recurrent Networks without Recurrencecited by 5
- 2026Chimera: Designing and Chinchilla-Scaling Hybrid Visual Diffusion Transformerscited by 3
Built on by
Not cited yet.