Expectation-Maximization Regularized Deep Learning for Weakly Supervised
Tumor Segmentation for Glioblastoma
- URL: http://arxiv.org/abs/2101.08757v3
- Date: Thu, 11 Mar 2021 21:08:01 GMT
- Title: Expectation-Maximization Regularized Deep Learning for Weakly Supervised
Tumor Segmentation for Glioblastoma
- Authors: Chao Li, Wenjian Huang, Xi Chen, Yiran Wei, Stephen J. Price,
Carola-Bibiane Sch\"onlieb
- Abstract summary: We present an Expectation-Maximization (EM) Regularized Deep Learning (EMReDL) model for the weakly supervised tumor segmentation.
The proposed framework was tailored to glioblastoma, a type of malignant tumor characterized by its diffuse infiltration into the surrounding brain tissue.
- Score: 8.24450401153384
- License: http://creativecommons.org/licenses/by-nc-nd/4.0/
- Abstract: We present an Expectation-Maximization (EM) Regularized Deep Learning
(EMReDL) model for the weakly supervised tumor segmentation. The proposed
framework was tailored to glioblastoma, a type of malignant tumor characterized
by its diffuse infiltration into the surrounding brain tissue, which poses
significant challenge to treatment target and tumor burden estimation based on
conventional structural MRI. Although physiological MRI can provide more
specific information regarding tumor infiltration, the relatively low
resolution hinders a precise full annotation. This has motivated us to develop
a weakly supervised deep learning solution that exploits the partial labelled
tumor regions.
EMReDL contains two components: a physiological prior prediction model and
EM-regularized segmentation model. The physiological prior prediction model
exploits the physiological MRI by training a classifier to generate a
physiological prior map. This map was passed to the segmentation model for
regularization using the EM algorithm. We evaluated the model on a glioblastoma
dataset with the available pre-operative multiparametric MRI and recurrence
MRI. EMReDL was shown to effectively segment the infiltrated tumor from the
partially labelled region of potential infiltration. The segmented core and
infiltrated tumor showed high consistency with the tumor burden labelled by
experts. The performance comparison showed that EMReDL achieved higher accuracy
than published state-of-the-art models. On MR spectroscopy, the segmented
region showed more aggressive features than other partial labelled region. The
proposed model can be generalized to other segmentation tasks with partial
labels, with the CNN architecture flexible in the framework.
Related papers
- Mask-Enhanced Segment Anything Model for Tumor Lesion Semantic Segmentation [48.107348956719775]
We introduce Mask-Enhanced SAM (M-SAM), an innovative architecture tailored for 3D tumor lesion segmentation.
We propose a novel Mask-Enhanced Adapter (MEA) within M-SAM that enriches the semantic information of medical images with positional data from coarse segmentation masks.
Our M-SAM achieves high segmentation accuracy and also exhibits robust generalization.
arXiv Detail & Related papers (2024-03-09T13:37:02Z) - Towards Generalizable Tumor Synthesis [48.45704270448412]
Tumor synthesis enables the creation of artificial tumors in medical images, facilitating the training of AI models for tumor detection and segmentation.
This paper made a progressive stride toward generalizable tumor synthesis by leveraging a critical observation.
We have ascertained that generative AI models, e.g., Diffusion Models, can create realistic tumors generalized to a range of organs even when trained on a limited number of tumor examples from only one organ.
arXiv Detail & Related papers (2024-02-29T18:57:39Z) - Cross-modality Guidance-aided Multi-modal Learning with Dual Attention
for MRI Brain Tumor Grading [47.50733518140625]
Brain tumor represents one of the most fatal cancers around the world, and is very common in children and the elderly.
We propose a novel cross-modality guidance-aided multi-modal learning with dual attention for addressing the task of MRI brain tumor grading.
arXiv Detail & Related papers (2024-01-17T07:54:49Z) - Fully Automated Tumor Segmentation for Brain MRI data using Multiplanner
UNet [0.29998889086656577]
This study evaluates the efficacy of the Multi-Planner U-Net (MPUnet) approach in segmenting different tumor subregions across three challenging datasets.
arXiv Detail & Related papers (2024-01-12T10:46:19Z) - Treatment-aware Diffusion Probabilistic Model for Longitudinal MRI
Generation and Diffuse Glioma Growth Prediction [0.5806504980491878]
We present a novel end-to-end network capable of generating future tumor masks and realistic MRIs of how the tumor will look at any future time points.
Our approach is based on cutting-edge diffusion probabilistic models and deep-segmentation neural networks.
arXiv Detail & Related papers (2023-09-11T12:12:52Z) - Brain tumor multi classification and segmentation in MRI images using
deep learning [3.1248717814228923]
The classification model is based on the EfficientNetB1 architecture and is trained to classify images into four classes: meningioma, glioma, pituitary adenoma, and no tumor.
The segmentation model is based on the U-Net architecture and is trained to accurately segment the tumor from the MRI images.
arXiv Detail & Related papers (2023-04-20T01:32:55Z) - Brain Tumor MRI Classification using a Novel Deep Residual and Regional
CNN [0.0]
A novel deep residual and regional-based Res-BRNet Convolutional Neural Network (CNN) is developed for effective brain tumor (Magnetic Resonance Imaging) MRI classification.
The efficiency of the developed Res-BRNet is evaluated on a standard dataset; collected from Kaggle and Figshare containing various tumor categories.
Experiments prove that the developed Res-BRNet outperforms the standard CNN models and attained excellent performances.
arXiv Detail & Related papers (2022-11-29T20:14:13Z) - MAG-Net: Mutli-task attention guided network for brain tumor
segmentation and classification [0.9176056742068814]
This paper proposes multi-task attention guided encoder-decoder network (MAG-Net) to classify and segment the brain tumor regions using MRI images.
The model achieved promising results as compared to existing state-of-the-art models.
arXiv Detail & Related papers (2021-07-26T16:51:00Z) - Spectral-Spatial Recurrent-Convolutional Networks for In-Vivo
Hyperspectral Tumor Type Classification [49.32653090178743]
We demonstrate the feasibility of in-vivo tumor type classification using hyperspectral imaging and deep learning.
Our best model achieves an AUC of 76.3%, significantly outperforming previous conventional and deep learning methods.
arXiv Detail & Related papers (2020-07-02T12:00:53Z) - Scale-Space Autoencoders for Unsupervised Anomaly Segmentation in Brain
MRI [47.26574993639482]
We show improved anomaly segmentation performance and the general capability to obtain much more crisp reconstructions of input data at native resolution.
The modeling of the laplacian pyramid further enables the delineation and aggregation of lesions at multiple scales.
arXiv Detail & Related papers (2020-06-23T09:20:42Z) - Stan: Small tumor-aware network for breast ultrasound image segmentation [68.8204255655161]
We propose a novel deep learning architecture called Small Tumor-Aware Network (STAN) to improve the performance of segmenting tumors with different size.
The proposed approach outperformed the state-of-the-art approaches in segmenting small breast tumors.
arXiv Detail & Related papers (2020-02-03T22:25:01Z)
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.