RESUMEN
PURPOSE: The current study explores the application of 3D U-Net architectures combined with Inception and ResNet modules for precise lung nodule detection through deep learning-based segmentation technique. This investigation is motivated by the objective of developing a Computer-Aided Diagnosis (CAD) system for effective diagnosis and prognostication of lung nodules in clinical settings. METHODS: The proposed method trained four different 3D U-Net models on the retrospective dataset obtained from AIIMS Delhi. To augment the training dataset, affine transformations and intensity transforms were utilized. Preprocessing steps included CT scan voxel resampling, intensity normalization, and lung parenchyma segmentation. Model optimization utilized a hybrid loss function that combined Dice Loss and Focal Loss. The model performance of all four 3D U-Nets was evaluated patient-wise using dice coefficient and Jaccard coefficient, then averaged to obtain the average volumetric dice coefficient (DSCavg) and average Jaccard coefficient (IoUavg) on a test dataset comprising 53 CT scans. Additionally, an ensemble approach (Model-V) was utilized featuring 3D U-Net (Model-I), ResNet (Model-II), and Inception (Model-III) 3D U-Net architectures, combined with two distinct patch sizes for further investigation. RESULTS: The ensemble of models obtained the highest DSCavg of 0.84 ± 0.05 and IoUavg of 0.74 ± 0.06 on the test dataset, compared against individual models. It mitigated false positives, overestimations, and underestimations observed in individual U-Net models. Moreover, the ensemble of models reduced average false positives per scan in the test dataset (1.57 nodules/scan) compared to individual models (2.69-3.39 nodules/scan). CONCLUSIONS: The suggested ensemble approach presents a strong and effective strategy for automatically detecting and delineating lung nodules, potentially aiding CAD systems in clinical settings. This approach could assist radiologists in laborious and meticulous lung nodule detection tasks in CT scans, improving lung cancer diagnosis and treatment planning.
RESUMEN
PURPOSE: The proposed work aims to develop an algorithm to precisely segment the lung parenchyma in thoracic CT scans. To achieve this goal, the proposed technique utilized a combination of deep learning and traditional image processing algorithms. The initial step utilized a trained convolutional neural network (CNN) to generate preliminary lung masks, followed by the proposed post-processing algorithm for lung boundary correction. METHODS: First, the proposed method trained an improved 2D U-Net CNN model with Inception-ResNet-v2 as its backbone. The model was trained on 32 CT scans from two different sources: one from the VESSEL12 grand challenge and the other from AIIMS Delhi. Further, the model's performance was evaluated on a test dataset of 16 CT scans with juxta-pleural nodules obtained from AIIMS Delhi and the LUNA16 challenge. The model's performance was assessed using evaluation metrics such as average volumetric dice coefficient (DSCavg), average IoU score (IoUavg), and average F1 score (F1avg). Finally, the proposed post-processing algorithm was implemented to eliminate false positives from the model's prediction and to include juxta-pleural nodules in the final lung masks. RESULTS: The trained model reported a DSCavg of 0.9791 ± 0.008, IoUavg of 0.9624 ± 0.007, and F1avg of 0.9792 ± 0.004 on the test dataset. Applying the post-processing algorithm to the predicted lung masks obtained a DSCavg of 0.9713 ± 0.007, IoUavg of 0.9486 ± 0.007, and F1avg of 0.9701 ± 0.008. The post-processing algorithm successfully included juxta-pleural nodules in the final lung mask. CONCLUSIONS: Using a CNN model, the proposed method for lung parenchyma segmentation produced precise segmentation results. Furthermore, the post-processing algorithm addressed false positives and negatives in the model's predictions. Overall, the proposed approach demonstrated promising results for lung parenchyma segmentation. The method has the potential to be valuable in the advancement of computer-aided diagnosis (CAD) systems for automatic nodule detection.