Iterative Semi-Supervised Learning for Abdominal Organs and Tumor
Segmentation
- URL: http://arxiv.org/abs/2310.01159v1
- Date: Mon, 2 Oct 2023 12:45:13 GMT
- Title: Iterative Semi-Supervised Learning for Abdominal Organs and Tumor
Segmentation
- Authors: Jiaxin Zhuang and Luyang Luo and Zhixuan Chen, and Linshan Wu
- Abstract summary: The FLARE23 challenge provides a large-scale dataset with both partially and fully annotated data.
We propose to use the strategy of Semi-Supervised Learning (SSL) and iterative pseudo labeling to address FLARE23.
Our approach achieves an average DSC score of 89.63% for organs and 46.07% for tumors on online validation leaderboard.
- Score: 4.952008176585512
- License: http://arxiv.org/licenses/nonexclusive-distrib/1.0/
- Abstract: Deep-learning (DL) based methods are playing an important role in the task of
abdominal organs and tumors segmentation in CT scans. However, the large
requirements of annotated datasets heavily limit its development. The FLARE23
challenge provides a large-scale dataset with both partially and fully
annotated data, which also focuses on both segmentation accuracy and
computational efficiency. In this study, we propose to use the strategy of
Semi-Supervised Learning (SSL) and iterative pseudo labeling to address
FLARE23. Initially, a deep model (nn-UNet) trained on datasets with complete
organ annotations (about 220 scans) generates pseudo labels for the whole
dataset. These pseudo labels are then employed to train a more powerful
segmentation model. Employing the FLARE23 dataset, our approach achieves an
average DSC score of 89.63% for organs and 46.07% for tumors on online
validation leaderboard. For organ segmentation, We obtain 0.9007\% DSC and
0.9493\% NSD. For tumor segmentation, we obtain 0.3785% DSC and 0.2842% NSD.
Our code is available at https://github.com/USTguy/Flare23.
Related papers
- SALT: Introducing a Framework for Hierarchical Segmentations in Medical Imaging using Softmax for Arbitrary Label Trees [1.004700727815227]
This study introduces a novel segmentation technique for CT imaging, which leverages conditional probabilities to map the hierarchical structure of anatomical landmarks.
The model was developed using the SAROS dataset from The Cancer Imaging Archive (TCIA), comprising 900 body region segmentations from 883 patients.
Performance was assessed using the Dice score across various datasets, including SAROS, CT-ORG, FLARE22, LCTSC, LUNA16, and WORD.
arXiv Detail & Related papers (2024-07-11T21:33:08Z) - TotalSegmentator MRI: Sequence-Independent Segmentation of 59 Anatomical Structures in MR images [62.53931644063323]
In this study we extended the capabilities of TotalSegmentator to MR images.
We trained an nnU-Net segmentation algorithm on this dataset and calculated similarity coefficients (Dice) to evaluate the model's performance.
The model significantly outperformed two other publicly available segmentation models (Dice score 0.824 versus 0.762; p0.001 and 0.762 versus 0.542; p)
arXiv Detail & Related papers (2024-05-29T20:15:54Z) - Towards Unifying Anatomy Segmentation: Automated Generation of a
Full-body CT Dataset via Knowledge Aggregation and Anatomical Guidelines [113.08940153125616]
We generate a dataset of whole-body CT scans with $142$ voxel-level labels for 533 volumes providing comprehensive anatomical coverage.
Our proposed procedure does not rely on manual annotation during the label aggregation stage.
We release our trained unified anatomical segmentation model capable of predicting $142$ anatomical structures on CT data.
arXiv Detail & Related papers (2023-07-25T09:48:13Z) - The impact of training dataset size and ensemble inference strategies on
head and neck auto-segmentation [0.0]
Convolutional neural networks (CNNs) are increasingly being used to automate segmentation of organs-at-risk in radiotherapy.
We investigated how much data is required to train accurate and robust head and neck auto-segmentation models.
An established 3D CNN was trained from scratch with different sized datasets (25-1000 scans) to segment the brainstem, parotid glands and spinal cord in CTs.
We evaluated multiple ensemble techniques to improve the performance of these models.
arXiv Detail & Related papers (2023-03-30T12:14:07Z) - TotalSegmentator: robust segmentation of 104 anatomical structures in CT
images [48.50994220135258]
We present a deep learning segmentation model for body CT images.
The model can segment 104 anatomical structures relevant for use cases such as organ volumetry, disease characterization, and surgical or radiotherapy planning.
arXiv Detail & Related papers (2022-08-11T15:16:40Z) - Reciprocal Adversarial Learning for Brain Tumor Segmentation: A Solution
to BraTS Challenge 2021 Segmentation Task [17.648013128690216]
This paper proposes an adversarial learning based training approach for brain tumor segmentation task.
We trained and evaluated network architecture on the RSNA-ASNR-MICCAI BraTS 2021 dataset.
Our approach achieved a Dice Similarity Score of 84.55%, 90.46% and 85.30%, as well as Hausdorff Distance (95%) of 13.48 mm, 6.32 mm and 16.98 mm.
arXiv Detail & Related papers (2022-01-11T04:44:29Z) - CvS: Classification via Segmentation For Small Datasets [52.821178654631254]
This paper presents CvS, a cost-effective classifier for small datasets that derives the classification labels from predicting the segmentation maps.
We evaluate the effectiveness of our framework on diverse problems showing that CvS is able to achieve much higher classification results compared to previous methods when given only a handful of examples.
arXiv Detail & Related papers (2021-10-29T18:41:15Z) - COVID-19 identification from volumetric chest CT scans using a
progressively resized 3D-CNN incorporating segmentation, augmentation, and
class-rebalancing [4.446085353384894]
COVID-19 is a global pandemic disease overgrowing worldwide.
Computer-aided screening tools with greater sensitivity is imperative for disease diagnosis and prognosis.
This article proposes a 3D Convolutional Neural Network (CNN)-based classification approach.
arXiv Detail & Related papers (2021-02-11T18:16:18Z) - 3D medical image segmentation with labeled and unlabeled data using
autoencoders at the example of liver segmentation in CT images [58.720142291102135]
This work investigates the potential of autoencoder-extracted features to improve segmentation with a convolutional neural network.
A convolutional autoencoder was used to extract features from unlabeled data and a multi-scale, fully convolutional CNN was used to perform the target task of 3D liver segmentation in CT images.
arXiv Detail & Related papers (2020-03-17T20:20:43Z) - VerSe: A Vertebrae Labelling and Segmentation Benchmark for
Multi-detector CT Images [121.31355003451152]
Large Scale Vertebrae Challenge (VerSe) was organised in conjunction with the International Conference on Medical Image Computing and Computer Assisted Intervention (MICCAI) in 2019 and 2020.
We present the the results of this evaluation and further investigate the performance-variation at vertebra-level, scan-level, and at different fields-of-view.
arXiv Detail & Related papers (2020-01-24T21:09:18Z)
This list is automatically generated from the titles and abstracts of the papers in this site.
This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.