Pulmonary Nodule Segmentation Framework Based on Fine-Tuned and Pretrained Deep Neural Network Using CT Images
Ananya Bhattacharjee, R. Murugan, Tripti Goel, Seyedali Mirjalili
Abstract
Deep learning is one of the most rapidly growing and emerging technologies in pulmonary nodule segmentation. However, the different shape, size, and location of the nodules make it extremely difficult to segment correctly. The purpose of this study is to obtain a fast and accurate segmentation algorithm with less number of stages. A fine-tuned dual skip connection-based segmentation framework is proposed that integrates pretrained residual neural network (ResNet) 152 with the U-Net architecture, namely, ResiU-Net. Nine different pretrained and fine-tuned encoder backbones, such as ResNet18, ResNet 34, ResNet 50, ResNet 101, ResNet 152, SEResNet18, SE-ResNet34, ResNext 101, and ResNext 50 are compared and the proposed ResiU-Net approach gives the best results. Also, the fine-tuned ResiU-Net performs better than nontuned ResiU-Net. 1224 computed tomography patient images with different nodule shapes and sizes are selected. The proposed method achieves 97.44% F score, 95.02% intersection over union score, 94.87% dice score, 0.34% binary cross-entropy loss, and 0.7585 combined dice coefficient and binary focal loss. The proposed ResiU-Net outperforms the state-of-the-art methods and reports the best evaluation metrics. The time taken by the model to train is 43 min. Hence, the proposed model is a fast and accurate segmentation approach.