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

面向特定领域且高效的RAG的多任务检索器微调

Multi-task retriever fine-tuning for domain-specific and efficient RAG

ServiceNow

Patrice Béchard, Orlando Marquez Ayala

AI总结:

本文提出对小型检索器编码器进行多任务指令微调,使单个编码器能服务多个特定领域RAG应用,在真实企业场景中实现低成本、可扩展且可泛化到未见任务的检索。

摘要:

检索增强生成(RAG)在部署大语言模型(LLM)时已变得无处不在,因为它可以解决生成幻觉或过时信息等典型局限。然而,在构建真实世界的RAG应用时,会出现实际问题。首先,检索到的信息通常是特定领域的。由于微调LLM的计算成本高昂,更可行的做法是微调检索器,以提高LLM输入中所包含数据的质量。其次,随着更多应用部署在同一真实世界系统中,无法承担为每个应用部署独立检索器的成本。此外,这些RAG应用通常会检索不同类型的数据。我们的解决方案是对一个小型检索器编码器在多种特定领域任务上进行指令微调,从而允许我们部署一个可以服务众多用例的编码器,实现低成本、可扩展性和高速度。我们展示了该编码器如何泛化到域外设置,以及在真实世界企业用例中泛化到一个未见过的检索任务。

英文摘要:

Retrieval-Augmented Generation (RAG) has become ubiquitous when deploying Large Language Models (LLMs), as it can address typical limitations such as generating hallucinated or outdated information. However, when building real-world RAG applications, practical issues arise. First, the retrieved information is generally domain-specific. Since it is computationally expensive to fine-tune LLMs, it is more feasible to fine-tune the retriever to improve the quality of the data included in the LLM input. Second, as more applications are deployed in the same real-world system, one cannot afford to deploy separate retrievers. Moreover, these RAG applications normally retrieve different kinds of data. Our solution is to instruction fine-tune a small retriever encoder on a variety of domain-specific tasks to allow us to deploy one encoder that can serve many use cases, thereby achieving low-cost, scalability, and speed. We show how this encoder generalizes to out-of-domain settings as well as to an unseen retrieval task on real-world enterprise use cases.

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

↑