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

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Béchard, Patrice, Ayala, Orlando Marquez
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866913945321734144
author Béchard, Patrice
Ayala, Orlando Marquez
author_facet Béchard, Patrice
Ayala, Orlando Marquez
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2501_04652
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Multi-task retriever fine-tuning for domain-specific and efficient RAG
Béchard, Patrice
Ayala, Orlando Marquez
Computation and Language
Information Retrieval
Machine Learning
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.
title Multi-task retriever fine-tuning for domain-specific and efficient RAG
topic Computation and Language
Information Retrieval
Machine Learning
url https://arxiv.org/abs/2501.04652