ME-Net: Multi-Encoder Net Framework for Brain Tumor Segmentation
- URL: http://arxiv.org/abs/2203.11213v1
- Date: Mon, 21 Mar 2022 14:42:05 GMT
- Title: ME-Net: Multi-Encoder Net Framework for Brain Tumor Segmentation
- Authors: Wenbo Zhang, Guang Yang, He Huang, Weiji Yang, Xiaomei Xu, Yongkai
Liu, Xiaobo Lai
- Abstract summary: We propose a model for brain tumor segmentation with multiple encoders.
Four encoders correspond to the four modalities of the MRI image, perform one-to-one feature extraction, and then merge the feature maps of the four modalities into the decoder.
We also introduced a new loss function named "Categorical Dice", and set different weights for different segmented regions at the same time, which solved the problem of voxel imbalance.
- Score: 6.643336433892116
- License: http://creativecommons.org/licenses/by-nc-nd/4.0/
- Abstract: Glioma is the most common and aggressive brain tumor. Magnetic resonance
imaging (MRI) plays a vital role to evaluate tumors for the arrangement of
tumor surgery and the treatment of subsequent procedures. However, the manual
segmentation of the MRI image is strenuous, which limits its clinical
application. With the development of deep learning, a large number of automatic
segmentation methods have been developed, but most of them stay in 2D images,
which leads to subpar performance. Moreover, the serious voxel imbalance
between the brain tumor and the background as well as the different sizes and
locations of the brain tumor makes the segmentation of 3D images a challenging
problem. Aiming at segmenting 3D MRI, we propose a model for brain tumor
segmentation with multiple encoders. The structure contains four encoders and
one decoder. The four encoders correspond to the four modalities of the MRI
image, perform one-to-one feature extraction, and then merge the feature maps
of the four modalities into the decoder. This method reduces the difficulty of
feature extraction and greatly improves model performance. We also introduced a
new loss function named "Categorical Dice", and set different weights for
different segmented regions at the same time, which solved the problem of voxel
imbalance. We evaluated our approach using the online BraTS 2020 Challenge
verification. Our proposed method can achieve promising results in the
validation set compared to the state-of-the-art approaches with Dice scores of
0.70249, 0.88267, and 0.73864 for the intact tumor, tumor core, and enhanced
tumor, respectively.
Related papers
- Exploration of Multi-Scale Image Fusion Systems in Intelligent Medical Image Analysis [3.881664394416534]
It is necessary to perform automatic segmentation of brain tumors on MRI images.
This project intends to build an MRI algorithm based on U-Net.
arXiv Detail & Related papers (2024-05-23T04:33:12Z) - Glioblastoma Tumor Segmentation using an Ensemble of Vision Transformers [0.0]
Glioblastoma is one of the most aggressive and deadliest types of brain cancer.
Brain Radiology Aided by Intelligent Neural NETworks (BRAINNET) generates robust tumor segmentation maks.
arXiv Detail & Related papers (2023-11-09T18:55:27Z) - 3DSAM-adapter: Holistic adaptation of SAM from 2D to 3D for promptable tumor segmentation [52.699139151447945]
We propose a novel adaptation method for transferring the segment anything model (SAM) from 2D to 3D for promptable medical image segmentation.
Our model can outperform domain state-of-the-art medical image segmentation models on 3 out of 4 tasks, specifically by 8.25%, 29.87%, and 10.11% for kidney tumor, pancreas tumor, colon cancer segmentation, and achieve similar performance for liver tumor segmentation.
arXiv Detail & Related papers (2023-06-23T12:09:52Z) - A Radiomics-Incorporated Deep Ensemble Learning Model for
Multi-Parametric MRI-based Glioma Segmentation [5.890417404600585]
We developed a deep ensemble learning model with a radiomics spatial encoding execution for improved glioma segmentation accuracy.
This model was developed using 369 glioma patients with a 4-modality mp-MRI protocol: T1, contrast-enhanced T1 (T1-Ce), T2, and FLAIR.
arXiv Detail & Related papers (2023-03-19T02:16:55Z) - Video4MRI: An Empirical Study on Brain Magnetic Resonance Image
Analytics with CNN-based Video Classification Frameworks [60.42012344842292]
3D CNN-based models dominate the field of magnetic resonance image (MRI) analytics.
In this paper, four datasets of Alzheimer's and Parkinson's disease recognition are utilized in experiments.
In terms of efficiency, the video framework performs better than 3D-CNN models by 5% - 11% with 50% - 66% less trainable parameters.
arXiv Detail & Related papers (2023-02-24T15:26:31Z) - Moving from 2D to 3D: volumetric medical image classification for rectal
cancer staging [62.346649719614]
preoperative discrimination between T2 and T3 stages is arguably both the most challenging and clinically significant task for rectal cancer treatment.
We present a volumetric convolutional neural network to accurately discriminate T2 from T3 stage rectal cancer with rectal MR volumes.
arXiv Detail & Related papers (2022-09-13T07:10:14Z) - CKD-TransBTS: Clinical Knowledge-Driven Hybrid Transformer with
Modality-Correlated Cross-Attention for Brain Tumor Segmentation [37.39921484146194]
Brain tumor segmentation in magnetic resonance image (MRI) is crucial for brain tumor diagnosis, cancer management and research purposes.
With the great success of the ten-year BraTS challenges, a lot of outstanding BTS models have been proposed to tackle the difficulties of BTS in different technical aspects.
We propose a clinical knowledge-driven brain tumor segmentation model, called CKD-TransBTS.
arXiv Detail & Related papers (2022-07-15T09:35:29Z) - Triplet Contrastive Learning for Brain Tumor Classification [99.07846518148494]
We present a novel approach of directly learning deep embeddings for brain tumor types, which can be used for downstream tasks such as classification.
We evaluate our method on an extensive brain tumor dataset which consists of 27 different tumor classes, out of which 13 are defined as rare.
arXiv Detail & Related papers (2021-08-08T11:26:34Z) - 3D AGSE-VNet: An Automatic Brain Tumor MRI Data Segmentation Framework [3.0261170901794308]
Glioma is the most common brain malignant tumor, with a high morbidity rate and a mortality rate of more than three percent.
The main method of acquiring brain tumors in the clinic is MRI of brain tumor regions from multi-modal MRI scan images.
We propose an automatic brain tumor MRI data segmentation framework which is called AGSE-VNet.
arXiv Detail & Related papers (2021-07-26T09:04:59Z) - Modality Completion via Gaussian Process Prior Variational Autoencoders
for Multi-Modal Glioma Segmentation [75.58395328700821]
We propose a novel model, Multi-modal Gaussian Process Prior Variational Autoencoder (MGP-VAE), to impute one or more missing sub-modalities for a patient scan.
MGP-VAE can leverage the Gaussian Process (GP) prior on the Variational Autoencoder (VAE) to utilize the subjects/patients and sub-modalities correlations.
We show the applicability of MGP-VAE on brain tumor segmentation where either, two, or three of four sub-modalities may be missing.
arXiv Detail & Related papers (2021-07-07T19:06:34Z) - 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)
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.