Overflow Prevention Enhances Long-Context Recurrent LLMs

Assaf Ben-Kish, Itamar Zimerman, M. Jehanzeb Mirza, Lior Wolf, James Glass +2 more
5/12/2025
cs.LGcs.AI

No official code found. Best implementation: assafbk/OPRM (18 stars, last commit 2025-07-08).

Abstract

A recent trend in LLMs is developing recurrent sub-quadratic models that improve long-context processing efficiency. We investigate leading large long-context models, focusing on how their fixed-size recurrent memory affects their performance. Our experiments reveal that, even when these models are trained for extended contexts, their use of long contexts remains underutilized. Specifically, we demonstrate that a chunk-based inference procedure, which identifies and processes only the most relevant portion of the input can mitigate recurrent memory failures and be effective for many long-context tasks: On LongBench, our method improves the overall performance of Falcon3-Mamba-Inst-7B by 14%, Falcon-Mamba-Inst-7B by 28%, RecurrentGemma-IT-9B by 50%, and RWKV6-Finch-7B by 51%. Surprisingly, this simple approach also leads to state-of-the-art results in the challenging LongBench v2 benchmark, showing competitive performance with equivalent size Transformers. Furthermore, our findings raise questions about whether recurrent models genuinely exploit long-range dependencies, as our single-chunk strategy delivers stronger performance - even in tasks that presumably require cross-context relations.

View on arXivView PDF

Code Implementations(1)

assafbk/OPRMCommunity100%

Overflow Prevention Enhances Long-Context Recurrent LLMs (COLM 2025)

182PythonMar 31, 20251 years ago

Cite this paper

@article{benkish2025overflow,
  title  = {Overflow Prevention Enhances Long-Context Recurrent LLMs},
  author = {Assaf Ben-Kish and Itamar Zimerman and M. Jehanzeb Mirza and Lior Wolf and James Glass and Leonid Karlinsky and Raja Giryes},
  year   = {2025},
  eprint = {2505.07793},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2505.07793v2}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2505.07793.svg)](https://codeofpaper.com/paper/2505.07793)

Discussion