← 返回 2026-02-05

PaperSearchQA:使用RLVR训练代理在科学文献上进行搜索和推理 PaperSearchQA: Learning to Search and Reason over Scientific Papers with RLVR

James Burgess, Jan N. Hansen, Duo Peng, Yuhui Zhang, Alejandro Lozano, Min Woo Sun, Emma Lundberg, Serena Yeung-Levy 📅 2026-01-26 👍 19 2026-07-13 08:35
RLVR 强化学习 搜索代理 检索增强生成 科学问答

提出科学文献问答环境,用RLVR训练搜索代理在1600万篇PubMed摘要上回答事实型问题

前置知识

强化学习与可验证奖励(RLVR)

RLVR是一种后训练方法,语言模型被要求回答查询,只有当自动验证器判定最终输出正确时才给予奖励。与监督微调(SFT)不同,RLVR不依赖人类标注的中间推理步骤或偏好分数,而是纯粹基于最终结果的正确性进行学习。这种方法通过'生成-验证-强化'循环来优化模型,使得模型能够学习更灵活和通用的推理策略。

本文的核心训练方法就是基于RLVR,理解这个范式对于把握论文的技术路线至关重要。

搜索代理(Search Agents)

搜索代理是能够交替进行推理和检索的语言模型系统。给定一个问题,代理可以生成推理步骤,然后输出搜索查询来检索相关文档,将检索到的信息加入上下文后继续推理,直到得出最终答案。这种交错式的推理-检索模式使得代理能够动态地获取和利用外部知识。

本文研究的就是如何用RLVR训练这样的搜索代理来处理科学文献问答任务。

Group Relative Policy Optimization(GRPO)

GRPO是一种策略梯度优化算法,它是本文使用的具体RL训练方法。对于每个问题,GRPO使用当前策略生成多个回答(rollouts),然后计算组内相对优势(advantage),通过归一化组内奖励来减少方差。该算法还包含裁剪机制和KL散度惩罚,以防止策略偏离参考模型太远。

理解GRPO的工作原理有助于把握论文中RL训练的具体实现细节。

事实型问答(Factoid QA)

事实型问答是指答案为单一实体的问答类型,例如'什么基因突变导致镰状细胞贫血?'的答案是'HBB'。这种问答类型的特点是答案明确、可验证,适合用精确匹配来评估。与之相对的是二元问答(是/否)、列表型问答和摘要型问答。

本文专注于事实型问答,因为这种类型适合当前的RLVR训练,答案可以无歧义地验证。

BM25与语义检索

BM25是一种基于词频的经典检索算法,通过统计查询词在文档中的出现频率来计算相关性。语义检索(如e5)则使用神经网络将文本编码为向量,通过向量相似度来衡量语义相关性。理论上语义检索能更好地处理同义词和改写,但在本文的实验中两者效果差异不大。

论文对比了这两种检索方式在科学文献问答中的表现,发现语义检索收益有限。

研究动机

当前使用RLVR训练的搜索代理主要关注通用领域问答,如TriviaQA、Natural Questions等简单常识问题,这些测试的是简单的事实检索而非深度技术推理。然而,科学、工程、法律和医学等领域需要更专业的技术知识、对复杂系统的推理能力,以及搜索技术知识库的能力。现有的搜索代理研究(如Search-R1、R1-Searcher)都强调通用知识问答,没有专门针对科学文献理解进行训练和评估。这导致训练出的代理缺乏处理专业领域问题的能力,无法满足实际科学研究中对文献检索和理解的需求。

本文的目标是本文的具体目标是创建一个专门用于训练科学文献搜索代理的环境,包括一个包含1600万篇PubMed生物医学论文摘要的检索语料库、一个包含60k样本的事实型QA数据集(PaperSearchQA),以及相应的评估基准。作者希望通过这个环境验证RLVR训练能否提升代理在科学文献问答上的性能,并分析代理在训练过程中学到的行为模式。

与已有工作不同的是,本文的独特切入角度在于:第一,专注于科学文献而非通用知识,这直接关联到实际的AI科学家系统需求;第二,强调可验证性,所有QA都是事实型的,答案可以无歧义地验证,这使得RLVR训练可行;第三,关注代理行为的可解释性,通过分析推理轨迹来理解代理学到了什么策略。作者指出,未来的AI科学家系统将需要知识密集型搜索能力,而本文正是为这个方向奠定基础。

核心方法

本文的方法可以类比为训练一个'科学文献检索专家'。就像一个新入职的研究助理需要学会如何提问、搜索文献、理解内容并给出答案一样,本文通过强化学习来训练语言模型掌握这些技能。技术路线分为三个主要部分:首先构建训练数据(从PubMed摘要生成QA对),然后准备检索环境(1600万篇摘要的语料库和索引),最后使用GRPO算法训练搜索代理。整个过程中,代理通过试错学习:尝试不同的搜索策略,只有当最终答案正确时才获得奖励。

本文的核心创新点是将RLVR训练范式从通用领域问答扩展到科学文献问答。与已有方法的本质区别在于:第一,训练环境专门针对科学领域设计,使用真实的生物医学文献而非通用知识库;第二,数据生成流程经过领域专家严格审核,确保问题对实际科学家有用;第三,通过最小化的系统提示设计,让代理在RL训练中自主学习推理和查询策略,而不是依赖人工设计的规则。这种设计使得代理能够学到更灵活、更通用的行为模式。

方法步骤详情

