用肾病患者病历数据训练时间嵌入,提升临床表征质量与预测能力。
Learning temporal embeddings from electronic health records of chronic kidney disease patients
- 采用时序感知LSTM架构学习病程动态表征
- 嵌入模型在分期聚类与死亡预测上准确率超端到端模型
- 适合需要可解释、通用表征的临床研究者
我们研究了在慢性肾病患者纵向电子病历上训练的时间嵌入模型,是否能在不牺牲预测性能的前提下学习到临床有意义的表征,并探讨了架构选择对嵌入质量的影响。模型引导医学需要能捕捉疾病演变且透明、任务无关的表征,而多数临床预测模型仅针对单一任务优化。表示学习有助于获得跨下游任务泛化的嵌入,循环架构则适合建模观察性临床数据中的时序结构。基于MIMIC-IV数据集,我们研究慢性肾病患者,对比三种循环架构:基础LSTM、注意力增强LSTM和时序感知LSTM(T-LSTM)。所有模型均以嵌入模型和端到端预测器两种方式训练。嵌入质量通过CKD分期聚类和院内死亡预测评估。T-LSTM生成更结构化嵌入,达更低戴维斯-鲍尔丁指数(DBI = 9.91)与更高分期分类准确率(0.74),优于基础LSTM(DBI = 15.85,准确率0.63)和注意力增强LSTM(DBI = 20.72,准确率0.67)。院内死亡预测中,嵌入模型持续优于端到端预测器,准确率从0.72–0.75提升至0.82–0.83,表明中间阶段学习嵌入比直接端到端学习更有效。
原文摘要 · Abstract (English)
We investigate whether temporal embedding models trained on longitudinal electronic health records can learn clinically meaningful representations without compromising predictive performance, and how architectural choices affect embedding quality. Model-guided medicine requires representations that capture disease dynamics while remaining transparent and task agnostic, whereas most clinical prediction models are optimised for a single task. Representation learning facilitates learning embeddings that generalise across downstream tasks, and recurrent architectures are well-suited for modelling temporal structure in observational clinical data. Using the MIMIC-IV dataset, we study patients with chronic kidney disease (CKD) and compare three recurrent architectures: a vanilla LSTM, an attention-augmented LSTM, and a time-aware LSTM (T-LSTM). All models are trained both as embedding models and as direct end-to-end predictors. Embedding quality is evaluated via CKD stage clustering and in-ICU mortality prediction. The T-LSTM produces more structured embeddings, achieving a lower Davies-Bouldin Index (DBI = 9.91) and higher CKD stage classification accuracy (0.74) than the vanilla LSTM (DBI = 15.85, accuracy = 0.63) and attention-augmented LSTM (DBI = 20.72, accuracy = 0.67). For in-ICU mortality prediction, embedding models consistently outperform end-to-end predictors, improving accuracy from 0.72-0.75 to 0.82-0.83, which indicates that learning embeddings as an intermediate step is more effective than direct end-to-end learning.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。