Brain tumor segmentation with self-ensembled, deeply-supervised 3D U-net
neural networks: a BraTS 2020 challenge solution
- URL: http://arxiv.org/abs/2011.01045v2
- Date: Fri, 27 Nov 2020 15:58:32 GMT
- Title: Brain tumor segmentation with self-ensembled, deeply-supervised 3D U-net
neural networks: a BraTS 2020 challenge solution
- Authors: Theophraste Henry, Alexandre Carre, Marvin Lerousseau, Theo Estienne,
Charlotte Robert, Nikos Paragios, Eric Deutsch
- Abstract summary: We automate and standardize the task of brain tumor segmentation with U-net like neural networks.
Two independent ensembles of models were trained, and each produced a brain tumor segmentation map.
Our solution achieved a Dice of 0.79, 0.89 and 0.84, as well as Hausdorff 95% of 20.4, 6.7 and 19.5mm on the final test dataset.
- Score: 56.17099252139182
- License: http://creativecommons.org/licenses/by/4.0/
- Abstract: Brain tumor segmentation is a critical task for patient's disease management.
In order to automate and standardize this task, we trained multiple U-net like
neural networks, mainly with deep supervision and stochastic weight averaging,
on the Multimodal Brain Tumor Segmentation Challenge (BraTS) 2020 training
dataset. Two independent ensembles of models from two different training
pipelines were trained, and each produced a brain tumor segmentation map. These
two labelmaps per patient were then merged, taking into account the performance
of each ensemble for specific tumor subregions. Our performance on the online
validation dataset with test time augmentation were as follows: Dice of 0.81,
0.91 and 0.85; Hausdorff (95%) of 20.6, 4,3, 5.7 mm for the enhancing tumor,
whole tumor and tumor core, respectively. Similarly, our solution achieved a
Dice of 0.79, 0.89 and 0.84, as well as Hausdorff (95%) of 20.4, 6.7 and 19.5mm
on the final test dataset, ranking us among the top ten teams. More complicated
training schemes and neural network architectures were investigated without
significant performance gain at the cost of greatly increased training time.
Overall, our approach yielded good and balanced performance for each tumor
subregion. Our solution is open sourced at
https://github.com/lescientifik/open_brats2020.
Related papers
- Prediction of brain tumor recurrence location based on multi-modal
fusion and nonlinear correlation learning [55.789874096142285]
We present a deep learning-based brain tumor recurrence location prediction network.
We first train a multi-modal brain tumor segmentation network on the public dataset BraTS 2021.
Then, the pre-trained encoder is transferred to our private dataset for extracting the rich semantic features.
Two decoders are constructed to jointly segment the present brain tumor and predict its future tumor recurrence location.
arXiv Detail & Related papers (2023-04-11T02:45:38Z) - Hybrid Window Attention Based Transformer Architecture for Brain Tumor
Segmentation [28.650980942429726]
We propose a volumetric vision transformer that follows two windowing strategies in attention for extracting fine features.
We trained and evaluated network architecture on the FeTS Challenge 2022 dataset.
Our performance on the online validation dataset is as follows: Dice Similarity Score of 81.71%, 91.38% and 85.40%.
arXiv Detail & Related papers (2022-09-16T03:55:48Z) - HNF-Netv2 for Brain Tumor Segmentation using multi-modal MR Imaging [86.52489226518955]
We extend our HNF-Net to HNF-Netv2 by adding inter-scale and intra-scale semantic discrimination enhancing blocks.
Our method won the RSNA 2021 Brain Tumor AI Challenge Prize (Segmentation Task)
arXiv Detail & Related papers (2022-02-10T06:34:32Z) - 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) - Redundancy Reduction in Semantic Segmentation of 3D Brain Tumor MRIs [2.946960157989204]
This work is a modification of network training process that minimizes redundancy under perturbations.
We evaluated the method on BraTS 2021 validation board, and achieved 0.8600, 0.8868 and 0.9265 average dice for enhanced tumor core, tumor core and whole tumor.
Our team (NVAUTO) submission was the top performing in terms of ET and TC scores and within top 10 performing teams in terms of WT scores.
arXiv Detail & Related papers (2021-11-01T07:39:06Z) - Brain Tumor Segmentation and Survival Prediction using Automatic Hard
mining in 3D CNN Architecture [0.30098583327398537]
We utilize 3-D fully convolutional neural networks (CNN) to segment gliomas and its constituents from multimodal Magnetic Resonance Images (MRI)
The architecture uses dense connectivity patterns to reduce the number of weights and residual connections and is 0.448 with weights obtained from training this model with BraTS 2018 dataset.
Hard mining is done during training to train for the difficult cases of segmentation tasks by increasing the dice similarity coefficient (DSC) threshold to choose the hard cases as epoch increases.
arXiv Detail & Related papers (2021-01-05T14:34:16Z) - H2NF-Net for Brain Tumor Segmentation using Multimodal MR Imaging: 2nd
Place Solution to BraTS Challenge 2020 Segmentation Task [96.49879910148854]
Our H2NF-Net uses the single and cascaded HNF-Nets to segment different brain tumor sub-regions.
We trained and evaluated our model on the Multimodal Brain Tumor Challenge (BraTS) 2020 dataset.
Our method won the second place in the BraTS 2020 challenge segmentation task out of nearly 80 participants.
arXiv Detail & Related papers (2020-12-30T20:44:55Z) - Solving Mixed Integer Programs Using Neural Networks [57.683491412480635]
This paper applies learning to the two key sub-tasks of a MIP solver, generating a high-quality joint variable assignment, and bounding the gap in objective value between that assignment and an optimal one.
Our approach constructs two corresponding neural network-based components, Neural Diving and Neural Branching, to use in a base MIP solver such as SCIP.
We evaluate our approach on six diverse real-world datasets, including two Google production datasets and MIPLIB, by training separate neural networks on each.
arXiv Detail & Related papers (2020-12-23T09:33:11Z) - A Two-Stage Cascade Model with Variational Autoencoders and Attention
Gates for MRI Brain Tumor Segmentation [2.055949720959582]
We propose a two-stage encoder-decoder based model for brain tumor subregional segmentation.
Variational autoencoder regularization is utilized in both stages to prevent the overfitting issue.
The proposed method achieves the mean Dice score of 0.9041, 0.8350, and 0.7958, and Hausdorff distance 95% of 4.953, 6.299, and 23.608 for the whole tumor, tumor core, and enhancing tumor.
arXiv Detail & Related papers (2020-11-04T05:55:06Z) - DR-Unet104 for Multimodal MRI brain tumor segmentation [7.786297008452384]
We propose a 2D deep residual Unet with 104 convolutional layers (DR-Unet104) for lesion segmentation in brain MRIs.
We make multiple additions to the Unet architecture, including adding the 'bottleneck' residual block to the Unet encoder and adding dropout after each convolution block stack.
We produced a competitive lesion segmentation architecture, despite only 2D convolutions, having the added benefit that it can be used on lower power computers.
arXiv Detail & Related papers (2020-11-04T01:24:26Z)
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.