arXiv:2504.13950cs.LGcs.AI2025-04被引 5

医学领域用强化学习提升模型推理,选对数据比调模型更重要。

Open-Medical-R1: How to Choose Data for RLVR Training at Medicine Domain

  • 用大模型筛选医学题数据,比随机选更有效。
  • 自过滤数据在医学任务上表现最好,但泛化能力弱。
  • 用更大同系列模型过滤数据,整体性能更稳。

本文研究了在医疗领域使用带验证奖励的强化学习(RLVR)训练时,最优数据选择策略。尽管RLVR在提升大模型推理能力方面潜力巨大,但以往研究主要集中在数学和逻辑谜题,对医疗等专业领域的探索有限。我们从MedQA-USMLE数据集中采用四种采样策略:随机采样(基线),以及利用Phi-4、Gemma-3-27b-it和Gemma-3-12b-it模型进行过滤。以Gemma-3-12b-it为基线模型,结合组相对策略优化(GRPO),在MMLU、GSM8K、MMLU-Pro和CMMLU等多个基准上评估性能。结果表明,经过滤的数据训练的模型普遍优于随机采样训练的模型。特别地,使用Gemma-3-12b-it自身进行数据过滤,在医疗任务中表现最优,但在其他基准上鲁棒性较差;而使用同系列更大模型进行过滤,则展现出更好的整体鲁棒性。研究为专业领域中RLVR的数据组织提供了重要参考,强调了数据选择的关键作用。代码已开源:https://github.com/Qsingle/open-medical-r1。

原文摘要 · Abstract (English)

This paper explores optimal data selection strategies for Reinforcement Learning with Verified Rewards (RLVR) training in the medical domain. While RLVR has shown exceptional potential for enhancing reasoning capabilities in large language models, most prior implementations have focused on mathematics and logical puzzles, with limited exploration of domain-specific applications like medicine. We investigate four distinct data sampling strategies from MedQA-USMLE: random sampling (baseline), and filtering using Phi-4, Gemma-3-27b-it, and Gemma-3-12b-it models. Using Gemma-3-12b-it as our base model and implementing Group Relative Policy Optimization (GRPO), we evaluate performance across multiple benchmarks including MMLU, GSM8K, MMLU-Pro, and CMMLU. Our findings demonstrate that models trained on filtered data generally outperform those trained on randomly selected samples. Notably, training on self-filtered samples (using Gemma-3-12b-it for filtering) achieved superior performance in medical domains but showed reduced robustness across different benchmarks, while filtering with larger models from the same series yielded better overall robustness. These results provide valuable insights into effective data organization strategies for RLVR in specialized domains and highlight the importance of thoughtful data selection in achieving optimal performance. You can access our repository (https://github.com/Qsingle/open-medical-r1) to get the codes.

强化学习医疗AI数据筛选模型训练

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