LLM Reasoning 相关度: 9/10

Eliciting Medical Reasoning with Knowledge-enhanced Data Synthesis: A Semi-Supervised Reinforcement Learning Approach

Haolin Li, Shuyang Jiang, Ruipeng Zhang, Jiangchao Yao, Ya Zhang, Yanfeng Wang
arXiv: 2604.11547v1 发布: 2026-04-13 更新: 2026-04-13

AI 摘要

MedSSR通过知识增强数据合成和半监督强化学习,提升LLM在罕见疾病等医学推理任务上的性能。

主要贡献

  • 提出MedSSR框架,结合知识增强数据合成和半监督强化学习
  • 利用罕见疾病知识合成可控分布的推理问题
  • 使用策略模型自身生成高质量伪标签,实现高效训练

方法论

使用罕见病知识合成数据,再用策略模型生成伪标签,进行自监督RL,最后在真实数据上进行监督RL。

原文摘要

While large language models hold promise for complex medical applications, their development is hindered by the scarcity of high-quality reasoning data. To address this issue, existing approaches typically distill chain-of-thought reasoning traces from large proprietary models via supervised fine-tuning, then conduct reinforcement learning (RL). These methods exhibit limited improvement on underrepresented domains like rare diseases while incurring substantial costs from generating complex reasoning chains. To efficiently enhance medical reasoning, we propose MedSSR, a Medical Knowledge-enhanced data Synthesis and Semi-supervised Reinforcement learning framework. Our framework first employs rare disease knowledge to synthesize distribution-controllable reasoning questions. We then utilize the policy model itself to generate high-quality pseudo-labels. This enables a two-stage, intrinsic-to-extrinsic training paradigm: self-supervised RL on the pseudo-labeled synthetic data, followed by supervised RL on the human-annotated real data. MedSSR scales model training efficiently without relying on costly trace distillation. Extensive experiments on Qwen and Llama demonstrate that our method outperforms existing methods across ten medical benchmarks, achieving up to +5.93% gain on rare-disease tasks. Our code is available at https://github.com/tdlhl/MedSSR.

标签

LLM 医学推理 强化学习 数据合成 罕见疾病

arXiv 分类

cs.LG cs.CL