arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2402.17010cs.CLcs.AI
收藏

MindRef:模拟人类记忆的细粒度位置感知分层参考检索

MindRef: Mimicking Human Memory for Hierarchical Reference Retrieval with Fine-Grained Location Awareness

Renmin University of China(中国人民大学);University of Chinese Academy of Sciences(中国科学院大学);Institute of Software, Chinese Academy of Sciences(中国科学院软件研究所)

Ye Wang, Xinrun Xu, Zhiming Ding

AI总结:

本文提出MindRef两阶段框架,利用大语言模型预训练参数化知识,通过先回忆文档标题再回忆细粒度段落的方式实现分层参考检索,并在KILT任务上验证其能独立定位参考段落并有效辅助下游任务。

摘要:

在完成知识密集型任务时,人类有时需要一个答案以及相应的参考段落用于辅助阅读。以往的方法需要通过额外的检索模型来获取预先切分好的文章块。本文探索利用大语言模型(LLMs)预训练阶段存储的参数化知识,从任意起始位置独立地回忆参考段落。我们提出了一个两阶段框架,模拟人类回忆易忘参考文献的场景。首先,提示大语言模型回忆文档标题标识符,以获得粗粒度的文档集合。然后,基于获得的粗粒度文档集合,回忆细粒度的段落。在两阶段回忆过程中,我们使用受限解码来确保不会生成存储文档之外的内容。为了提高速度,我们在第二阶段只回忆一个短前缀,然后定位其位置以检索完整段落。在KILT知识敏感任务上的实验验证了LLMs能够在各种任务形式中独立回忆参考段落位置,并且获得的参考文献显著辅助了下游任务。

英文摘要:

When completing knowledge-intensive tasks, humans sometimes need an answer and a corresponding reference passage for auxiliary reading. Previous methods required obtaining pre-segmented article chunks through additional retrieval models. This paper explores leveraging the parameterized knowledge stored during the pre-training phase of large language models (LLMs) to recall reference passage from any starting position independently. We propose a two-stage framework that simulates the scenario of humans recalling easily forgotten references. Initially, the LLM is prompted to recall document title identifiers to obtain a coarse-grained document set. Then, based on the acquired coarse-grained document set, it recalls fine-grained passage. In the two-stage recall process, we use constrained decoding to ensure that content outside of the stored documents is not generated. To increase speed, we only recall a short prefix in the second stage, and then locate its position to retrieve a complete passage. Experiments on KILT knowledge-sensitive tasks have verified that LLMs can independently recall reference passage locations in various task forms, and the obtained reference significantly assists downstream tasks.

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

↑