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

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.

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.

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.

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.