A machine learning-based pulmonary venous obstruction prediction model using clinical data and CT image

Int J Comput Assist Radiol Surg. 2021 Apr;16(4):609-617. doi: 10.1007/s11548-021-02335-y. Epub 2021 Mar 31.

Abstract

Purpose: In this study, we try to consider the most common type of total anomalous pulmonary venous connection and established a machine learning-based prediction model for postoperative pulmonary venous obstruction by using clinical data and CT images jointly.

Method: Patients diagnosed with supracardiac TPAVC from January 1, 2009, to December 31, 2018, in Guangdong Province People's Hospital were enrolled. Logistic regression were applied for clinical data features selection, while a convolutional neural network was used to extract CT images features. The prediction model was established by integrating the above two kinds of features for PVO prediction. And the proposed methods were evaluated using fourfold cross-validation.

Result: Finally, 131 patients were enrolled in our study. Results show that compared with traditional approaches, the machine learning-based joint method using clinical data and CT image achieved the highest average AUC score of 0.943. In addition, the joint method also achieved a higher sensitivity of 0.828 and a higher positive prediction value of 0.864.

Conclusion: Using clinical data and CT images jointly can improve the performance significantly compared with other methods that using only clinical data or CT images. The proposed machine learning-based joint method demonstrates the practicability of fully using multi-modality clinical data.

Keywords: Deep learning; Prediction; Pulmonary venous obstruction; Total anomalous pulmonary venous connection.

MeSH terms

  • Area Under Curve
  • Humans
  • Logistic Models
  • Lung / diagnostic imaging*
  • Machine Learning*
  • Neural Networks, Computer*
  • Predictive Value of Tests
  • Pulmonary Veins / diagnostic imaging*
  • ROC Curve
  • Tomography, X-Ray Computed / methods
  • Vascular Diseases / diagnostic imaging*