论文

用神经网络学习患者世界模型,模拟ICU医疗策略效果

Learning a patient world model for health system simulation in intensive care

精选理由

有人把ICU患者的电子病历训练成世界模型,一次建模就能模拟床位、调度、转院多种政策,比传统Markov模型多捕捉恢复速度差异。

研究团队提出从MIMIC-IV纵向电子病历中训练患者世界模型,为ICU运营模拟提供共享的患者轨迹表征。模型冻结后通过场景映射接入ICU日历、资源和队列,一次建模即可评估床位容量、护理调度和转院容量等多个模拟问题。在总体占用率和对容量与调度政策的方向性响应上,模型与Markov和semi-Markov基线持平。对不同历史轨迹进入同一支持状态的患者,模型捕捉到24小时恢复差异(预测-6.4对比实际-6.7个百分点),这是紧凑状态模型无法识别的。

原文 · arXiv cs.AI

Learning a patient world model for health system simulation in intensive care

Designing healthcare strategies requires anticipating their clinical and operational consequences before implementation, a task that often relies on health system simulation. Such a simulation necessitates modeling how patient states evolve and generate care demands. However, conventional approaches typically tailor compact state transitions to a single policy question and omit history-dependent dynamics, thus requiring the redefinition of states and transitions when there is a new question that entails different clinical information. We propose to learn a patient world model from longitudinal EHRs using a neural network to provide a shared clinical representation of patient trajectories that can serve simulation questions across scenarios. Taking ICU operations as an example, we train the model on MIMIC-IV, freeze it, and couple its time-queryable predictions to shared ICU calendars, resources, and queues via scenario-specific mappings, so that one model renders admission demand, care tasks, and discharge proposals across simulations evaluating ICU bed capacity, care dispatch, and post-ICU transfer capacity. Experiments show that our model matches Markov and semi-Markov comparators in aggregate occupancy and directional responses to capacity and dispatch policies. It also reproduces policy gains from replayed observed trajectories. For patients entering the same support state from different prior trajectories, the learned representation captures a 24-h recovery difference (-6.4 predicted versus -6.7 observed percentage points), a contrast missed by compact state models, which propagates into distinct simulated care demand, queue burden, and bed use. These results indicate that a learned world model of patient dynamics has the potential to replace question-specific state models, allowing new care policies to be evaluated by reconfiguring the scenario without model rebuilding.