Fast Training of Recurrent Neural Networks with Stationary State Feedbacks
Not on the map. Why?
Abstract (arXiv)
Recurrent neural networks (RNNs) have recently demonstrated strong performance and faster inference than Transformers at comparable parameter budgets. However, the recursive gradient computation with the backpropagation through time (or BPTT) algorithm remains the major computational bottleneck. In this work, we propose a novel method that replaces BPTT with a fixed gradient feedback mechanism, yielding an efficient approximation of the exact gradient propagation based on the assumption of time stationarity. Our approach leverages state-space model (SSM) principles to define a structured feedback matrix that directly propagates gradients from future time steps. This formulation bypasses the need for recursive gradient backpropagation, significantly reducing training overhead while preserving the network's ability to capture long-term dependencies. The experiments on language modeling benchmarks exhibit competitive perplexity scores, while significantly reducing the training costs. These promising results suggest that designing a feedback method like an SSM can fully exploit the efficiency advantages of RNNs for many practical applications.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 13 of its 22 references that are in the map. First posted on arXiv 29 Mar 2025.
Builds on
13 of 22 references.
References · 13
- 2017Attention Is All You Needcited by 60,361
- 2023Mamba: Linear-Time Sequence Modeling with Selective State Spacescited by 4,718
- 2014Neural Machine Translation by Jointly Learning to Align and Translatecited by 4,548
- 2014Learning Phrase Representations using RNN Encoder-Decoder for Statistical Machine Translationcited by 4,057
- 2021Efficiently Modeling Long Sequences with Structured State Spacescited by 2,465
- 2022Simplified State Space Layers for Sequence Modelingcited by 788
- 2020HiPPO: Recurrent Memory with Optimal Polynomial Projectionscited by 605
- 2022On the Parameterization and Initialization of Diagonal State Space Modelscited by 424
- 2023Resurrecting Recurrent Neural Networks for Long Sequencescited by 408
- 2022Diagonal State Spaces are as Effective as Structured State Spacescited by 350
- 2024xLSTM: Extended Long Short-Term Memorycited by 327
- 2016Direct Feedback Alignment Provides Learning in Deep Neural Networkscited by 240
1 more
- 2022Mega: Moving Average Equipped Gated Attentioncited by 179
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2024Recurrent neural networks: vanishing and exploding gradients are not the end of the story
- 2025ParaRNN: Unlocking Parallel Training of Nonlinear RNNs for Large Language Models
- 2023State space models can express n-gram languages
- 2026When Learning Hurts: Fixed-Pole RNN for Real-Time Online Training
- 2025Implicit Language Models are RNNs: Balancing Parallelization and Expressivity
- 2024State-Space Modeling in Long Sequence Processing: A Survey on Recurrence in the Transformer Era
- 2025Memory Determines Learning Direction: A Theory of Gradient-Based Optimization in State Space Models
- 2026M2RNN: Non-Linear RNNs with Matrix-Valued States for Scalable Language Modeling