ConvMem:卷积记忆模型提升长上下文推理
ConvMem: Convolutional Memory for Long-Context Reasoning
ConvMem解决了长上下文推理的高延迟问题,无需强化学习训练,在多个基准测试中表现优异。
ConvMem是一种无需训练、高度并行化的框架,将长上下文推理重新表述为分层卷积。该模型在RULER-HotpotQA和RULER-2WikiMultiHopQA基准测试中表现优于无需训练的基线模型。ConvMem采用可配置步长和跳跃连接确保稳健的证据捕获和传播,同时使用多核卷积将复杂查询解耦为独立的语义通道。该设计不仅减少了错误累积,还实现了文本段落和推理线程的大规模并行化。
ConvMem: Convolutional Memory for Long-Context Reasoning
While Large Language Models (LLMs) have demonstrated impressive capabilities, they often struggle with extremely long contexts due to fixed context limits. To address this, sequential approaches like MemAgent extend the effective context by reading text in segments and iteratively updating a fixed-size memory. However, this sequential paradigm suffers from high latency and requires costly reinforcement learning (RL) training, which can lead to overfitting on specific datasets. To overcome these limitations, we propose ConvMem, a training-free, highly parallelizable framework that reformulates long-context reasoning as a hierarchical convolution. Inspired by CNNs, ConvMem treats an LLM prompted with a specific query as a convolutional kernel. This kernel summarizes text segments hierarchically, shortening the reasoning path from a linear chain into a logarithmic tree. Specifically, ConvMem integrates \textit{Configurable Strides} and \textit{Skip Connections} to ensure robust evidence capture and propagation, while employing \textit{Multi-Kernel Convolution} to decompose complex queries into disentangled semantic channels. This design not only mitigates error accumulation but also enables massive parallelization across both text segments and reasoning threads. Experiments on RULER-HotpotQA and RULER-2WikiMultiHopQA demonstrate that ConvMem outperforms training-free baselines and avoids the risk of overfitting to parametric priors often observed in RL-trained models on out-of-distribution tasks.