Intelligent Segmentor: Self Supervised Deep Learning based Multi Organ and Tumor Segmentation with Pseudo Lables Generation from CT Images
P. Savitha1,*, Laxmi Raja1, R. Santhosh1
1Department of Computer Science and Engineering, Faculty of Engineering, Karpagam Academy of Higher Education
Coimbatore, Tamil Nadu, India
Emails: saviinfo@gmail.com; santhoshrd@gmail.com; laxmirajaphd@gmail.com
|
Abstract
Multi Organ and tumor segmentation is the challenging task in medical imaging and surgical planning scenarios due to its diverse applications includes lesions and organs measurements and disease diagnosis respectively. Although collecting and examining labels for all classes pose severe challenges. Furthermore, Graphical Processing Unit (GPU) optimization emerge as another critical factor for multi organ and tumor segmentation. To address the mentioned conventional challenge, we designed a deep learning-based model named “Intelligent Segmentor” which performs automated segmentation in end-to-end fashion with novel semi supervised training approach. Initially, the obtained multi organ CT images is then subjected to pre-processing in terms of geometric standardization, noise removal, and intensity normalization respectively. The pre-processed image is then further provided to dual view training for effective Pseudolabel generation. The labelled data along with generated pseudolabels are provided to train the model for amplifying the model performance. After that, there are two inputs are provided to the designed segmentation model which includes dual encoders such as GoogleNet and VGG-16 for contextual and spatial information extraction in five stages, Tweaked Feature Pyramidal Network (TFPN) for dimensionality reduction and side features extraction, and Gated Fusion Module (GFM) for fusing the side features to form unified feature map. Finally, the unified feature map is the examined through convolution layers for multi organ and tumor output. We adopted FLARE 2023 dataset for validating the proposed work with existing works on 13 various organs and tumor segmentation tasks. From the results, the proposed research achieves better Dice Similarity Coefficient (DSC) and Normalized Surface Dice (NSD) through online validation and final testing than the existing works. |
Received: February 05, 2024 Revised: April 14, 2024 Accepted: July 02, 2024
Keywords: Multi Organ and Tumor Segmentation; Computed Tomography (CT); Deep Learning (DL); Pseudo Label; Self Supervised