arXiv:2503.12759cs.CL2025-03Conference of the …被引 33

让大模型学会从大量文本中挑选并引用关键信息,提升问答准确率。

RAG-RL: Advancing Retrieval-Augmented Generation via RL and Curriculum Learning

  • 通过课程学习,先教模型处理简单问题,逐步过渡到复杂场景。
  • 在三个多跳问答数据集上,答案和引用准确率显著提升。
  • 适合想提升生成模型引用能力的研究者和工程师。

检索增强生成(RAG)系统依赖检索模型识别相关上下文,再由生成模型利用这些上下文回答问题。然而,检索器存在召回率和精确率不足的问题,限制了下游性能。我们提出RAG-RL,一种训练方式不仅让生成模型产出答案,还使其能从更大量的检索上下文中识别并引用相关内容,将识别相关文档的负担部分转移到生成模型。该方法采用课程学习策略,模型先在仅含相关上下文的简单样本上训练。实验表明,这类训练样本使模型以更高样本效率和更强泛化能力掌握引用与推理技能,即使在无关段落数量增加时仍表现良好。我们在三个开放域多跳问答数据集上进行了基准测试,报告了答案准确率和引用准确率的显著提升。实验揭示了较易样本如何为模型提供更强的学习信号(如引用生成),以及后训练阶段不同组件(如训练集构建、规则奖励、样本排序等)对最终性能的影响。

原文摘要 · Abstract (English)

Retrieval-augmented generation (RAG) systems rely on retrieval models for identifying relevant contexts and answer generation models for utilizing those contexts. However, retrievers exhibit imperfect recall and precision, limiting downstream performance. We introduce RAG-RL, an answer generation model trained not only to produce answers but also to identify and cite relevant information from larger sets of retrieved contexts, shifting some of the burden of identifying relevant documents from the retriever to the answer generator. Our approach uses curriculum learning, where the model is first trained on easier examples that include only relevant contexts. Our experiments show that these training samples enable models to acquire citation and reasoning skills with greater sample efficiency and generalizability, demonstrating strong model performance even as the number of irrelevant passages increases. We benchmark our methods on three open-domain multi-hop question answering datasets and report significant gains in answer and citation accuracy. Our experiments provide empirical insights into how easier training samples can give models stronger signals for learning specific skills (e.g., citation generation) and how different components of post-training (e.g., training set construction, rule-based rewards, training sample ordering, etc.) impact final model performance.

RAG强化学习课程学习问答系统

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