arXiv:2504.12323cs.CLcs.AI2025-04

小模型用RAG反而更不公平,提出两种方法改善

The Other Side of the Coin: Exploring Fairness in Retrieval-Augmented Generation

  • 让检索器与大模型在公平性上对齐,避免引入偏见
  • 小模型(<8B)用RAG后不公平性加剧,如1B/7B模型
  • 适合关注生成公平性的大模型应用开发者

检索增强生成(RAG)通过引入外部知识源提升大语言模型(LLM)的生成质量,减少幻觉。然而,其在具有社会影响领域的应用引发公平性关注:RAG是否会影响模型公平性?我们通过改变不同LLM、检索器和检索源进行大量实验,发现当模型规模小于80亿时,集成检索机制反而加剧了小模型(如LLaMA3.2-1B、Mistral-7B、LLaMA3-8B)的不公平性。为此,我们提出两种缓解方案:FairFT通过在公平性上对齐检索器与模型,使其检索更公平的文档;FairFilter则在检索后过滤偏见内容。在真实数据集上的验证表明,所提方法有效提升公平性且不牺牲性能。

原文摘要 · Abstract (English)

Retrieval-Augmented Generation (RAG) enhances Large Language Models (LLMs) by retrieving relevant document from external knowledge sources. By referencing this external knowledge, RAG effectively reduces the generation of factually incorrect content and addresses hallucination issues within LLMs. Recently, there has been growing attention to improving the performance and efficiency of RAG systems from various perspectives. While these advancements have yielded significant results, the application of RAG in domains with considerable societal implications raises a critical question about fairness: What impact does the introduction of the RAG paradigm have on the fairness of LLMs? To address this question, we conduct extensive experiments by varying the LLMs, retrievers, and retrieval sources. Our experimental analysis reveals that the scale of the LLMs plays a significant role in influencing fairness outcomes within the RAG framework. When the model scale is smaller than 8B, the integration of retrieval mechanisms often exacerbates unfairness in small-scale LLMs (e.g., LLaMA3.2-1B, Mistral-7B, and LLaMA3-8B). To mitigate the fairness issues introduced by RAG for small-scale LLMs, we propose two approaches, FairFT and FairFilter. Specifically, in FairFT, we align the retriever with the LLM in terms of fairness, enabling it to retrieve documents that facilitate fairer model outputs. In FairFilter, we propose a fairness filtering mechanism to filter out biased content after retrieval. Finally, we validate our proposed approaches on real-world datasets, demonstrating their effectiveness in improving fairness while maintaining performance.

RAG公平性小模型

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。