Decoupling Reasoning and Reward: A Modular Approach for Stable Alignment of Small Clinical Language Models
Bhattacharyya, K.; Kamabattula, S.
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.
Similar papers in this journal
Similar papers in this journal
- LCD Benchmark: Long Clinical Document Benchmark on Mortality Prediction for Language Models 94%
- A Novel Question-Answering Framework for Automated Abstract Screening Using Large Language Models 93%
- Analysis of Eligibility Criteria Clusters Based on Large Language Models for Clinical Trial Design 92%
Similar papers in this journal
- Building a Best-in-Class De-identification Tool for Electronic Medical Records Through Ensemble Learning 93%
- Inferring global-scale temporal latent topics from news reports to predict public health interventions for COVID-19 92%
- Generating hard-to-obtain information from easy-to-obtain information: applications in drug discovery and clinical inference 92%
Similar papers in this journal
"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.