通过多视角偏好优化,提升RAG生成器的准确性与可靠性
PA-RAG: RAG Alignment via Multi-Perspective Preference Optimization
- 构建多场景响应数据,从不同文档质量中采样生成质量差异
- 结合SFT与直接偏好优化,显著提升生成信息量与引用质量
- 适用于希望改进RAG生成效果的研究者与开发者
检索增强生成(RAG)缓解了大语言模型(LLM)内容过时与幻觉问题,但仍存在响应信息量不足、鲁棒性差及引用质量低等局限。现有方法或增加额外步骤,或通过监督微调(SFT)优化生成器,未能充分对齐RAG需求。为此,我们提出PA-RAG:一种面向检索增强生成的多视角偏好对齐方法。通过在不同提示文档质量场景下采样生成器的多样化响应,构建高质量指令微调与多视角偏好数据集,并采用SFT与直接偏好优化(DPO)联合训练生成器。在三个LLM和四个问答数据集上的实验表明,PA-RAG能显著提升RAG生成器性能。代码与数据集已开源。
原文摘要 · Abstract (English)
The emergence of Retrieval-augmented generation (RAG) has alleviated the issues of outdated and hallucinatory content in the generation of large language models (LLMs), yet it still reveals numerous limitations. When a general-purpose LLM serves as the RAG generator, it often suffers from inadequate response informativeness, response robustness, and citation quality. Past approaches to tackle these limitations, either by incorporating additional steps beyond generating responses or optimizing the generator through supervised fine-tuning (SFT), still failed to align with the RAG requirement thoroughly. Consequently, optimizing the RAG generator from multiple preference perspectives while maintaining its end-to-end LLM form remains a challenge. To bridge this gap, we propose Multiple Perspective Preference Alignment for Retrieval-Augmented Generation (PA-RAG), a method for optimizing the generator of RAG systems to align with RAG requirements comprehensively. Specifically, we construct high-quality instruction fine-tuning data and multi-perspective preference data by sampling varied quality responses from the generator across different prompt documents quality scenarios. Subsequently, we optimize the generator using SFT and Direct Preference Optimization (DPO). Extensive experiments conducted on four question-answer datasets across three LLMs demonstrate that PA-RAG can significantly enhance the performance of RAG generators. Our code and datasets are available at https://github.com/wujwyi/PA-RAG.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。