提出新基准与记忆框架,显著提升大模型长对话记忆能力
Beyond a Million Tokens: Benchmarking and Enhancing Long-Term Memory in LLMs
- 自动生成长达1000万词的连贯多主题对话并设计测评问题
- 在超长对话中,现有模型性能随长度下降,而新框架提升3.5%-12.69%
- 模仿人类认知,构建三重记忆系统,适合长程交互场景研究者
评估大语言模型在需长期记忆的任务(如对话)中的表现,受限于现有基准:缺乏叙事连贯性、领域狭窄、仅测试简单回忆。本文提出综合性解决方案:首先,构建自动生成长达1000万词、内容连贯且主题多样的对话框架,并配套覆盖多种记忆能力的探测问题;基于此构建新基准BEAM,包含100场对话和2000个验证问题。其次,提出LIGHT框架,受人类认知启发,为模型配备三种互补记忆系统:长期情景记忆、短期工作记忆和用于积累关键事实的草稿区。在BEAM上的实验表明,即使具备100万词上下文窗口的模型(含检索增强)在对话变长后仍表现不佳。相比之下,LIGHT在各类模型上持续提升性能,平均优于最强基线3.5%至12.69%,具体取决于底层模型。消融实验进一步证实各记忆组件的有效性。
原文摘要 · Abstract (English)
Evaluating the abilities of large language models (LLMs) for tasks that require long-term memory and thus long-context reasoning, for example in conversational settings, is hampered by the existing benchmarks, which often lack narrative coherence, cover narrow domains, and only test simple recall-oriented tasks. This paper introduces a comprehensive solution to these challenges. First, we present a novel framework for automatically generating long (up to 10M tokens), coherent, and topically diverse conversations, accompanied by probing questions targeting a wide range of memory abilities. From this, we construct BEAM, a new benchmark comprising 100 conversations and 2,000 validated questions. Second, to enhance model performance, we propose LIGHT-a framework inspired by human cognition that equips LLMs with three complementary memory systems: a long-term episodic memory, a short-term working memory, and a scratchpad for accumulating salient facts. Our experiments on BEAM reveal that even LLMs with 1M token context windows (with and without retrieval-augmentation) struggle as dialogues lengthen. In contrast, LIGHT consistently improves performance across various models, achieving an average improvement of 3.5%-12.69% over the strongest baselines, depending on the backbone LLM. An ablation study further confirms the contribution of each memory component.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。