Back to Search Start Over

ERV-Net: An efficient 3D residual neural network for brain tumor segmentation.

Authors :
Zhou, Xinyu
Li, Xuanya
Hu, Kai
Zhang, Yuan
Chen, Zhineng
Gao, Xieping
Source :
Expert Systems with Applications. May2021, Vol. 170, pN.PAG-N.PAG. 1p.
Publication Year :
2021

Abstract

• We propose an efficient 3D residual neural network for brain tumor segmentation. • We propose a fusion loss function based on Dice and Cross-entropy. • We introduce a concise but effective post-processing method. • The evaluation is performed on the BRATS 2018 dataset. • The results demonstrate that our method outperforms the state-of-the-art approaches. Brain tumors are the most aggressive and mortal cancers, which lead to short life expectancy. A reliable and efficient automatic or semi-automatic segmentation method is significant for clinical practice. In recent years, deep learning-based methods achieve great success in brain tumor segmentation. However, due to the limitation of parameters and computational complexity, there is still much room for improvement in these methods. In this paper, we propose an efficient 3D residual neural network (ERV-Net) for brain tumor segmentation, which has less computational complexity and GPU memory consumption. In ERV-Net, a computation-efficient network, 3D ShuffleNetV2, is firstly utilized as encoder to reduce GPU memory and improve the efficiency of ERV-Net, and then the decoder with residual blocks (Res-decoder) is introduced to avoid degradation. Furthermore, a fusion loss function, which is composed of Dice loss and Cross-entropy loss, is developed to solve the problems of network convergence and data imbalance. Moreover, a concise and effective post-processing method is proposed to refine the coarse segmentation result of ERV-Net. The experimental results on the dataset of multimodal brain tumor segmentation challenge 2018 (BRATS 2018) demonstrate that ERV-Net achieves the best performance with Dice of 81.8%, 91.21% and 86.62% and Hausdorff distance of 2.70 mm, 3.88 mm and 6.79 mm for enhancing tumor, whole tumor and tumor core, respectively. Besides, ERV-Net also achieves high efficiency compared to the state-of-the-art methods. [ABSTRACT FROM AUTHOR]

Details

Language :
English
ISSN :
09574174
Volume :
170
Database :
Academic Search Index
Journal :
Expert Systems with Applications
Publication Type :
Academic Journal
Accession number :
148986717
Full Text :
https://doi.org/10.1016/j.eswa.2021.114566