Back

Decoupling Reasoning and Reward: A Modular Approach for Stable Alignment of Small Clinical Language Models

Bhattacharyya, K.; Kamabattula, S.

2026-03-13 health informatics
10.64898/2026.03.12.26348283 medRxiv
Show abstract

Deploying language models (LMs) in clinical settings requires navigating competing demands between accuracy, auditability, and on-device efficiency for privacy. While smaller LMs are desirable for this purpose, aligning them with methods like Group Relative Policy Optimization (GRPO) is often hindered by training instability and objective conflicts. Prior work has shown that Chain-of-Thought (CoT) supervision during supervised fine-tuning (SFT) can stabilize GRPO, but existing approaches typically entangle these objectives within a single, monolithic model. In this work, we introduce a modular, adapter-based alignment framework that decouples reasoning supervision and reward tuning into separate, composable parameter-efficient fine-tuning (PEFT) stages using LoRA adapters. We evaluate five alignment configurations on a medically grounded question answering dataset, using Qwen2.5 models from 0.5B to 7B to analyze how alignment stability, factual accuracy, and structural auditability scale with model size. Our findings demonstrate that this modular approach resolves key training instabilities, especially in smaller models, and produces structurally consistent, auditable reasoning without sacrificing accuracy. To support further research, we release, upon publication, (1) our dataset comprising over 100K clinically relevant QA pairs with CoT traces and (2) our multi-stage alignment codebase. We conclude that decoupling reasoning and reward offers a flexible and robust foundation for building privacy-preserving, verifiably aligned clinical LLMs that successfully address the competing demands in clinical AI.

Matching journals

The top 8 journals account for 50% of the predicted probability mass.

50% of probability mass above

"Similar papers" are the closest papers from that journal in the model's embedding space. They show what the match is built on, but the ranking comes mostly from a classifier over the whole training set, not from these examples alone.