In plain words: Mamba reads text quickly but loses track of long documents; ReMamba re-reads the input twice, squeezing key details into a compact form so nothing is dropped. On two long-text benchmarks it beat the original by 3.2 and 1.6 points, nearly matching same-size transformer models.
Abstract
While the Mamba architecture demonstrates superior inference efficiency and competitive performance on short-context natural language processing (NLP) tasks, empirical evidence suggests its capacity to comprehend long contexts is limited compared to transformer-based models. In this study, we investigate the long-context efficiency issues of the Mamba models and propose ReMamba, which enhances Mamba's ability to comprehend long contexts. ReMamba incorporates selective compression and adaptation techniques within a two-stage re-forward process, incurring minimal additional inference costs overhead. Experimental results on the LongBench and L-Eval benchmarks demonstrate ReMamba's efficacy, improving over the baselines by 3.2 and 1.6 points, respectively, and attaining performance almost on par with same-size transformer models.
Danlong Yuan, Jiahao Liu, Bei Li, Huishuai Zhang, Jingang Wang, Xunliang Cai, Dongyan Zhao
arXiv:2408.15496 · cs.CL · submitted Aug 28, 2024 · updated Jan 1, 2025
abstract · pdf · html