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

面向冷门知识的微调与检索增强生成比较

Fine Tuning vs. Retrieval Augmented Generation for Less Popular Knowledge

Radboud University(拉德堡德大学);University of Amsterdam(阿姆斯特丹大学)

Heydar Soudani, Evangelos Kanoulas, Faegheh Hasibi

AI总结:

本文比较 RAG 与合成数据微调对语言模型掌握冷门知识的效果,发现 RAG 对最低频事实知识优势显著,并提出成本更低且优于微调的 Stimulus RAG。

摘要:

语言模型(LM)会记忆大量事实性知识,并在多种任务和领域中表现出强劲性能。然而,人们观察到,在处理不太流行或低频的概念与实体时(例如在领域特定应用中),其性能会下降。提升 LM 在低频主题上性能的两种突出方法是:检索增强生成(RAG)和基于合成数据的微调(FT)。本文探索并评估了 RAG 与 FT 在定制 LM 以处理问答任务中低频实体方面的影响。我们对 12 个不同规模和类型的 LM,以及不同的微调、数据增强和检索模型进行了大量实验。研究结果表明,尽管 FT 提升了模型在不同流行度实体上的性能,但 RAG 大幅超越 FT,尤其是在最不流行的事实性知识方面。此外,改进检索和数据增强技术会放大 RAG 与 FT 两种方法的成功效果。微调虽然有益于小型 LM,但需要大量资源。为解决这一问题,我们提出了新的 Stimulus RAG 方法,其效果超越了基于微调的方法,从而无需为丰富 LM 的冷门事实性知识而执行代价高昂的数据增强和微调步骤。代码可在 https://github.com/informagi/RAGvsFT 获取。

英文摘要:

Language Models (LMs) memorize a vast amount of factual knowledge, exhibiting strong performance across diverse tasks and domains. However, it has been observed that the performance diminishes when dealing with less-popular or low-frequency concepts and entities, for example in domain specific applications. The two prominent approaches to enhance the performance of LMs on low-frequent topics are: Retrieval Augmented Generation (RAG) and fine-tuning (FT) over synthetic data. This paper explores and evaluates the impact of RAG and FT on customizing LMs in handling low-frequency entities on question answering tasks. We conduct extensive experiments on twelve LMs of varying size and type and different fine tuning, data augmentation, and retrieval models. Our findings indicate that while FT boosts the performance across entities of varying popularity, RAG surpasses FT by a large margin particularly for least popular factual knowledge. Additionally, the success of both RAG and FT approaches is amplified by improving retrieval and data augmentation techniques. Fine tuning, while beneficial for small LMs, requires extensive resources. To address this issue, we propose the new Stimulus RAG approach that surpasses the effectiveness of fine tuning based approaches, thereby eliminating the need for the costly data augmentation and fine tuning step for enriching LMs with less popular factual knowledge. The code is available at \url{https://github.com/informagi/RAGvsFT}.

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

↑