arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2305.15294cs.CL

利用迭代检索-生成协同增强检索增强型大语言模型

Enhancing Retrieval-Augmented Large Language Models with Iterative Retrieval-Generation Synergy

Tsinghua University(清华大学);Microsoft Research Asia(微软亚洲研究院);Microsoft Azure AI(微软Azure人工智能)

Zhihong Shao, Yeyun Gong, Yelong Shen, Minlie Huang, Nan Duan, Weizhu Chen

AI总结:

本文提出 Iter-RetGen 方法,通过迭代方式协同检索与生成,利用模型输出作为上下文检索相关知识,在多跳问答等任务上以更少开销优于或媲美现有基线。

摘要:

大语言模型是强大的文本处理器和推理器,但仍存在知识过时和幻觉等局限性,这需要将其与世界相连。检索增强型大语言模型因将模型生成建立在外部知识之上而引起了广泛关注。然而,检索器难以捕捉相关性,尤其是对于具有复杂信息需求的查询。最近的工作提出通过让大语言模型积极参与检索来改进相关性建模,即通过生成来改进检索。在本文中,我们展示了一种名为 Iter-RetGen 的方法可以实现出色的性能,该方法以迭代方式协同检索和生成。模型输出展示了完成任务可能需要的内容,从而为检索更多相关知识提供了信息丰富的上下文,这反过来又有助于在下一次迭代中生成更好的输出。与最近在生成输出时将检索与生成交替进行的工作相比,Iter-RetGen 将所有检索到的知识作为一个整体进行处理,并在很大程度上保持了生成的灵活性,没有结构约束。我们在多跳问答、事实验证和常识推理上评估了 Iter-RetGen,结果表明它可以灵活利用参数化知识和非参数化知识,并且在检索和生成开销更少的情况下,优于或与最先进的检索增强基线具有竞争力。我们还可以通过生成增强的检索适应来进一步提升性能。

英文摘要:

Large language models are powerful text processors and reasoners, but are still subject to limitations including outdated knowledge and hallucinations, which necessitates connecting them to the world. Retrieval-augmented large language models have raised extensive attention for grounding model generation on external knowledge. However, retrievers struggle to capture relevance, especially for queries with complex information needs. Recent work has proposed to improve relevance modeling by having large language models actively involved in retrieval, i.e., to improve retrieval with generation. In this paper, we show that strong performance can be achieved by a method we call Iter-RetGen, which synergizes retrieval and generation in an iterative manner. A model output shows what might be needed to finish a task, and thus provides an informative context for retrieving more relevant knowledge which in turn helps generate a better output in the next iteration. Compared with recent work which interleaves retrieval with generation when producing an output, Iter-RetGen processes all retrieved knowledge as a whole and largely preserves the flexibility in generation without structural constraints. We evaluate Iter-RetGen on multi-hop question answering, fact verification, and commonsense reasoning, and show that it can flexibly leverage parametric knowledge and non-parametric knowledge, and is superior to or competitive with state-of-the-art retrieval-augmented baselines while causing fewer overheads of retrieval and generation. We can further improve performance via generation-augmented retrieval adaptation.

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

↑