Lin Mingquan, Wang Song, Ding Ying, Zhao Lihui, Wang Fei, Peng Yifan
Department of Population Health Sciences, Weill Cornell Medicine, New York, USA.
Cockrell School of Engineering, The University of Texas at Austin, Austin, USA.
Proc (IEEE Int Conf Healthc Inform). 2021 Aug;2021:497-498. doi: 10.1109/ichi52183.2021.00088. Epub 2021 Oct 15.
The predictive Intensive Care Unit (ICU) scoring system plays an important role in ICU management for its capability of predicting important outcomes, especially mortality. There are many scoring systems that have been developed and used in the ICU. These scoring systems are primarily based on the structured clinical data contained in the electronic health record (EHR), which may suffer the loss of the important clinical information contained in the narratives and images. In this work, we build a deep learning based survival prediction model with multi-modality data to predict ICU-mortality. Four sets of features are investigated: (1) physiological measurements of Simplified Acute Physiology Score (SAPS) II, (2) common thorax diseases pre-defined by radiologists, (3) BERT-based text representations, and (4) chest X-ray image features. We use the Medical Information Mart for Intensive Care IV (MIMIC-IV) dataset to evaluate the proposed model. Our model achieves the average C-index of 0.7847 (95% confidence interval, 0.7625-0.8068), which substantially exceeds that of the baseline with SAPS-II features (0.7477 (0.7238-0.7716)). Ablation studies further demonstrate the contributions of pre-defined labels (2.12%), text features (2.68%), and image features (2.96%). Our model achieves a higher average C-index than the traditional machine learning methods under the same feature fusion setting, which suggests that the deep learning methods can outperform the traditional machine learning methods in ICU-mortality prediction. These results highlight the potential of deep learning models with multimodal information to enhance ICU-mortality prediction. We make our work publicly available at https://github.com/bionlplab/mimic-icu-mortality.
预测性重症监护病房(ICU)评分系统因其预测重要结局尤其是死亡率的能力,在ICU管理中发挥着重要作用。ICU中已开发并使用了许多评分系统。这些评分系统主要基于电子健康记录(EHR)中包含的结构化临床数据,而这些数据可能会丢失叙述和图像中包含的重要临床信息。在这项工作中,我们构建了一个基于深度学习的多模态数据生存预测模型,以预测ICU死亡率。我们研究了四组特征:(1)简化急性生理学评分(SAPS)II的生理测量值,(2)放射科医生预先定义的常见胸部疾病,(3)基于BERT的文本表示,以及(4)胸部X光图像特征。我们使用重症监护医学信息集市IV(MIMIC-IV)数据集来评估所提出的模型。我们的模型平均C指数达到0.7847(95%置信区间,0.7625 - 0.8068),大大超过了具有SAPS-II特征的基线模型(0.7477(0.7238 - 0.7716))。消融研究进一步证明了预定义标签(2.12%)、文本特征(2.68%)和图像特征(2.96%)的贡献。在相同的特征融合设置下,我们的模型比传统机器学习方法获得了更高的平均C指数,这表明深度学习方法在ICU死亡率预测方面可以优于传统机器学习方法。这些结果凸显了具有多模态信息的深度学习模型在增强ICU死亡率预测方面的潜力。我们将我们的工作公开在https://github.com/bionlplab/mimic-icu-mortality上。