用强化学习提升大模型推理时的检索与思考质量
From Sufficiency to Reflection: Reinforcement-Guided Thinking Quality in Retrieval-Augmented Reasoning for LLMs
- 引入思考-检索-反思三步流程,多维度奖励引导高质量推理
- 在四个多跳问答数据集上超越现有方法,复杂任务表现更优
- 适合关注大模型推理可靠性与鲁棒性的研究者与开发者
基于强化学习的检索增强生成(RAG)方法可提升大语言模型(LLMs)的推理能力。然而,多数方法仅依赖最终答案奖励,忽视中间推理质量。本文分析现有RAG推理模型,识别出三种主要失败模式:(1) 信息不足,即模型未能检索足够支持;(2) 推理错误,即使信息充足也存在逻辑或内容缺陷;(3) 答案-推理不一致,合理推理链却得出错误答案。为此提出TIRESRAG-R1框架,采用思考-检索-反思流程与多维奖励机制:(1) 充足性奖励鼓励充分检索;(2) 推理质量奖励评估推理合理性与准确性;(3) 反思奖励用于检测并修正错误。同时引入难度感知重加权策略与训练样本过滤,提升复杂任务表现。在四个多跳问答数据集上的实验表明,TIRESRAG-R1优于现有RAG方法,并能良好泛化至单跳任务。代码与数据见:https://github.com/probe2/TIRESRAG-R1。
原文摘要 · Abstract (English)
Reinforcement learning-based retrieval-augmented generation (RAG) methods enhance the reasoning abilities of large language models (LLMs). However, most rely only on final-answer rewards, overlooking intermediate reasoning quality. This paper analyzes existing RAG reasoning models and identifies three main failure patterns: (1) information insufficiency, meaning the model fails to retrieve adequate support; (2) faulty reasoning, where logical or content-level flaws appear despite sufficient information; and (3) answer-reasoning inconsistency, where a valid reasoning chain leads to a mismatched final answer. We propose TIRESRAG-R1, a novel framework using a think-retrieve-reflect process and a multi-dimensional reward system to improve reasoning and stability. TIRESRAG-R1 introduces: (1) a sufficiency reward to encourage thorough retrieval; (2) a reasoning quality reward to assess the rationality and accuracy of the reasoning chain; and (3) a reflection reward to detect and revise errors. It also employs a difficulty-aware reweighting strategy and training sample filtering to boost performance on complex tasks. Experiments on four multi-hop QA datasets show that TIRESRAG-R1 outperforms prior RAG methods and generalizes well to single-hop tasks. The code and data are available at: https://github.com/probe2/TIRESRAG-R1.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。