Coordinate-Unet 3D for segmentation of lung parenchyma

Abstract

Lung segmentation is an initial step to provide accurate lung parenchyma in many studies on lung diseases based on analyzing the Computed Tomography (CT) scan, especially in Non-Small Cell Lung Cancer (NSCLC) detection. In this work, Coordinate-UNet 3D, a model inspired by UNet, is proposed to improve the accuracy of lung segmentation in the CT scan. Like UNet, the proposed model consists of a contracting/encoder path to extract the high-level information and an expansive/decoder path to recover the features to provide the segmentation. However, we have considered modifying the structure inside each level of the model and using the Coordinate Convolutional layer as the final layer to provide the segmentation. This network was trained end-to-end from a small set of CT scans of NSCLC patients. The experimental results show the proposed network can provide a highly accurate segmentation for the validation set with a Dice Coefficient index of 0.991, an F1 score of 0.976, and a Jaccard index (IOU) of 0.9535.