Teaching LLMs How ICU Physicians Approach Clinical Reasoning Through OMOP-Aligned Retrieval Improves Reasoning Across Clinical Domains
Organizations: Department of Biomedical Engineering, University of Florida, Gainesville, FL, USA. · Intelligent Clinical Care Center (IC3), University of Florida, Gainesville, FL, USA. · Department of Electrical and Computer Engineering, University of Florida, Gainesville, FL, USA. · Google, Seattle, WA, USA. ·
Abstract
Clinical decision-making relies on identifying relevant patient information to guide diagnosis and treatment, a challenge that is especially difficult in the data-dense and rapidly changing intensive care unit (ICU). Large language models (LLMs) could support this task. However, existing applications and datasets mostly emphasize surface-level retrieval or factual recall rather than the inductive and deductive reasoning clinicians practice to select and reason over decision-relevant evidence. We hypothesized that training LLMs on expert ICU reasoning could yield clinical reasoning skills that generalize beyond critical care. Here we introduce ICU-REACT, a reasoning dataset developed with 19 clinicians through a clinician-in-the-loop framework to teach LLMs to perform information retrieval and context-aware clinical reasoning in the ICU. Using ICU-REACT, we fine-tuned Clin-REACT models spanning 8B-70B parameters and three model families. Across five clinical reasoning benchmarks, Clin-REACT consistently outperformed its backbone models and open-source general-purpose and medical LLMs. Gains extended to different tasks including script concordance tests, and downstream diagnosis and treatment tasks. These findings suggest that expert reasoning supervision in critical care can improve broader clinical reasoning, although prospective evaluation is needed before real-world clinical use.
Figures & tables
Appendix figures & tables22 assets
Supplementary material from the paper’s appendix.
Appendix
| Hyperparameter | Clin-REACT 8B | Clin-REACT 14B | Clin-REACT 31B | Clin-REACT 70B |
| LoRA rank ( ) | 16 | 16 | 16 | 16 |
| LoRA alpha ( ) | 32 | 32 | 32 | 32 |
| LoRA dropout | 0.05 | 0.05 | 0.05 | 0.05 |
| Learning rate | ||||
| Weight decay | 0.01 | 0.01 | 0.01 | 0.01 |
| Warmup steps | 100 | 100 | 100 | 100 |
| Model | Average | MMLU-Med | MedMCQA | MedQA | MedXpertQA |
|---|---|---|---|---|---|
| Llama 3.1 8B Instruct | 43.3 | 64.0 | 47.5 | 47.1 | 14.8 |
| Clin-REACT 8B | 34.6 ( 8.7 ) | 55.2 ( 8.8 ) | 42.0 ( 5.5 ) | 27.3 ( 19.8 ) | 13.8 ( 1.0 ) |
| Baichuan M1 14B Instruct | 56.5 | 81.4 | 62.5 | 61.2 | 21.0 |
| Clin-REACT 14B | 55.3 ( 1.2 ) | 78.8 ( 2.6 ) | 61.3 ( 1.2 ) | 59.5 ( 1.7 ) | 21.7 ( 0.7 ) |
| Gemma-4-31B-It | 80.2 | 94.2 | 78.5 | 92.4 | 55.8 |
| Clin-REACT 31B | 78.8 ( 1.4 ) | 94.9 ( 0.7 ) | 78.4 ( 0.1 ) | 89.8 ( 2.6 ) | 52.2 ( 3.6 ) |
| Domain | Subcategory | Variables |
|---|---|---|
| Diagnosis | Cardiac | acute coronary syndrome; atrial fibrillation; bradycardia; cardiac arrest; cardiogenic shock; complete heart block; endocarditis; heart failure; myocardial infarction; pericardial effusion; supraventricular tachycardia; ventricular fibrillation; ventricular tachycardia; Aortic dissection; Cardiomyopathy; Myocarditis; Valvular heart disease; DVT; Pulmonary Embolism |
| Diagnosis | Pulmonary | respiratory acidosis; respiratory alkalosis; ventilator-associated pneumonia; ARDS; COPD exacerbation; aspiration pneumonitis; asthma exacerbation; hemothorax; pleural effusion; pneumonia; pneumothorax; pulmonary contusion; pulmonary edema; pulmonary embolism; pulmonary hypertension; respiratory failure; air embolism; fat embolism syndrome; Atelectasis; Pulmonary fibrosis; Bronchiectasis; Lung cancer |
| Diagnosis | Neurological | hemorrhage; encephalitis; myxedema coma; Guillain-Barré syndrome; brain death; cerebral edema; encephalopathy; epidural hematoma; intracranial hemorrhage; ischemic stroke; myasthenia gravis; status epilepticus; subarachnoid hemorrhage; subdural hematoma; traumatic brain injury; Spinal cord injury; Brain tumor; Hydrocephalus; ICU-acquired weakness; Delirium (Hypoactive) |
| Diagnosis | Infectious | bacteremia; catheter-related bloodstream infection; fungemia; necrotizing fasciitis; anaphylaxis; sepsis; septic shock; COVID-19; Influenza; Cellulitis; Abscess; Ventilator-associated pneumonia; CLABSI; CAUTI |
| Diagnosis | Renal | urinary tract infection; adrenal insufficiency; acute kidney injury; acute tubular necrosis; chronic kidney disease; nephrotic syndrome; rhabdomyolysis; urinary retention; Nephrolithiasis; Glomerulonephritis; Hydronephrosis; Pyelonephritis; Hepatorenal Syndrome |
| Topic | Dimension | Items |
|---|---|---|
| Hemodynamic Instability / Shock | Clinical scenarios | Septic shock from pneumonia; Septic shock from intra-abdominal source; Postoperative hypotension after major surgery; Acute upper GI hemorrhage with volume loss; Traumatic hemorrhagic shock; Cardiogenic shock after acute MI; Acute decompensated heart failure with low output; Acute right-ventricular failure from massive PE; Cardiac tamponade with obstructive physiology; Anaphylactic distributive shock; Neurogenic/spinal shock after trauma; Adrenal/relative cortisol insufficiency in shock; Tension pneumothorax causing obstructive shock; Dialysis/CRRT-associated hypotension in ICU |
| Hemodynamic Instability / Shock | Presentations | Sustained low MAP despite initial fluids; Rising lactate and signs of poor perfusion; Cool, mottled extremities and delayed cap refill; Warm, vasodilatory shock with wide pulse pressure; Oliguria or declining urine output; Acute change in mental status from hypoperfusion; Tachycardia with narrow pulse pressure; Escalating vasopressor requirement over hours; Hypotension with new arrhythmia; Need for closer hemodynamic monitoring; Hypotension in setting of limited fluid tolerance (ARDS/CHF) |
| Hemodynamic Instability / Shock | Management decisions | Initiate norepinephrine as first-line vasopressor; Add a second vasopressor such as vasopressin; Start inotropic support for low cardiac output; Administer another fluid bolus vs hold further fluids; Transfuse packed RBCs for hemorrhagic component; Obtain bedside echocardiography to define etiology; Place arterial and/or central line for monitoring; Initiate stress-dose corticosteroids for refractory shock; Arrange source-control procedure for shock driver; Escalate to mechanical/advanced circulatory support; Begin conservative fluid strategy after stabilization |
| Hemodynamic Instability / Shock | Contextual modifiers | Patient has significant cardiopulmonary comorbidities; High risk of fluid overload/ARDS so resuscitate cautiously; Limited vascular access or line-related infection concern; Post-cardiac surgery physiology alters pressor choice; Immunocompromised host with atypical infection; Goals-of-care or DNR/DNI status limits escalation |
| Respiratory Failure | Clinical scenarios | ARDS secondary to severe pneumonia; Community-acquired pneumonia with hypoxemic failure; COPD exacerbation with hypercapnic respiratory failure; Obesity hypoventilation/OSA causing ventilatory failure; Postoperative atelectasis with impaired gas exchange; Aspiration event in intubated patient; Ventilator-associated pneumonia in ICU; Neuromuscular weakness (GBS/MG) causing hypoventilation; Cardiogenic pulmonary edema with respiratory compromise; Prolonged mechanical ventilation needing weaning; Post-extubation stridor or upper-airway compromise; Trauma/chest wall injury with poor ventilation; Severe asthma in ICU not responding to usual therapy |
Medical LLMs
Medical large language models are AI systems adapted to clinical and biomedical data for health reasoning, information extraction, and communication.
See all 402 papers