A Two-Stage Cascade Model with Variational Autoencoders and Attention Gates for MRI Brain Tumor Segmentation

by   Chenggang Lyu, et al.

Automatic MRI brain tumor segmentation is of vital importance for the disease diagnosis, monitoring, and treatment planning. In this paper, 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 second-stage network adopts attention gates and is trained additionally using an expanded dataset formed by the first-stage outputs. On the BraTS 2020 validation dataset, the proposed method achieves the mean Dice score of 0.9041, 0.8350, and 0.7958, and Hausdorff distance (95 4.953, 6.299, and 23.608 for the whole tumor, tumor core, and enhancing tumor, respectively. The corresponding results on the BraTS 2020 testing dataset are 0.8729, 0.8357, and 0.8205 for Dice score, and 11.4288, 19.9690, and 15.6711 for Hausdorff distance.


page 2

page 11


Multi-stage Deep Layer Aggregation for Brain Tumor Segmentation

Gliomas are among the most aggressive and deadly brain tumors. This pape...

Multi-Threshold Attention U-Net (MTAU) based Model for Multimodal Brain Tumor Segmentation in MRI scans

Gliomas are one of the most frequent brain tumors and are classified int...

Multi-step Cascaded Networks for Brain Tumor Segmentation

Automatic brain tumor segmentation method plays an extremely important r...

3D MRI brain tumor segmentation using autoencoder regularization

Automated segmentation of brain tumors from 3D magnetic resonance images...

Kidney tumor segmentation using an ensembling multi-stage deep learning approach. A contribution to the KiTS19 challenge

Precise characterization of the kidney and kidney tumor characteristics ...

Brain tumour segmentation using cascaded 3D densely-connected U-net

Accurate brain tumour segmentation is a crucial step towards improving d...

Redundancy Reduction in Semantic Segmentation of 3D Brain Tumor MRIs

Another year of the multimodal brain tumor segmentation challenge (BraTS...