A Multi Brain Tumor Region Segmentation Model Based on 3D U-Net
Source abstract
Abstract
Accurate segmentation of different brain tumor regions from MR images is of great significance in the diagnosis and treatment of brain tumors. In this paper, an enhanced 3D U-Net model was proposed to address the shortcomings of 2D U-Net in the segmentation tasks of brain tumors. While retaining the U-shaped characteristics of the original U-Net network, an enhanced encoding module and decoding module were designed to increase the extraction and utilization of image features. Then, a hybrid loss function combining the binary cross-entropy loss function and dice similarity coefficient was adopted to speed up the model’s convergence and to achieve accurate and fast automatic segmentation. The model’s performance in the segmentation of brain tumor’s whole tumor region, tumor core region, and enhanced tumor region was studied. The results showed that the proposed 3D U-Net model can achieve better segmentation performance, especially for the tumor core region and enhanced tumor region tumor regions.
Source: crossref · Quality: baseline
Research context
OpenAlex topics: Brain Tumor Detection and Classification · Advanced Neural Network Applications · Medical Image Segmentation Techniques
SciVX discovery tags are terminology-based. OpenAlex topics are source-provided classifications.
Affiliations
School of Medical Technology and Engineering, Henan University of Science and Technology, Luoyang 471023, China
References (35)
Show source-provided references
1. Review of Brain Tumor Segmentation Methods Based on Convolutional Neural Networks
2. The multimodal brain tumor image segmentation benchmark (BraTS)10.1109/TMI.2014.2377694 ↗
3. Masood, M., Nazir, T., Nawaz, M., Mehmood, A., Rashid, J., Kwon, H.-Y., Mahmood, T., and Hussain, A. (2021). A Novel Deep Learning Method for Recognition and Classification of Brain Tumors from MRI Images. Diagnostics, 11.10.3390/diagnostics11050744 ↗
4. Hao, L., Guanhua, W., and Qiang, Z. (2019). Optimization of Dice Loss Function for 3D Brain Tumor Segmentation. China Med. Devices, 34.
5. Mishra, P., Garg, A., and Gupta, D. (2020, January 6–7). Review on brain tumor segmentation: Hard and soft computing approaches. Proceedings of the International Conference on Image Processing and Capsule Networks, ICIPCN 2020, Bangkok, Thailand.10.1007/978-3-030-51859-2_18 ↗
6. SegNet: A Deep Convolutional Encoder-Decoder Architecture for Image Segmentation10.1109/TPAMI.2016.2644615 ↗
7. DeepLab-Based Spatial Feature Extraction for Hyperspectral Image Classification10.1109/LGRS.2018.2871507 ↗
8. Fully Convolutional Networks for Semantic Segmentation
9. Brain Tumor Segmentation Using Convolutional Neural Networks in MRI Images10.1109/TMI.2016.2538465 ↗
10. Jiang, Z., Ding, C., and Liu, M. (2020). Two-Stage Cascaded U-Net: 1st Place Solution to BraTS Challenge 2019 Segmentation Task, Springer.10.1007/978-3-030-46640-4_22 ↗
11. Fast and robust segmentation of the striatum using deep convolutional neural networks10.1016/j.jneumeth.2016.10.007 ↗
12. Sundaresan, V., Griffanti, L., and Jenkinson, M. (2021). Brain Tumor Segmentation Using a Tri-Planar Ensemble of U-Nets, Springer.10.1007/978-3-030-72084-1_31 ↗
13. Guoxiu, J. (2017). An introduction to convolutional neural networks. Heilongjiang Sci. Technol. Inf., 43–47.
14. 3D Convolutional Neural Networks for Tumor Segmentation using Long-range 2D Context10.1016/j.compmedimag.2019.02.001 ↗
15. DR-Unet104 for Multimodal MRI brain tumor segmentation10.1007/978-3-030-72087-2_36 ↗
16. Multi-scale feature fusion and additive attention guide brain tumor MR image segmentation10.11834/jig.211073 ↗
17. Design of tumor segmentation method based on 3D U-Net network
18. Brain Tumor Segmentation and Radiomics Survival Prediction: Contribution to the BRATS 2017 Challenge
19. Chen, W., Liu, B., Peng, S., Sun, J., and Qiao, X. (2019). Glioma, Multiple Sclerosis, Stroke and Traumatic Brain Injuries, Proceedings of the 4th International Workshop, BrainLes 2018, Granada, Spain, 16 September 2018, Springer International Publishing.
20. Myroneko, A. (2019). 3D MRI Brain Tumor Segmentation Using Autoencoder Regularization, Springer International Publishing.10.1007/978-3-030-11726-9_28 ↗
21. Adaptive Batch Normalization for practical domain adaptation10.1016/j.patcog.2018.03.005 ↗
22. Optimal approximation of piecewise smooth functions using deep ReLU neural networks10.1016/j.neunet.2018.08.019 ↗
23. Comparison of Jaccard, Dice, Cosine Similarity Coefficient To Find Best Fitness Value for Web Retrieved Documents Using Genetic Algorithm
24. Arora, A., Jayal, A., Gupta, M., Mittal, P., and Satapathy, S.C. (2021). Brain Tumor Segmentation of MRI Images Using Processed Image Driven U-Net Architectur. Computers, 10.10.3390/computers10110139 ↗
25. Advancing The Cancer Genome Atlas glioma MRI collections with expert segmentation labels and radiomic features10.1038/sdata.2017.117 ↗
26. Brain Tumor Segmentation Using Multi-Cascaded Convolutional Neural Networks and Conditional Random Field10.1109/ACCESS.2019.2927433 ↗
27. Saeed, M.U., Ali, G., Bin, W., Almotiri, S.H., AlGhamdi, M.A., Nagra, A.A., Masood, K., and Amin, R.U. (2021). RMU-Net: A Novel Residual Mobile U-Net Model for Brain Tumor Segmentation from MR Images. Electronics, 10.10.3390/electronics10161962 ↗
28. Segmentation Method of Brain Tumor MR Image Based on Improved U—Net Model
29. Fu, J., Liu, J., and Tian, H. (2020, January 15–20). Dual Attention Network for Scene Segmentation. Proceedings of the 2019 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), Long Beach, CA, USA.10.1109/CVPR.2019.00326 ↗
30. Recurrent residual U-Net for medical image segmentation10.1117/1.JMI.6.1.014006 ↗
31. Attention gated networks: Learning to leverage salient regions in medical images10.1016/j.media.2019.01.012 ↗
32. Cai, Y., and Wang, Y. (2020). MA-Unet: An improved version of Unet based on multi-scale and attention mechanism for medical image segmentation. arXiv.
33. SAU-Net: Efficient 3D Spine MRI Segmentation Using Inter-Slice Attention
34. ANU-Net: Attention-based Nested U-Net to exploit full resolution features for medical image segmentation10.1016/j.cag.2020.05.003 ↗
35. Oktay, O., Schlemper, J., and Folgoc, L.L. (2018). Attention U-Net: Learning Where to Look for the Pancreas. arXiv.