arXiv:2604.16878cs.LG2026-04被引 2

用医学本体增强对比学习,提升重症监护风险预测精度

OC-Distill: Ontology-aware Contrastive Learning with Cross-Modal Distillation for ICU Risk Prediction

  • 基于ICD疾病层级构建患者相似性,优化对比学习
  • 通过多模态知识蒸馏,融合病历文本信息提升模型表现
  • 仅需生命体征即可推理,适合临床部署

早期预测重症患者临床恶化及剩余住院时间,有助于及时干预与资源优化。现有方法存在两大局限:对比预训练将所有患者视为同等负样本,忽略具有相关诊断患者的临床相似性;下游微调常忽略病历文本等互补模态,而这些信息无法仅从生理信号中获得。为此,我们提出OC-Distill框架,分两阶段训练,推理时仅需生命体征。第一阶段引入本体感知对比目标,利用ICD疾病层级量化患者相似性,学习具有临床意义的表征。第二阶段通过跨模态知识蒸馏,将病历文本中的补充信息迁移至模型。在MIMIC数据集上,该方法在多个重症预测任务中实现更优标签效率,并达到仅使用生命体征推理方法中的最佳性能。

原文摘要 · Abstract (English)

Early prediction of severe clinical deterioration and remaining length of stay can enable timely intervention and better resource allocation in high-acuity settings such as the ICU. This has driven the development of machine learning models that leverage continuous streams of vital signs and other physiological signals for real-time risk prediction. Despite their promise, existing methods have important limitations. Contrastive pretraining treats all patients as equally strong negatives, failing to capture clinically meaningful similarity between patients with related diagnoses. Meanwhile, downstream fine-tuning typically ignores complementary modalities such as clinical notes, which provide rich contextual information unavailable in physiological signals alone. To address these challenges, we propose OC-Distill, a two-stage framework that leverages multimodal supervision during training while requiring only vital signs at inference. In the first stage, we introduce an ontology-aware contrastive objective that exploits the ICD hierarchy to quantify patient similarity and learn clinically grounded representations. In the second stage, we fine-tune the pretrained encoder via cross-modal knowledge distillation, transferring complementary information from clinical notes into the model. Across multiple ICU prediction tasks on MIMIC, OC-Distill demonstrates improved label efficiency and achieves state-of-the-art performance among methods that use only vital signs at inference.

ICU预测多模态学习知识蒸馏临床决策

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