Using stacked deep learning models based on PET/CT images and clinical data to predict EGFR mutations in lung cancer

Abstract

Purpose: To determine whether stacked deep learning models based on PET/CT images and clinical data can help to predict epidermal growth factor receptor (EGFR) mutations in lung cancer. Methods: We analyzed data from two public datasets of patients who underwent 18F-FDG PET/CT. Three PET deep learning ResNet models and one CT deep learning ResNet model were trained as low-level predictors based on PET and CT images, respectively. A high-level Support Vector Machine model (Stack PET/CT and Clinical model) was trained using the prediction results of the low-level predictors and clinical data. The clinical data included sex, age, smoking history, SUVmax and SUVmean of the lesion. Fivefold cross-validation was used in this study to validate the prediction performance of the models. The predictive performance of the models was evaluated by receiver operator characteristic (ROC) curves. The area under the curve (AUC) was calculated. Results: One hundred forty-seven patients were included in this study. Among them, 37/147 cases were EGFR mutations, and 110/147 cases were EGFR wild-type. The ROC analysis showed that the Stack PET/CT & Clinical model had the best performance (AUC = 0.85 ± 0.09), with 0.76, 0.85 and 0.83 in sensitivity, specificity and accuracy, respectively. Three ResNet PET models had relatively higher AUCs (0.82 ± 0.07, 0.80 ± 0.08 and 0.79 ± 0.07) and outperformed the CT model (AUC = 0.58 ± 0.12). Conclusion: Using stack generalization, the deep learning model was able to efficiently combine the anatomic and biological imaging information gathered from PET/CT images with clinical data. This stacked deep learning model showed a strong ability to predict EGFR mutations with high accuracy.