跳到正文
arXiv:cs.AI· Arash Ahmadi, Parisa Masnadi Khiabani, Sarah Sharif, Charles Nicholson, David Ebert, Aswathy K. Vaikom House, Mike Banad·· 5 小时前

用方差感知评分奖励与 GRPO 提升 LLM 心脏医学问答能力

Improving Heart-Focused Medical Question Answering in LLMs via Variance-Aware Rubric Rewards with GRPO

AI 导读

研究用 GRPO 结合 RaR-Medicine 的评分监督,为心脏专科医学问答构建方差感知的连续奖励,保留部分得分并缓解奖励稀疏。基于低秩适配器和量化权重微调的 Qwen3-14B,在留出的心脏相关 HealthBench 子集上准确率从 0.362 提升至 0.502,F1 从 0.532 提升至 0.668。

正文

View PDF HTML (experimental)

Abstract:Large language models can support medical question answering, but privacy, cost, and reliability still limit their clinical use in focused domains. Can rubric-based reinforcement learning bring a locally deployable model close to much larger systems? We study Group Relative Policy Optimization (GRPO) for heart-focused medical question answering with rubric supervision from RaR-Medicine. Rather than rely on binary rubric aggregation or a single overall score, our variance-aware reward framework builds continuous rewards from criterion-level outcomes. It preserves partial credit and reduces reward sparsity. We adapt a Qwen3-14B policy with low-rank adapters and quantized weights. On a held-out heart-related HealthBench subset, the best GRPO variant improves accuracy from 0.362 to 0.502 and F1 from 0.532 to 0.668 over the base model, close to GPT-OSS-120B. An independent clinician rated most responses clinically acceptable (61%) and patient ready (63%), both above the automated accuracy; fewer than 2% were unsafe. Taken together, the results support rubric-based reinforcement learning as a practical route for focused medical language models.
Comments: 29 pages main text plus 30 pages Supplementary Information
Subjects: Computation and Language (cs.CL); Artificial Intelligence (cs.AI)
Cite as: arXiv:2606.05174 [cs.CL]
  (or arXiv:2606.05174v2 [cs.CL] for this version)
  https://doi.org/10.48550/arXiv.2606.05174

arXiv-issued DOI via DataCite

Submission history

From: Yaser Banad [view email]
[v1] Fri, 17 Apr 2026 03:08:55 UTC (746 KB)
[v2] Tue, 13 Oct 2026 01:33:59 UTC (1,562 KB)

来源:arXiv:cs.AI · arxiv.org