Self-Supervised Learning for Organs At Risk and Tumor Segmentation with Uncertainty Quantification
In this study, our goal is to show the impact of self-supervised pre-training of transformers for organ at risk (OAR) and tumor segmentation as compared to costly fully-supervised learning. The proposed algorithm is called Monte Carlo Transformer based U-Net (MC-Swin-U). Unlike many other available models, our approach presents uncertainty quantification with Monte Carlo dropout strategy while generating its voxel-wise prediction. We test and validate the proposed model on both public and one private datasets and evaluate the gross tumor volume (GTV) as well as nearby risky organs' boundaries. We show that self-supervised pre-training approach improves the segmentation scores significantly while providing additional benefits for avoiding large-scale annotation costs.
Code (0)
등록된 구현이 없습니다.
Tasks
SegmentationSelf-Supervised LearningTumor SegmentationUncertainty QuantificationMethods 이 논문이 사용한 방법론
Similar Papers 제목 키워드 기반
A Modality-Adaptive Method for Segmenting Brain Tumors and Organs-at-Risk in Radiation Therapy Planning
In this paper we present a method for simultaneously segmenting brain tumors and an extensive set of organs-at-risk for radiation therapy planning of glioblastomas. The method combines a contrast-adaptive generative mode…
Brain SegmentationSegmentationTumor SegmentationA unified 3D framework for Organs at Risk Localization and Segmentation for Radiation Therapy Planning
Automatic localization and segmentation of organs-at-risk (OAR) in CT are essential pre-processing steps in medical image analysis tasks, such as radiation therapy planning. For instance, the segmentation of OAR surround…
Medical Image AnalysisOrgan SegmentationSegmentationIterative Semi-Supervised Learning for Abdominal Organs and Tumor Segmentation
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…
Computational EfficiencyOrgan SegmentationSegmentationTumor SegmentationLearning from partially labeled data for multi-organ and tumor segmentation
Medical image benchmarks for the segmentation of organs and tumors suffer from the partially labeling issue due to its intensive cost of labor and expertise. Current mainstream approaches follow the practice of one netwo…
Image SegmentationMedical Image SegmentationPartially Labeled DatasetsSegmentation+3Multi-Scale Fusion Methodologies for Head and Neck Tumor Segmentation
Head and Neck (H\&N) organ-at-risk (OAR) and tumor segmentations are essential components of radiation therapy planning. The varying anatomic locations and dimensions of H\&N nodal Gross Tumor Volumes (GTVn) and H\&N pri…
Tumor Segmentation