Development and validation of a lung graph-based machine learning model to predict acute pulmonary thromboembolism on chest noncontrast computed tomography.

BACKGROUND

Computed tomography pulmonary angiography (CTPA) is a first-line noninvasive method to diagnose acute pulmonary thromboembolism (APE); however, whether chest noncontrast CT (NC-CT) could aid in the diagnosis of APE remains unknown. The aim of this study was to build and evaluate a holistic lung graph-based machine learning (HLG-ML) using NC-CT for the diagnosis of APE and to compare its performance with that of radiologists and the YEARS algorithm.

METHODS

This study enrolled 178 cases (77 males; age 63.9±16.7 years) who underwent NC-CT and CTPA in the same day from January 2019 to December 2020. Of these patients, 133 (75% of cases; 58 males; age 65.4±15.6 years) were placed into a training group and 45 (25% of cases; 19 males; age 59.6±19.2 years) into a testing group. The other 43 cases (18 males; age 62.8±20.0 years) were used to externally validate the model between January 2021 and March 2022. A HLG was developed with a pulmonary radiomics descriptor derived from NC-CT images. The approach extracted local radiomics features and encoded these local features into a radiomics descriptor as a characterization of global radiomics feature distribution. Subsequently, 8 ML models were trained and compared based on the radiomics descriptor. In the validation group, area under the curves (AUCs) of the HLG-ML model in the diagnosis of APE were compared with those of the 3 radiologists and the YEARS algorithm.

RESULTS

Among the 8 ML models, gradient boosting decision tree demonstrated the best classification performance (AUC =0.772) on the training set. In the testing set, the AUC of gradient boosting decision trees was 0.857 [95% confidence intervals (CIs): 0.699-0.951]. In the validation set, the performance of gradient boosting decision tree (AUC =0.810; 95% CI: 0.669-0.952; Youden index =0.621) outperformed 3 radiologists (AUC =0.508, 95% CI: 0.335-0.681, Youden index =0.016; AUC =0.504, 95% CI: 0.354-0.654, Youden index =0.008; AUC =0.527, 95% CI: 0.363-0.691, Youden index =0.050) and the YEARS algorithm (AUC =0.618; 95% CI: 0.469-0.767; Youden index =0.237).

CONCLUSIONS

Compared to all 3 radiologists and the YEARS algorithm, the proposed HLG-based gradient boosting decision tree model achieved a superior performance in the diagnosis of APE on the NC-CT and may thus serve as a valuable tool for physicians in the diagnosis of APE.