Icu models improve patient vital sign predictions with drug effect rules
Sign-Constrained Intervention Effects for Domain-Generalizable ICU World Models
Computational Engineering, Finance, and Science
Summary
Predicting how a patient's vital signs change when given medicine in the ICU is tricky, especially when doctors try new doses that the model hasn't seen before. The authors show that usual models get confused because they learn wrong dose-outcome connections from past data. They created a new method called PHYSIO WORLD that uses medical knowledge about how drugs are supposed to work, to guide the model’s predictions. This makes the model better at predicting outcomes even with new dosing practices, while keeping accuracy on familiar cases.
What this means in practice
- •For intensive care data teams: Improve ICU patient monitoring systems by integrating pharmacological sign constraints to predict patient responses to new drug dosages more reliably.
- •For medical device engineers: Develop smarter ICU decision support tools that predict patient reactions to treatments with better reliability under varied dosing conditions.
Authors
Zhen Xu, Nicholas Konz, Zhen Tan, Zachary Plotkin, Tianlong Chen
Abstract
Predicting how a patient's vital signs respond to an intervention is a central question in intensive care. World models can do so by learning dynamics as a function of prior actions. However, such models tend to be brittle outside of training data. Clinicians choose drug dosages based on the patient's state, so the association a model learns between dose and outcome runs opposite to the drug's effect, generalizing poorly to out-of-distribution (OOD) dosing practices. Yet, pharmacological information provides the directional effects of different drugs, which are invariant to OOD shifts. To resolve the OOD sensitivity of current ICU world models, we introduce PHYSIO WORLD, which isolates an intervention's effect by evaluating a forward pass twice: once under recorded doses, and once with those doses set to zero. The difference is projected onto pharmacologically admissible effect directions stated by 70 rules over 28 drugs, leaving the magnitude to be learned from data. Across five distribution-shift parameters over cohorts from three intensive-care databases, and against forecasters, counterfactual models, and invariance objectives, PHYSIO WORLD attains the lowest OOD RMSE in every setting, improving on the strongest external baseline by 8-12%, while matching the in-distribution error of its backbone. Reversing the pharmacological directions degrades accuracy below the backbone, and data-mined directions recover almost no gain, indicating the improvement derives from the content of the pharmacological knowledge. The construction may extend to other settings where the sign of an effect is known in advance and its magnitude is confounded by treatment assignment.