arXiv:2603.04406cs.CLcs.AI2026-03被引 1

用对比似然奖励提升RAG模型对上下文的忠实度

CTRL-RAG: Contrastive Likelihood Reward Based Reinforcement Learning for Context-Faithful RAG Models

  • 设计内部对比似然奖励,直接优化有无证据时的响应差异
  • 在单跳、多跳、垂直领域任务上均显著提升忠实性表现
  • 适合需要高可信生成的场景,如医疗、法律问答

随着检索增强生成(RAG)应用增多,训练大语言模型进行上下文敏感推理与忠实生成愈发重要。现有RAG导向的强化学习方法依赖外部奖励,难以评估文档忠实性,且在开放域中易误判相似答案。此外,缺乏基于RAG的自奖励机制。尽管该机制理论上可估计答案置信度,但自判断缺乏客观反馈,易导致幻觉累积并最终模型崩溃。为此,我们提出一种以对比似然奖励(CLR)为核心的“内外结合”奖励框架。CLR直接优化在有无支持证据条件下响应的对数似然差距,促使模型提取相关证据,并在特定上下文中提高信心。实验表明,该方法(单独使用或结合外部正确性奖励)在单跳、多跳、垂直领域及忠实性基准上均表现优异。训练代码与模型即将发布。

原文摘要 · Abstract (English)

With the growing use of Retrieval-Augmented Generation (RAG), training large language models (LLMs) for context-sensitive reasoning and faithfulness is increasingly important. Existing RAG-oriented reinforcement learning (RL) methods rely on external rewards that often fail to evaluate document faithfulness, and may misjudge similar answers in open-domain settings. In addition, there is no RAG-based selfreward mechanism. Moreover, although such a mechanism could in principle estimate answer confidence given documents, the absence of objective feedback in a self-judgment can cause hallucination accumulation and eventual model collapse. To tackle these issues, we propose a novel "internal-external" hybrid reward framework centered on a Contrastive Likelihood Reward (CLR). CLR directly optimizes the log-likelihood gap between responses conditioned on prompts with and without supporting evidence. This encourages the model to extract relevant evidence and increases its confidence when grounded in a specific context. Experiments show that our method (used alone or combined with external correctness rewards) achieves strong performance on singlehop, multi-hop, vertical-domain, and faithfulness benchmarks. Our training code and models are coming soon.

RAG强化学习忠实性大模型

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