训练模型在1600万篇医学论文中搜索并推理,回答科学问题。
PaperSearchQA: Learning to Search and Reason over Scientific Papers with RLVR
- 用强化学习让模型在论文库中检索并推理答案
- 构建了含6万样本的PaperSearchQA数据集,可验证答案正确性
- 适合科研人员和未来AI科学家系统研究使用
搜索代理是能够通过语言模型在知识库或网络中检索并推理以回答问题的系统;近期方法仅使用可验证奖励(RLVR)进行强化学习,以监督最终答案的准确性。大多数RLVR搜索代理针对通用领域问答,限制了其在科学、工程和医学等技术领域的相关性。本文提出在科学论文中训练搜索与推理代理——这测试了技术性问答能力,直接服务于真实科学家,并为未来人工智能科学家系统提供关键支持。具体而言,我们发布了包含1600万篇生物医学论文摘要的搜索语料库,构建了一个名为PaperSearchQA的挑战性事实型问答数据集,包含6万个可从语料库中回答的样本,并提供了基准测试。我们在该环境中训练搜索代理,使其性能优于非强化学习的检索基线;还进行了定量分析,观察到代理展现出规划、推理和自我验证等有趣行为。我们的语料库、数据集和基准可与流行的Search-R1代码库结合使用,已发布于https://huggingface.co/collections/jmhb/papersearchqa。此外,我们的数据生成方法具有可扩展性,可轻松推广至其他科学领域。
原文摘要 · Abstract (English)
Search agents are language models (LMs) that reason and search knowledge bases (or the web) to answer questions; recent methods supervise only the final answer accuracy using reinforcement learning with verifiable rewards (RLVR). Most RLVR search agents tackle general-domain QA, which limits their relevance to technical AI systems in science, engineering, and medicine. In this work we propose training agents to search and reason over scientific papers -- this tests technical question-answering, it is directly relevant to real scientists, and the capabilities will be crucial to future AI Scientist systems. Concretely, we release a search corpus of 16 million biomedical paper abstracts and construct a challenging factoid QA dataset called PaperSearchQA with 60k samples answerable from the corpus, along with benchmarks. We train search agents in this environment to outperform non-RL retrieval baselines; we also perform further quantitative analysis and observe interesting agent behaviors like planning, reasoning, and self-verification. Our corpus, datasets, and benchmarks are usable with the popular Search-R1 codebase for RLVR training and released on https://huggingface.co/collections/jmhb/papersearchqa. Finally, our data creation methods are scalable and easily extendable to other scientific domains.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。