arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2402.09727cs.CLcs.AIcs.IR
收藏

一种具有长上下文要点记忆的人类启发式阅读代理

A Human-Inspired Reading Agent with Gist Memory of Very Long Contexts

Google DeepMind(谷歌DeepMind)

Kuang-Huei Lee, Xinyun Chen, Hiroki Furuta, John Canny, Ian Fischer

AI总结:

提出ReadAgent,一种模拟人类阅读的LLM代理,通过记忆片段压缩为要点记忆并支持原文查找,将有效上下文长度提升3.5-20倍,在三个长文档阅读理解任务上超越基线。

摘要:

当前的大型语言模型(LLMs)不仅受限于最大上下文长度,而且无法稳健地处理长输入。为解决这些限制,我们提出了ReadAgent,一个LLM代理系统,在我们的实验中可将有效上下文长度提升至20倍。受人类交互式阅读长文档方式的启发,我们将ReadAgent实现为一个简单的提示系统,利用LLM的高级语言能力来(1)决定哪些内容应一起存储在记忆片段中,(2)将这些记忆片段压缩为称为要点记忆的短时情景记忆,以及(3)在ReadAgent需要提醒自己相关细节以完成任务时,采取行动查找原文中的段落。我们使用检索方法、原始长上下文以及要点记忆对ReadAgent进行了基线评估。这些评估在三个长文档阅读理解任务上进行:QuALITY、NarrativeQA和QMSum。ReadAgent在所有三个任务上均优于基线,同时将有效上下文窗口扩展了3.5至20倍。

英文摘要:

Current Large Language Models (LLMs) are not only limited to some maximum context length, but also are not able to robustly consume long inputs. To address these limitations, we propose ReadAgent, an LLM agent system that increases effective context length up to 20x in our experiments. Inspired by how humans interactively read long documents, we implement ReadAgent as a simple prompting system that uses the advanced language capabilities of LLMs to (1) decide what content to store together in a memory episode, (2) compress those memory episodes into short episodic memories called gist memories, and (3) take actions to look up passages in the original text if ReadAgent needs to remind itself of relevant details to complete a task. We evaluate ReadAgent against baselines using retrieval methods, using the original long contexts, and using the gist memories. These evaluations are performed on three long-document reading comprehension tasks: QuALITY, NarrativeQA, and QMSum. ReadAgent outperforms the baselines on all three tasks while extending the effective context window by 3.5-20x.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