arXiv:2410.08133cs.CLcs.AI2024-10被引 9

用排序回忆任务评估大模型的事件记忆能力,发现其在上下文提示下表现良好但训练时未见长时记忆。

Assessing Episodic Memory in LLMs with Sequence Order Recall Tasks

  • 设计序列顺序回忆任务(SORT),模拟人类对事件发生顺序的记忆。
  • 在36000个书段对上测试,模型在上下文提示下准确率达高,但仅靠训练数据表现差。
  • 适合研究记忆增强型模型,填补大模型记忆评估的空白。

当前大模型评估主要关注事实与语义关系的记忆,侧重长期记忆的语义层面。然而,人类长期记忆还包括情境记忆,即把记忆与时间、地点等背景关联。这种情境化记忆对认知任务和日常功能至关重要,但现有基准尚未评估大模型的此类能力。为此,本文引入序列顺序回忆任务(SORT),源自认知心理学中研究情境记忆的方法。SORT要求模型回忆文本片段的正确顺序,具有通用性且无需额外标注。我们构建了初始数据集Book-SORT,包含从9本近期进入公共领域的书籍中提取的36,000对文本段落。基于155名参与者的实验,证明人类可依靠长期记忆回忆书本顺序。结果显示,当相关文本在推理时提供上下文时,模型能以高准确率完成任务;但若仅在训练阶段接触书籍内容,模型表现不佳。SORT有助于更全面评估大模型的记忆能力,推动记忆增强模型的发展。

原文摘要 · Abstract (English)

Current LLM benchmarks focus on evaluating models' memory of facts and semantic relations, primarily assessing semantic aspects of long-term memory. However, in humans, long-term memory also includes episodic memory, which links memories to their contexts, such as the time and place they occurred. The ability to contextualize memories is crucial for many cognitive tasks and everyday functions. This form of memory has not been evaluated in LLMs with existing benchmarks. To address the gap in evaluating memory in LLMs, we introduce Sequence Order Recall Tasks (SORT), which we adapt from tasks used to study episodic memory in cognitive psychology. SORT requires LLMs to recall the correct order of text segments, and provides a general framework that is both easily extendable and does not require any additional annotations. We present an initial evaluation dataset, Book-SORT, comprising 36k pairs of segments extracted from 9 books recently added to the public domain. Based on a human experiment with 155 participants, we show that humans can recall sequence order based on long-term memory of a book. We find that models can perform the task with high accuracy when relevant text is given in-context during the SORT evaluation. However, when presented with the book text only during training, LLMs' performance on SORT falls short. By allowing to evaluate more aspects of memory, we believe that SORT will aid in the emerging development of memory-augmented models.

大模型记忆情境记忆评估基准序列理解

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。