方法分为四个主要步骤:第一步是数据集构建,包括定义10个问题类别(如基因突变、治疗方法、蛋白质功能等),从PubMed摘要中随机采样,使用GPT-4.1生成3个问题,经过质量检查后生成答案同义词,并对50%的问题进行改写以避免关键词匹配。第二步是检索环境准备,使用1600万篇PubMed摘要构建语料库,建立BM25和e5两种检索索引。第三步是训练设置,使用最小化的系统提示,让代理通过``标签进行推理,``标签发起搜索,``标签给出答案。第四步是RLVR训练,使用GRPO算法,批量大小512,小批量大小256,训练150步,使用8张A100 GPU。

技术新颖性

本文的技术新颖性体现在三个方面:首先,这是第一个专门针对科学文献问答的RLVR训练环境,填补了通用领域和专业领域之间的空白;其次,数据生成流程结合了LLM自动化和人类专家审核,既保证了规模(60k样本)又确保了质量;最后,通过最小化提示设计,让代理在训练中自主发现有效的推理策略,这比人工设计的规则更具扩展性。作者还发现,训练后的代理会展现出规划、推理和自我验证等有趣行为,这些都不是显式编程进去的。

搜索代理在问答中交替进行推理和检索
Figure 1: 搜索代理在问答中交替进行推理和检索
数据生成流程
Figure 3: 数据生成流程

实验结果

实验结果显示RLVR训练确实能提升科学文献问答性能,但整体得分仍然较低,表明该数据集具有挑战性。具体来看,在PaperSearchQA测试集上,3B模型的直接推理得分为16.7%,思维链推理为20.3%,RAG为32.0%,而经过RLVR训练的Search-R1达到41.6%,比RAG高出9.6个百分点。7B模型上,直接推理27.5%,RAG 36.5%,Search-R1达到51.0%,比RAG高出14.5个百分点。在BioASQ评估集上也观察到类似趋势。有趣的是,语义检索器(e5)相比语法检索器(BM25)的收益很小,在所有实验中差异都在2分以内。此外,无检索的LLM基线得分(15-30%)表明模型在预训练中已经记住了部分科学知识,但远非完美。训练动态分析显示,基础模型和指令调优模型的性能差异不大,但基础模型需要更多训练时间收敛。

基线方法与Search-R1训练的主要结果对比
Table 1: 基线方法与Search-R1训练的主要结果对比
按问题类别划分的主要结果
Table 2: 按问题类别划分的主要结果
查看结构化数据
任务指标本文基线提升
PaperSearchQA测试集 准确率(精确匹配) 51.0%(7B模型) 36.5%(RAG基线) +14.5个百分点
PaperSearchQA测试集 准确率(精确匹配) 41.6%(3B模型) 32.0%(RAG基线) +9.6个百分点
BioASQ事实型问答 准确率(精确匹配) 44.8%(7B模型) 29.7%(RAG基线) +9.3个百分点
问题类别分析 各类别准确率 生物标记物与诊断类最高(56%) 基因突变类最低(26%) 类别间差异显著

局限与改进

论文承认的主要局限包括:第一,数据生成流程是自动化的,使用LLM可能导致事实错误的QA对,尽管风险较小因为使用了强模型(GPT-4.1);第二,从单一摘要推断'通用QA'具有挑战性,例如某个摘要声称'基因X突变与疾病Y相关',但可能还有其他基因也相关,而我们的流程无法检测这种冲突;第三,范围限制在事实型QA,这只是一种问题类型,未来需要扩展到多跳问答、列表答案、长篇摘要等更复杂的类型;第四,仅覆盖生物医学领域,未涉及化学、材料科学、计算机科学等领域;第五,仅处理文本,未考虑图像等多模态信息。从我的观察来看,整体准确率仍然较低(最高约50%),说明科学文献问答本身就是一个困难任务,当前方法还有很大提升空间。

独立分析的弱点

从独立分析来看,本文存在几个关键弱点:第一,数据生成的可验证性假设过于严格,要求答案必须是单一实体且无歧义,这限制了问题类型的多样性,实际科学问题往往更复杂;第二,检索收益有限,语义检索相比语法检索几乎没有优势,可能是因为科学术语的专业性使得语义相似度计算不准确,或者检索模型本身在专业领域表现不佳;第三,代理行为趋于单一化,随着训练进行,代理越来越依赖简单的关键词提取和搜索策略,缺乏更复杂的推理模式;第四,评估仅基于精确匹配,未考虑答案的部分正确性或语义等价性。改进方向包括:开发更灵活的验证机制支持复杂答案类型,设计专门针对科学领域的检索模型,在训练数据中引入更多样化的问题类型以鼓励复杂行为,以及开发更精细的评估指标。

未来方向

作者提出和可延伸的未来方向包括:第一,扩展到多跳QA,需要代理从多个文档中综合信息;第二,支持列表型答案和长篇摘要;第三,处理证据冲突,如系统性文献综述中的矛盾结果;第四,引入LLM-as-judge进行更灵活的奖励建模;第五,结合图像和多模态信息,因为科学论文包含大量图表;第六,添加科学家实际使用的工具和元数据,如引用遍历、来源可靠性指标;第七,将数据生成流程扩展到其他科学领域如化学、材料科学;第八,开发评估检索信息质量的指标,判断结论是否得到论文数据的支持。

复现评估

复现评估显示本文的可复现性较好。数据和代码完全开源:PaperSearchQA数据集在Hugging Face Hub上发布,使用CC-BY-4.0许可证;代码在GitHub上公开;检索语料库和索引也可获取。训练资源方面,需要8张A100 GPU(80GB显存)训练约30小时,这是中等规模的算力需求。数据生成成本约600美元(主要是OpenRouter API费用)。复现难度中等,主要挑战在于设置检索环境和调优训练超参数。作者提供了完整的训练脚本和超参数配置,降低了复现门槛。不过,需要注意BioASQ评估集有其自己的许可证(CC-BY-2.5),且原始PubMed摘要需要遵守NLM的使用条款。