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

RA-DIT:检索增强的双重指令微调

RA-DIT: Retrieval-Augmented Dual Instruction Tuning

FAIR at Meta(Meta FAIR实验室)

Xi Victoria Lin, Xilun Chen, Mingda Chen, Weijia Shi, Maria Lomeli, Rich James, Pedro Rodriguez, Jacob Kahn, Gergely Szilvasy, Mike Lewis, Luke Zettlemoyer, Scott Yih

AI总结:

本文提出RA-DIT轻量级微调方法,通过分别优化语言模型利用检索信息和检索器返回更相关结果,使任意大语言模型获得检索增强能力,并在知识密集型基准上取得最先进性能。

摘要:

检索增强语言模型(RALMs)通过访问外部数据存储中的长尾知识和最新知识来提升性能,但构建起来颇具挑战。现有方法要么需要对语言模型预训练进行昂贵的检索特定修改,要么采用事后集成数据存储的方式,导致性能欠佳。我们提出了检索增强的双重指令微调(RA-DIT),这是一种轻量级微调方法,通过为任意大语言模型(LLM)加装检索能力提供了第三种选择。我们的方法分为两个不同的微调步骤:(1)更新预训练语言模型以更好地利用检索到的信息;(2)更新检索器以返回更符合语言模型偏好的相关结果。通过在同时需要知识利用和上下文感知的任务上进行微调,我们证明每个阶段都能带来显著的性能提升,而同时使用两个阶段则能获得额外增益。我们最好的模型 RA-DIT 65B 在一系列知识密集型的零样本和少样本学习基准上取得了最先进的性能,在平均情况下,显著优于现有的上下文检索增强语言模型方法,在 0-shot 设置下提升高达 +8.9%,在 5-shot 设置下提升高达 +1.4%。

英文摘要:

Retrieval-augmented language models (RALMs) improve performance by accessing long-tail and up-to-date knowledge from external data stores, but are challenging to build. Existing approaches require either expensive retrieval-specific modifications to LM pre-training or use post-hoc integration of the data store that leads to suboptimal performance. We introduce Retrieval-Augmented Dual Instruction Tuning (RA-DIT), a lightweight fine-tuning methodology that provides a third option by retrofitting any LLM with retrieval capabilities. Our approach operates in two distinct fine-tuning steps: (1) one updates a pre-trained LM to better use retrieved information, while (2) the other updates the retriever to return more relevant results, as preferred by the LM. By fine-tuning over tasks that require both knowledge utilization and contextual awareness, we demonstrate that each stage yields significant performance improvements, and using both leads to additional gains. Our best model, RA-DIT 65B, achieves state-of-the-art performance across a range of knowledge-intensive zero- and few-shot learning benchmarks, significantly outperforming existing in-context RALM approaches by up to +8.9% in 0-shot setting and +1.4% in 5-shot setting on average.

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

↑