MARCH: Scaling Recurrent Memory with Content-Routed State Anchors
Abstract
MARCH enhances recurrent state-space models by caching historical state anchors with content-conditioned keys, enabling scalable long-range memory and improved retrieval without sacrificing efficiency.
Transformers owe much of their strong long-context retrieval capability to a token-level memory that grows with context length. This flexibility, however, incurs a quadratic computation complexity during training and a key--value cache that grows linearly during autoregressive inference. Recurrent alternatives offer efficient decoding by compressing the entire history into a fixed-size state, but often underperform on recall-intensive tasks since earlier associations usually get overwritten by subsequent updates, and only the most recent contextual information is retained. In this paper, we introduce Memory-Anchor Routing across Context History (MARCH), a network architecture that effectively scales state-space models beyond a fixed-size dimension, while maintaining computational efficiency over long-sequences. MARCH periodically caches cumulative recurrent-state checkpoints as state anchors and associates each anchor with a compact, content-conditioned anchor key. This lets MARCH maintain a memory bank, which can grow as context length increases, providing a controllable trade-off between historical resolution and memory cost. At each token, MARCH produces an anchor query to attend all causally available state anchors, and the output is calculated as an attention-style aggregation over all historical anchors along the current state. We show that after standard pretraining, MARCH consistently outperforms multiple linear attention variants across commonsense reasoning, LongBench, and in-context retrieval. These results demonstrate that content-routed state caching substantially strengthens recurrent long-range memory while preserving its native computation path.
Community
Hi, folks! I just came across your cool paper! We had a very relevant paper called RAT at NeurIPS ’25: https://arxiv.org/abs/2507.04416, where we apply cross-chunk softmax attention over summarized recurrent states. It would be great if you could consider citing it!
We also have a follow-up paper that makes the chunk size selection flexible at inference: https://arxiv.org/pdf/2602.18196. Happy to connect/chat!
Hi folks, thanks for sharing RAT and RAT+! Both are relevant to MARCH. The connection through attention over recurrently compressed representations is clear, and RAT+’s full-sequence recurrence and flexible inference-time sparsification are particularly interesting.
MARCH focuses on retrieving from cumulative matrix-valued recurrent-state checkpoints, with separate routing and state-read queries, while preserving the original recurrent readout. We’ll cite both papers and discuss these connections and differences in the next revision.
Thanks again for bringing these works to our attention—we’d be happy to connect and exchange ideas!
Get this paper in your agent:
hf papers read 2608.12435 Don't have the latest CLI?
curl -LsSf https://hf.co/cli/install.sh | bash Models citing this paper 0
No model linking this paper
Datasets citing this paper 0
No dataset linking this paper
Spaces citing this paper 0
No Space linking this paper
Collections including this paper 0
No Collection including this paper