Parallel Recursive LSTM
Not on the map. Why?
Abstract (arXiv)
Transformers have become the dominant architecture for sequence modeling by using self-attention to enable expressive and highly parallel processing. However, the resulting quadratic time and memory costs limit efficiency in long-context settings. Recurrent models such as LSTMs provide explicit nonlinear state updates and strong state-tracking capabilities, yet their strictly sequential computation limits parallelism. We introduce the Parallel Recursive LSTM (PR-LSTM), a hierarchical recurrent architecture that replaces left-to-right recurrence with recursive nonlinear state composition over a balanced computation tree. Tokens are first mapped independently to latent states, which are then recursively merged by a learned gated composition block. This structure uses the reduction pattern underlying parallel scans as a fixed execution schedule, rather than assuming an associative recurrence. As a result, PR-LSTM retains nonlinear gated state representations while reducing recurrent parallel depth from linear to logarithmic. Empirically, PR-LSTM achieves strong sequence-length generalization on formal-language benchmarks, solving more tasks than standard RNN, LSTM, and Transformer baselines, while avoiding the quadratic scaling of attention. These results suggest that recurrent computation can be reorganized hierarchically to expose parallelism without restricting the transition dynamics to linear or associative forms.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 19 of its 25 references that are in the map. First posted on arXiv 16 May 2026.
Builds on
19 of 25 references.
References · 19
- 2017Attention Is All You Needcited by 59,932
- 2023Mamba: Linear-Time Sequence Modeling with Selective State Spacescited by 4,616
- 2021Efficiently Modeling Long Sequences with Structured State Spacescited by 2,424
- 2020Transformers are RNNs: Fast Autoregressive Transformers with Linear Attentioncited by 1,979
- 2020Rethinking Attention with Performerscited by 1,584
- 2024Transformers are SSMs: Generalized Models and Efficient Algorithms Through Structured State Space Dualitycited by 1,283
- 2022Simplified State Space Layers for Sequence Modelingcited by 776
- 2023Retentive Network: A Successor to Transformer for Large Language Modelscited by 594
- 2023Hyena Hierarchy: Towards Larger Convolutional Language Modelscited by 388
- 2022Diagonal State Spaces are as Effective as Structured State Spacescited by 350
- 2024xLSTM: Extended Long Short-Term Memorycited by 320
- 2015Improved Semantic Representations From Tree-Structured Long Short-Term Memory Networkscited by 305
7 more
- 2022Neural Networks and the Chomsky Hierarchycited by 214
- 2022The Parallelism Tradeoff: Limitations of Log-Precision Transformerscited by 191
- 2024The Illusion of State in State-Space Modelscited by 181
- 2023Parallelizing non-linear sequential models over the sequence lengthcited by 45
- 2017Understanding Hidden Memories of Recurrent Neural Networkscited by 43
- 2024Towards Scalable and Stable Parallelization of Nonlinear RNNscited by 38
- 2025ParaRNN: Unlocking Parallel Training of Nonlinear RNNs for Large Language Modelscited by 13
Built on by
Not cited yet.