arXiv:2511.16839cs.LGcs.AI2025-11

用短序列Transformer预测心衰患者出院后一年内再住院或死亡风险

Trajectory-guided discharge stratification for heart failure using short-context electronic health record sequence modeling

  • 基于患者住院期间的多模态医疗轨迹,用紧凑自回归Transformer建模
  • 在4.2万瑞典心衰患者数据上,再住院和死亡预测AUPRC分别达0.555和0.574
  • 模型在数据少或概念缺失时仍稳定,适合临床实际应用

目的:心衰出院计划依赖于识别高恶化或死亡风险患者,但从常规电子健康记录(EHR)中准确预测仍具挑战。方法:我们提出轨迹引导的出院分层方法(TGDS-HF),采用轻量级短上下文自回归Transformer端到端读取患者住院期间的诊断、生命体征、检验、用药和操作等多源轨迹,并用于分层评估出院后一年内临床不稳定(再住院表型)或死亡风险。在瑞典心衰队列(N=42,820)上,以初始心衰诊断入院时为起点进行预测。TGDS-HF包含三部分:类别级标记化、时间加权表示与序列模型配置。通过消融实验验证其有效性。结果:相比传统XGBoost及基于BERT的基线模型,使用Llama主干的TGDS-HF在默认设置下,对两项任务的精确率-召回率曲线下面积(AUPRC)分别为0.555(95%置信区间:0.535–0.575)和0.574(0.550–0.599),且校准性能稳健。通过每日聚合重复连续事件的任务特异性优化,死亡预测提升至0.582(0.558–0.608)。此外,该模型在临床概念减少或训练数据有限时仍保持强性能。结论:结合不稳定性与死亡风险的联合预测可支持个性化出院规划,涵盖初级保健随访、专科管理乃至必要时的姑息治疗。

原文摘要 · Abstract (English)

Purpose: Heart failure (HF) discharge planning depends on identifying patients at risk of deterioration or death, yet accurate prediction from routinely collected electronic health records (EHRs) remains challenging. Methods: We develop trajectory-guided discharge stratification for heart failure (TGDS-HF), a methodology that reads the patient in-hospital trajectory of diagnoses, vital signs, laboratories, medications, and procedures end-to-end with a compact short-context autoregressive Transformer, and uses it to stratify one-year risks of clinical instability (a rehospitalization phenotype) or mortality for discharge care. We instantiate TGDS-HF on a Swedish HF cohort (N = 42,820) to predict one-year clinical instability or mortality at the initial HF diagnosis in-hospital. TGDS-HF has three components: category-level tokenization, recency-weighted temporal representation, and sequence model configuration. We run ablations on these components to show the effectiveness of TGDS-HF. Results: Against traditional eXtreme gradient boosting machine (XGBoost) and bidirectional encoder representations from Transformers (BERT)-based EHR sequence-modeling baselines, TGDS-HF (Llama backbone) achieved area under the precision-recall curves (AUPRCs) with 95% confidence intervals of 0.555 (0.535-0.575) and 0.574 (0.550-0.599) across the two tasks at the method default, with robust calibration. A task-specific refinement using daily aggregation of repeated continuous events improves the mortality task to 0.582 (0.558-0.608). Further, TGDS-HF maintains strong performance under reduced clinical concept availability and limited training data. Conclusion: Combined predictions of instability and mortality from TGDS-HF may support personalized discharge planning, ranging from follow-up in primary care to specialist-led management and, when appropriate, palliative care.

心衰预测电子病历时间序列风险分层

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。