'RNN Homunculus' paper: gated RNNs can match Mamba on long memory
TL;DR
- Gated RNNs can match or exceed Mamba and xLSTM test accuracy when their gates are learned properly, the authors claim.
- The analysis finds gated architectures suffer the vanishing gradient problem when all training samples require long-range memory.
- Adding short-dependency samples lets gated models learn solutions that capture long dependencies and generalize across class labels.
Gated RNNs, trained the right way, can match or beat Mamba and xLSTM on long-memory tasks. That is the claim in a new arxiv preprint by Zeyu Fan, Alexandru-Dragos Manolache and Jan van Gemert.
The authors look at why gated recurrent networks like GRUs and LSTMs are supposed to handle long-range dependencies and often don't. The standard story is that gates open additive copy paths through time. The paper says that story is 'correct, but incomplete': the gates themselves have to be learned, and learning them runs into the same vanishing-gradient problem the gates were meant to solve. 'Gated architectures struggle when all training samples require long-range memory and thus equally suffer from the vanishing gradient problem,' the paper reports.
Their fix is unglamorous. The authors mix short-dependency samples into training alongside the long ones, which lets gated models 'learn solutions that capture long-dependency samples, but also generalize across different class labels.' With that curriculum, they write, 'gated RNNs match or even exceed the test accuracy of modern architectures like Mamba and xLSTM.' The experiments are controlled sequence classification tasks, not language modeling, so the direct comparison to Mamba in its best-known setting is not what the paper claims to settle.
Shared on Bluesky by 2 AI experts
-
I really enjoy finding gaps in our understanding of text-book deep learning. The RNN Homunculus: Long-range input vanishes RNN gradients. Gates (GRU, LSTM) help, but gates are themselves an RNN. Ie: gates are not suffic…
View on Bluesky →
Originally reported by arxiv.org
Read the original article →Original headline: The RNN Homunculus: Learning to Gate Long-Term Memory