Performance Evaluation of a Deep Learning System for Differential Diagnosis of Lung Cancer With Conventional CT and FDG PET/CT Using Transfer Learning and Metadata
Yong Jin Park, Dongmin Choi, Joon Young Choi, Seung Hyup Hyun
Abstract
PURPOSE: We aimed to evaluate the performance of a deep learning system for differential diagnosis of lung cancer with conventional CT and FDG PET/CT using transfer learning (TL) and metadata. METHODS: A total of 359 patients with a lung mass or nodule who underwent noncontrast chest CT and FDG PET/CT prior to treatment were enrolled retrospectively. All pulmonary lesions were classified by pathology (257 malignant, 102 benign). Deep learning classification models based on ResNet-18 were developed using the pretrained weights obtained from ImageNet data set. We propose a deep TL model for differential diagnosis of lung cancer using CT imaging data and metadata with SUVmax and lesion size derived from PET/CT. The area under the receiver operating characteristic curve (AUC) of the deep learning model was measured as a performance metric and verified by 5-fold cross-validation. RESULTS: The performance metrics of the conventional CT model were generally better than those of the CT of PET/CT model. Introducing metadata with SUVmax and lesion size derived from PET/CT into baseline CT models improved the diagnostic performance of the CT of PET/CT model (AUC = 0.837 vs 0.762) and the conventional CT model (AUC = 0.877 vs 0.817). CONCLUSIONS: Deep TL models with CT imaging data provide good diagnostic performance for lung cancer, and the conventional CT model showed overall better performance than the CT of PET/CT model. Metadata information derived from PET/CT can improve the performance of deep learning systems.