RESUMO
Accurate segmentation of lung nodules from pulmonary computed tomography (CT) slices plays a vital role in the analysis and diagnosis of lung cancer. Convolutional Neural Networks (CNNs) have achieved state-of-the-art performance in the automatic segmentation of lung nodules. However, they are still challenged by the large diversity of segmentation targets, and the small inter-class variances between the nodule and its surrounding tissues. To tackle this issue, we propose a features complementary network according to the process of clinical diagnosis, which made full use of the complementarity and facilitation among lung nodule location information, global coarse area, and edge information. Specifically, we first consider the importance of global features of nodules in segmentation and propose a cross-scale weighted high-level feature decoder module. Then, we develop a low-level feature decoder module for edge feature refinement. Finally, we construct a complementary module to make information complement and promote each other. Furthermore, we weight pixels located at the nodule edge on the loss function and add an edge supervision to the deep supervision, both of which emphasize the importance of edges in segmentation. The experimental results demonstrate that our model achieves robust pulmonary nodule segmentation and more accurate edge segmentation.
RESUMO
BACKGROUND: Accurate, noninvasive, and reliable assessment of epidermal growth factor receptor (EGFR) mutation status and EGFR molecular subtypes is essential for treatment plan selection and individualized therapy in lung adenocarcinoma (LUAD). Radiomics models based on 18F-FDG PET/CT have great potential in identifying EGFR mutation status and EGFR subtypes in patients with LUAD. The validation of multi-center data, model visualization, and interpretation are significantly important for the management, application and trust of machine learning predictive models. However, few EGFR-related research involved model visualization and interpretation, and multi-center trial. PURPOSE: To develop explainable optimal predictive models based on handcrafted radiomics features (HRFs) extracted from multi-center 18F-FDG PET/CT to predict EGFR mutation status and molecular subtypes in LUAD. METHODS: Baseline 18F-FDG PET/CT images of 383 LUAD patients from three hospitals and one public data set were collected. Further, 1808 HRFs were extracted from the primary tumor regions using Pyradiomics. Predictive models were built based on cross-combination of seven feature selection methods and seven machine learning algorithms. Yellowbrick and explainable artificial intelligence technology were used for model visualization and interpretation. Receiver operating characteristic curve, classification report and confusion matrix were used for model performance evaluation. Clinical applicability of the optimal models was assessed by decision curve analysis. RESULTS: STACK feature selection method combined with light gradient boosting machine (LGBM) reached optimal performance in identifying EGFR mutation status ([area under the curve] AUC = 0.81 in the internal test cohort; AUC = 0.62 in the external test cohort). Random forest feature selection method combined with LGBM reached optimal performance in predicting EGFR mutation molecular subtypes (AUC = 0.89 in the internal test cohort; AUC = 0.61 in the external test cohort). CONCLUSIONS: Explainable machine learning models combined with radiomics features extracted from multi-center/scanner 18F-FDG PET/CT have certain potential to identify EGFR mutation status and subtypes in LUAD, which might be helpful to the treatment of LUAD.