ProMedical: Hierarchical Fine-Grained Criteria Modeling for Medical LLM Alignment via Explicit Injection

He Geng, Yangmin Huang, Lixian Lai, Qianyun Du, Hui Chu +3 more
4/9/2026
cs.AI

No official code found. Best implementation: genghe02/ProMedical (1 star, last commit 2026-04-14).

Abstract

Aligning Large Language Models (LLMs) with high-stakes medical standards remains a significant challenge, primarily due to the dissonance between coarse-grained preference signals and the complex, multi-dimensional nature of clinical protocols. To bridge this gap, we introduce ProMedical, a unified alignment framework grounded in fine-grained clinical criteria. We first construct ProMedical-Preference-50k, a dataset generated via a human-in-the-loop pipeline that augments medical instructions with rigorous, physician-derived rubrics. Leveraging this corpus, we propose the Explicit Criteria Injection paradigm to train a multi-dimensional reward model. Unlike traditional scalar reward models, our approach explicitly disentangles safety constraints from general proficiency, enabling precise guidance during reinforcement learning. To rigorously validate this framework, we establish ProMedical-Bench, a held-out evaluation suite anchored by double-blind expert adjudication. Empirical evaluations demonstrate that optimizing the Qwen3-8B base model via ProMedical-RM-guided GRPO yields substantial gains, improving overall accuracy by 22.3% and safety compliance by 21.7%, effectively rivaling proprietary frontier models. Furthermore, the aligned policy generalizes robustly to external benchmarks, demonstrating performance comparable to state-of-the-art models on UltraMedical. We publicly release our datasets, reward models, and benchmarks to facilitate reproducible research in safety-aware medical alignment.

View on arXivView PDF

Code Implementations(1)

Official Repository of "ProMedical: Hierarchical Fine-Grained Criteria Modeling for Medical LLM Alignment via Explicit Injection". ACL 2026

10Apr 14, 20266 months ago

Similar Papers

Cite this paper

@article{geng2026promedical,
  title  = {ProMedical: Hierarchical Fine-Grained Criteria Modeling for Medical LLM Alignment via Explicit Injection},
  author = {He Geng and Yangmin Huang and Lixian Lai and Qianyun Du and Hui Chu and Zhiyang He and Jiaxue Hu and Xiaodong Tao},
  year   = {2026},
  eprint = {2604.08326},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2604.08326v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2604.08326.svg)](https://codeofpaper.com/paper/2604.08326)

Discussion