用思维链提升大模型对错误检索样本的免疫力
VaccineRAG: Boosting Multimodal Large Language Models' Immunity to Harmful RAG Samples
- 设计新数据集,让模型先分析每个检索样本的合理性再回答
- 在不同干扰比例下测试,发现现有模型极易被误导
- 提出部分强化学习算法,更好学习复杂推理过程
检索增强生成(RAG)通过结合外部知识提升大语言模型(LLM)的响应准确率,在实时查询和视觉问答任务中表现突出。然而,其效果常受限于检索器的精度:大量无关或误导性样本被引入生成阶段,成为性能瓶颈。为此,我们提出VaccineRAG,一种基于思维链(CoT)的检索增强生成数据集。一方面,该数据集通过不同正负样本比例的基准测试,系统暴露当前模型的内在弱点;另一方面,通过引导模型在生成最终答案前对每个样本进行显式的思维链分析,增强其样本辨别能力。此外,为提升模型对长序列复杂思维链内容的学习能力,我们提出部分强化策略(Partial-GRPO),将模型输出视为多个组件而非单一整体,从而实现更精准的偏好选择,增强对复杂推理序列的学习能力。在VaccineRAG上的全面评估与消融实验验证了该方案的有效性。代码与数据集即将开源。
原文摘要 · Abstract (English)
Retrieval Augmented Generation enhances the response accuracy of Large Language Models (LLMs) by integrating retrieval and generation modules with external knowledge, demonstrating particular strength in real-time queries and Visual Question Answering tasks. However, the effectiveness of RAG is frequently hindered by the precision of the retriever: many retrieved samples fed into the generation phase are irrelevant or misleading, posing a critical bottleneck to LLMs' performance. To address this challenge, we introduce VaccineRAG, a novel Chain-of-Thought-based retrieval-augmented generation dataset. On one hand, VaccineRAG employs a benchmark to evaluate models using data with varying positive/negative sample ratios, systematically exposing inherent weaknesses in current LLMs. On the other hand, it enhances models' sample-discrimination capabilities by prompting LLMs to generate explicit Chain-of-Thought (CoT) analysis for each sample before producing final answers. Furthermore, to enhance the model's ability to learn long-sequence complex CoT content, we propose Partial-GRPO. By modeling the outputs of LLMs as multiple components rather than a single whole, our model can make more informed preference selections for complex sequences, thereby enhancing its capacity to learn complex CoT. Comprehensive evaluations and ablation studies on VaccineRAG validate the effectiveness of the proposed scheme. The code and dataset will be publicly released soon.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。