用自监督预训练提升脓毒症患者48小时用药趋势预测精度
A Novel Multi-Task Teacher-Student Architecture with Self-Supervised Pretraining for 48-Hour Vasoactive-Inotropic Trend Analysis in Sepsis Mortality Prediction
- 教师-学生多任务框架+掩码自编码器预训练,提取稳定时序特征
- 在MIMIC-IV数据集上达到0.82的AUROC,优于基线0.74
- 揭示社会因素如婚姻状态、医保类型对死亡率影响显著
脓毒症是重症监护病房(ICU)死亡的主要原因,早期识别与有效干预对改善预后至关重要。然而,血管活性药物-正性肌力评分(VIS)随血流动力学状态动态变化,受用药不规律、数据缺失和混杂因素干扰,使预测难度增加。为此,我们提出一种新型教师-学生多任务框架,结合掩码自编码器(MAE)进行自监督VIS预训练。教师模型执行死亡率分类与病情严重度回归,学生模型则蒸馏出鲁棒的时间序列表示,增强对异构VIS数据的适应能力。相较于基于LSTM的方法,本方法在MIMIC-IV 3.0数据集(9,476名患者)上取得0.82的AUROC,优于基线(0.74)。SHAP分析显示,SOFA评分(0.147)对ICU死亡率影响最大,其次为LODS(0.033)、单身状态(0.031)和医疗补助保险(0.023),表明社会人口学因素亦具关键作用。SAPSII(0.020)也贡献显著。结果表明,临床与社会因素应共同纳入ICU决策支持系统。该多任务与知识蒸馏策略有助于更早识别高风险患者,提升预测准确性与疾病管理水平。
原文摘要 · Abstract (English)
Sepsis is a major cause of ICU mortality, where early recognition and effective interventions are essential for improving patient outcomes. However, the vasoactive-inotropic score (VIS) varies dynamically with a patient's hemodynamic status, complicated by irregular medication patterns, missing data, and confounders, making sepsis prediction challenging. To address this, we propose a novel Teacher-Student multitask framework with self-supervised VIS pretraining via a Masked Autoencoder (MAE). The teacher model performs mortality classification and severity-score regression, while the student distills robust time-series representations, enhancing adaptation to heterogeneous VIS data. Compared to LSTM-based methods, our approach achieves an AUROC of 0.82 on MIMIC-IV 3.0 (9,476 patients), outperforming the baseline (0.74). SHAP analysis revealed that SOFA score (0.147) had the greatest impact on ICU mortality, followed by LODS (0.033), single marital status (0.031), and Medicaid insurance (0.023), highlighting the role of sociodemographic factors. SAPSII (0.020) also contributed significantly. These findings suggest that both clinical and social factors should be considered in ICU decision-making. Our novel multitask and distillation strategies enable earlier identification of high-risk patients, improving prediction accuracy and disease management, offering new tools for ICU decision support.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。