该研究提出一种面向电子健康记录(EHR)的预测性世界模型,将患者病史表示为以临床动作为条件的潜在轨迹,显式分离患者状态与观测到的临床动作,并通过状态与动作共同约束的转移算子学习其演化。其核心是状态依赖的线性化预测器,可传播潜在状态的局部变化,支持时间分辨预测、多步推演和动作条件模拟。实验显示,生成的潜在轨迹保留了编码器表示中的下游预测信号、时间结构及亚组统计关系;将自由推演解码回原始临床特征空间后,与任务匹配的预测基线相比,所学动态仍保留超出潜在空间一致性的预测信息。在二分类与时间到事件任务上,该模型以约小30倍的参数量取得与大型EHR基础模型相当但依赖任务的表现;扰动动作序列会引起生成轨迹的相应变化,表明动态对动作敏感。结果支持将世界建模作为从纵向EHR学习显式、时间索引的患者演化表示的补充框架。
| 世界模型 (World Model) | 一种学习环境动态并预测未来状态的模型,常用于模拟和规划。 |
| 动作条件化 (Action-Conditioned) | 模型根据输入的动作(如治疗干预)来调节其预测或生成过程。 |
| 潜在轨迹 (Latent Trajectory) | 在低维潜在空间中表示的患者状态随时间变化的序列。 |
| 状态依赖线性化预测器 (State-Dependent Linearization Predictor) | 一种根据当前状态动态线性化转移算子以预测状态局部变化的模块。 |
| 多步展开 (Multi-step Rollout) | 在时间序列上迭代应用模型进行多步预测的过程。 |
📱 每天一份 AI 前沿日报
关注公众号,每天 09:00 推送 · 不错过任何重磅