Two-Stage Hybrid Supervision Framework for Fast, Low-resource, and
Accurate Organ and Pan-cancer Segmentation in Abdomen CT
- URL: http://arxiv.org/abs/2309.05405v2
- Date: Thu, 16 Nov 2023 04:04:59 GMT
- Title: Two-Stage Hybrid Supervision Framework for Fast, Low-resource, and
Accurate Organ and Pan-cancer Segmentation in Abdomen CT
- Authors: Wentao Liu, Tong Tian, Weijin Xu, Lemeng Wang, Haoyuan Li, Huihua Yang
- Abstract summary: We propose a hybrid supervised framework, StMt, that integrates self-training and mean teacher for the segmentation of abdominal organs and tumors.
Experiments on the validation set of FLARE2023 demonstrate that our method achieves excellent segmentation performance.
- Score: 12.506232623163665
- License: http://arxiv.org/licenses/nonexclusive-distrib/1.0/
- Abstract: Abdominal organ and tumour segmentation has many important clinical
applications, such as organ quantification, surgical planning, and disease
diagnosis. However, manual assessment is inherently subjective with
considerable inter- and intra-expert variability. In the paper, we propose a
hybrid supervised framework, StMt, that integrates self-training and mean
teacher for the segmentation of abdominal organs and tumors using partially
labeled and unlabeled data. We introduce a two-stage segmentation pipeline and
whole-volume-based input strategy to maximize segmentation accuracy while
meeting the requirements of inference time and GPU memory usage. Experiments on
the validation set of FLARE2023 demonstrate that our method achieves excellent
segmentation performance as well as fast and low-resource model inference. Our
method achieved an average DSC score of 89.79\% and 45.55 \% for the organs and
lesions on the validation set and the average running time and area under GPU
memory-time cure are 11.25s and 9627.82MB, respectively.
Related papers
- MAST-Pro: Dynamic Mixture-of-Experts for Adaptive Segmentation of Pan-Tumors with Knowledge-Driven Prompts [54.915060471994686]
We propose MAST-Pro, a novel framework that integrates dynamic Mixture-of-Experts (D-MoE) and knowledge-driven prompts for pan-tumor segmentation.
Specifically, text and anatomical prompts provide domain-specific priors guiding tumor representation learning, while D-MoE dynamically selects experts to balance generic and tumor-specific feature learning.
Experiments on multi-anatomical tumor datasets demonstrate that MAST-Pro outperforms state-of-the-art approaches, achieving up to a 5.20% improvement in average improvement while reducing trainable parameters by 91.04%, without compromising accuracy.
arXiv Detail & Related papers (2025-03-18T15:39:44Z) - Deep Learning-Based Automated Workflow for Accurate Segmentation and Measurement of Abdominal Organs in CT Scans [0.0]
The purpose of this study is to develop and validate an automated workflow for the segmentation and measurement of abdominal organs in CT scans.
The proposed approach offers an automated, efficient, and reliable solution for abdominal organ measurement in CT scans.
arXiv Detail & Related papers (2025-03-13T06:50:44Z) - Enhancing Brain Tumor Segmentation Using Channel Attention and Transfer learning [5.062500255359342]
We present an enhanced ResUNet architecture for automatic brain tumor segmentation.
The EfficientNetB0 encoder leverages pre-trained features to improve feature extraction efficiency.
The channel attention mechanism enhances the model's focus on tumor-relevant features.
arXiv Detail & Related papers (2025-01-19T23:58:16Z) - A Localization-to-Segmentation Framework for Automatic Tumor
Segmentation in Whole-Body PET/CT Images [8.0523823243864]
This paper proposes a localization-to-segmentation framework (L2SNet) for precise tumor segmentation.
L2SNet first localizes the possible lesions in the lesion localization phase and then uses the location cues to shape the segmentation results in the lesion segmentation phase.
Experiments with the MII Automated Lesion in Whole-Body FDG-PET/CT challenge dataset show that our method achieved a competitive result.
arXiv Detail & Related papers (2023-09-11T13:39:15Z) - A region and category confidence-based multi-task network for carotid
ultrasound image segmentation and classification [6.162577404860473]
We propose a multi-task learning framework (RCCM-Net) for ultrasound carotid plaque segmentation and classification.
The framework uses a region confidence module (RCM) and a sample category confidence module ( CCM) to exploit the correlation between these two tasks.
The results show that the proposed method can improve both segmentation and classification performance compared to existing single-task networks.
arXiv Detail & Related papers (2023-07-02T14:43:59Z) - Segmentation of glioblastomas in early post-operative multi-modal MRI
with deep neural networks [33.51490233427579]
Two state-of-the-art neural network architectures for pre-operative segmentation were trained for the task.
The best performance achieved was a 61% Dice score, and the best classification performance was about 80% balanced accuracy.
The predicted segmentations can be used to accurately classify the patients into those with residual tumor, and those with gross total resection.
arXiv Detail & Related papers (2023-04-18T10:14:45Z) - Generative Adversarial Networks for Weakly Supervised Generation and Evaluation of Brain Tumor Segmentations on MR Images [0.0]
This work presents a weakly supervised approach to segment anomalies in 2D magnetic resonance images.
We train a generative adversarial network (GAN) that converts cancerous images to healthy variants.
Non-cancerous variants can also be used to evaluate the segmentations in a weakly supervised fashion.
arXiv Detail & Related papers (2022-11-10T00:04:46Z) - Combining Hybrid Architecture and Pseudo-label for Semi-supervised
Abdominal Organ Segmentation [8.392397691020232]
In this work, we employ a hybrid architecture (PHTrans) with CNN and Transformer for both teacher and student models to generate precise pseudo-labels.
Experiments on the validation set of FLARE2022 demonstrate that our method achieves excellent segmentation performance as well as fast and low-resource model inference.
arXiv Detail & Related papers (2022-07-23T13:02:43Z) - Improving Classification Model Performance on Chest X-Rays through Lung
Segmentation [63.45024974079371]
We propose a deep learning approach to enhance abnormal chest x-ray (CXR) identification performance through segmentations.
Our approach is designed in a cascaded manner and incorporates two modules: a deep neural network with criss-cross attention modules (XLSor) for localizing lung region in CXR images and a CXR classification model with a backbone of a self-supervised momentum contrast (MoCo) model pre-trained on large-scale CXR data sets.
arXiv Detail & Related papers (2022-02-22T15:24:06Z) - EMT-NET: Efficient multitask network for computer-aided diagnosis of
breast cancer [58.720142291102135]
We propose an efficient and light-weighted learning architecture to classify and segment breast tumors simultaneously.
We incorporate a segmentation task into a tumor classification network, which makes the backbone network learn representations focused on tumor regions.
The accuracy, sensitivity, and specificity of tumor classification is 88.6%, 94.1%, and 85.3%, respectively.
arXiv Detail & Related papers (2022-01-13T05:24:40Z) - Lung Cancer Lesion Detection in Histopathology Images Using Graph-Based
Sparse PCA Network [93.22587316229954]
We propose a graph-based sparse principal component analysis (GS-PCA) network, for automated detection of cancerous lesions on histological lung slides stained by hematoxylin and eosin (H&E)
We evaluate the performance of the proposed algorithm on H&E slides obtained from an SVM K-rasG12D lung cancer mouse model using precision/recall rates, F-score, Tanimoto coefficient, and area under the curve (AUC) of the receiver operator characteristic (ROC)
arXiv Detail & Related papers (2021-10-27T19:28:36Z) - Learning Inductive Attention Guidance for Partially Supervised
Pancreatic Ductal Adenocarcinoma Prediction [73.96902906734522]
Pancreatic ductal adenocarcinoma (PDAC) is the third most common cause of cancer death in the United States.
In this paper, we consider a partially supervised setting, where cheap image-level annotations are provided for all the training data, and the costly per-voxel annotations are only available for a subset of them.
We propose an Inductive Attention Guidance Network (IAG-Net) to jointly learn a global image-level classifier for normal/PDAC classification and a local voxel-level classifier for semi-supervised PDAC segmentation.
arXiv Detail & Related papers (2021-05-31T08:16:09Z) - Segmentation of the Myocardium on Late-Gadolinium Enhanced MRI based on
2.5 D Residual Squeeze and Excitation Deep Learning Model [55.09533240649176]
The aim of this work is to develop an accurate automatic segmentation method based on deep learning models for the myocardial borders on LGE-MRI.
A total number of 320 exams (with a mean number of 6 slices per exam) were used for training and 28 exams used for testing.
The performance analysis of the proposed ensemble model in the basal and middle slices was similar as compared to intra-observer study and slightly lower at apical slices.
arXiv Detail & Related papers (2020-05-27T20:44:38Z) - A Global Benchmark of Algorithms for Segmenting Late Gadolinium-Enhanced
Cardiac Magnetic Resonance Imaging [90.29017019187282]
" 2018 Left Atrium Challenge" using 154 3D LGE-MRIs, currently the world's largest cardiac LGE-MRI dataset.
Analyse of the submitted algorithms using technical and biological metrics was performed.
Results show the top method achieved a dice score of 93.2% and a mean surface to a surface distance of 0.7 mm.
arXiv Detail & Related papers (2020-04-26T08:49:17Z)
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.