用连续轨迹建模病历数据,提升生存预测的准确性与可解释性。
TrajSurv: Learning Continuous Latent Trajectories from Electronic Health Records for Trustworthy Survival Prediction
- 基于神经微分方程学习不规则采样病历的连续潜变量轨迹。
- 在MIMIC-III和eICU上实现媲美顶尖模型的预测精度。
- 通过轨迹分解与聚类,揭示临床进展与生存结局的关联路径。
可信的生存预测对临床决策至关重要。纵向电子健康记录(EHR)为预测提供了独特机会,但准确建模患者连续临床进展并透明关联其与生存结果仍具挑战。为此,我们提出TrajSurv,一种从纵向EHR中学习连续潜变量轨迹以实现可信生存预测的模型。TrajSurv采用神经控制微分方程(NCDE)从不规则采样数据中提取连续时间潜状态,形成连续潜轨迹。为确保潜空间反映真实临床进展,模型通过时间感知对比学习将潜状态空间与患者状态空间对齐。为透明关联临床进展与生存结果,采用两步解耦解释策略:首先利用学习到的向量场解释临床特征变化如何驱动潜轨迹演化;其次对潜轨迹聚类,识别与不同生存结局相关的关键临床进展模式。在两个真实世界医疗数据集MIMIC-III和eICU上的评估显示,TrajSurv在预测准确率上具有竞争力,且在可解释性方面显著优于现有深度学习方法。
原文摘要 · Abstract (English)
Trustworthy survival prediction is essential for clinical decision making. Longitudinal electronic health records (EHRs) provide a uniquely powerful opportunity for the prediction. However, it is challenging to accurately model the continuous clinical progression of patients underlying the irregularly sampled clinical features and to transparently link the progression to survival outcomes. To address these challenges, we develop TrajSurv, a model that learns continuous latent trajectories from longitudinal EHR data for trustworthy survival prediction. TrajSurv employs a neural controlled differential equation (NCDE) to extract continuous-time latent states from the irregularly sampled data, forming continuous latent trajectories. To ensure the latent trajectories reflect the clinical progression, TrajSurv aligns the latent state space with patient state space through a time-aware contrastive learning approach. To transparently link clinical progression to the survival outcome, TrajSurv uses latent trajectories in a two-step divide-and-conquer interpretation process. First, it explains how the changes in clinical features translate into the latent trajectory's evolution using a learned vector field. Second, it clusters these latent trajectories to identify key clinical progression patterns associated with different survival outcomes. Evaluations on two real-world medical datasets, MIMIC-III and eICU, show TrajSurv's competitive accuracy and superior transparency over existing deep learning methods.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。