Pulmonary Embolism Mortality Prediction Using Multimodal Learning Based on Computed Tomography Angiography and Clinical Data

Zhusi Zhong,Helen Zhang,Fayez H. Fayad,Andrew C. Lancaster,John Sollee,Shreyas Kulkarni,Cheng Ting Lin,Jie Li,Xinbo Gao,Scott Collins,Colin Greineder,Sun H. Ahn,Harrison X. Bai,Zhicheng Jiao,Michael K. Atalay
2024-06-05
Abstract:Purpose: Pulmonary embolism (PE) is a significant cause of mortality in the United States. The objective of this study is to implement deep learning (DL) models using Computed Tomography Pulmonary Angiography (CTPA), clinical data, and PE Severity Index (PESI) scores to predict PE mortality. Materials and Methods: 918 patients (median age 64 years, range 13-99 years, 52% female) with 3,978 CTPAs were identified via retrospective review across three institutions. To predict survival, an AI model was used to extract disease-related imaging features from CTPAs. Imaging features and/or clinical variables were then incorporated into DL models to predict survival outcomes. Four models were developed as follows: (1) using CTPA imaging features only; (2) using clinical variables only; (3) multimodal, integrating both CTPA and clinical variables; and (4) multimodal fused with calculated PESI score. Performance and contribution from each modality were evaluated using concordance index (c-index) and Net Reclassification Improvement, respectively. Performance was compared to PESI predictions using the Wilcoxon signed-rank test. Kaplan-Meier analysis was performed to stratify patients into high- and low-risk groups. Additional factor-risk analysis was conducted to account for right ventricular (RV) dysfunction. Results: For both data sets, the PESI-fused and multimodal models achieved higher c-indices than PESI alone. Following stratification of patients into high- and low-risk groups by multimodal and PESI-fused models, mortality outcomes differed significantly (both p<0.001). A strong correlation was found between high-risk grouping and RV dysfunction. Conclusions: Multiomic DL models incorporating CTPA features, clinical data, and PESI achieved higher c-indices than PESI alone for PE survival prediction.
Computer Vision and Pattern Recognition
What problem does this paper attempt to address?
This paper aims to predict the mortality rate of pulmonary embolism (PE) using deep learning (DL) models, combining computed tomography pulmonary angiography (CTPA) images and clinical data, as well as the Pulmonary Embolism Severity Index (PESI) score. The research team developed four models through retrospective analysis of data from 918 patients in three institutions: CTPA image-based model, clinical variable-based model, multimodal model (combining CTPA and clinical variables), and multimodal model incorporating PESI score. They found that the multimodal DL model integrating CTPA features, clinical data, and PESI outperformed the use of PESI alone in predicting PE survival rate. Furthermore, the study underscores the importance of right ventricular dysfunction as a risk factor for PE. The conclusion of the paper is that the multimodal DL model provides higher predictive accuracy than PESI and contributes to the improvement of prognostic assessment for PE patients.