arXivDaily arXiv每日学术速递 周一至周五更新
北美退伍军人节(Veterans Day),12 日无更新
arXiv 2005.11401cs.CLcs.LG
收藏

知识密集型NLP任务的检索增强生成

Retrieval-Augmented Generation for Knowledge-Intensive NLP Tasks

Facebook AI Research(Facebook AI研究院);University College London(伦敦大学学院);New York University(纽约大学)
Patrick Lewis, Ethan Perez, Aleksandra Piktus, Fabio Petroni, Vladimir Karpukhin, Naman Goyal, Heinrich Küttler, Mike Lewis, Wen-tau Yih, Tim Rocktäschel, Sebas… 展开作者

Patrick Lewis, Ethan Perez, Aleksandra Piktus, Fabio Petroni, Vladimir Karpukhin, Naman Goyal, Heinrich Küttler, Mike Lewis, Wen-tau Yih, Tim Rocktäschel, Sebastian Riedel, Douwe Kiela

摘要:

大型预训练的语言模型已被证明可以在其参数中存储事实知识,并在对下游NLP任务进行微调时实现最先进的结果。然而,它们获取和精确操作知识的能力仍然有限,因此在知识密集型任务中,它们的性能落后于特定任务的架构。此外,为他们的决定提供出处和更新他们的世界知识仍然是开放的研究问题。具有显式非参数记忆的可微分访问机制的预训练模型可以克服这个问题,但到目前为止仅针对提取下游任务进行了研究。我们探索一个通用的微调配方检索增强生成(RAG)-模型结合了预先训练的参数和非参数的记忆语言生成。我们引入RAG模型,其中参数记忆是预训练的seq 2seq模型,非参数记忆是维基百科的密集向量索引,使用预训练的神经检索器访问。我们比较了两种RAG公式,一种是在整个生成的序列中以相同的检索通道为条件,另一种是每个标记可以使用不同的通道。我们在广泛的知识密集型NLP任务上微调和评估我们的模型,并在三个开放领域QA任务上设置最先进的模型,性能优于参数seq 2seq模型和特定于任务的检索和提取架构。对于语言生成任务,我们发现RAG模型比最先进的仅参数seq 2seq基线生成更具体,更多样和更真实的语言。

英文摘要:

Large pre-trained language models have been shown to store factual knowledge in their parameters, and achieve state-of-the-art results when fine-tuned on downstream NLP tasks. However, their ability to access and precisely manipulate knowledge is still limited, and hence on knowledge-intensive tasks, their performance lags behind task-specific architectures. Additionally, providing provenance for their decisions and updating their world knowledge remain open research problems. Pre-trained models with a differentiable access mechanism to explicit non-parametric memory can overcome this issue, but have so far been only investigated for extractive downstream tasks. We explore a general-purpose fine-tuning recipe for retrieval-augmented generation (RAG) -- models which combine pre-trained parametric and non-parametric memory for language generation. We introduce RAG models where the parametric memory is a pre-trained seq2seq model and the non-parametric memory is a dense vector index of Wikipedia, accessed with a pre-trained neural retriever. We compare two RAG formulations, one which conditions on the same retrieved passages across the whole generated sequence, the other can use different passages per token. We fine-tune and evaluate our models on a wide range of knowledge-intensive NLP tasks and set the state-of-the-art on three open domain QA tasks, outperforming parametric seq2seq models and task-specific retrieve-and-extract architectures. For language generation tasks, we find that RAG models generate more specific, diverse and factual language than a state-of-the-art parametric-only seq2seq baseline.

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

↑