W-RAG: Weakly Supervised Dense Retrieval in RAG for Open-domain Question Answering
This addresses the problem of high annotation costs for retrieval training in RAG systems, offering a weakly supervised alternative that is incremental but practical for improving open-domain QA.
The paper tackles the challenge of training dense retrievers in RAG systems for open-domain question answering by proposing W-RAG, which uses weak supervision from LLM outputs to fine-tune the retriever, achieving retrieval and QA performance comparable to models trained with human-labeled data across four datasets.
In knowledge-intensive tasks such as open-domain question answering (OpenQA), large language models (LLMs) often struggle to generate factual answers, relying solely on their internal (parametric) knowledge. To address this limitation, Retrieval-Augmented Generation (RAG) systems enhance LLMs by retrieving relevant information from external sources, thereby positioning the retriever as a pivotal component. Although dense retrieval demonstrates state-of-the-art performance, its training poses challenges due to the scarcity of ground-truth evidence, largely attributed to the high costs of human annotation. In this paper, we propose W-RAG, a method that draws weak training signals from the downstream task (such as OpenQA) of an LLM, and fine-tunes the retriever to prioritize passages that most benefit the task. Specifically, we rerank the top-$k$ passages retrieved via BM25 by assessing the probability that the LLM will generate the correct answer for a question given each passage. The highest-ranking passages are then used as positive fine-tuning examples for dense retrieval. We conduct comprehensive experiments across four publicly available OpenQA datasets to demonstrate that our approach enhances both retrieval and OpenQA performance compared to baseline models, achieving results comparable to models fine-tuned with human-labeled data.