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。
正文
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