3D S-UNET an Efficient Architecture for 3 Dimensional Segmentation of Brain Tumors on MRI Images
DOI:
https://doi.org/10.33005/itij.v3i2.53Keywords:
Brain Tumors, Deep Learning, Lightweight Deep Learning, Brain Tumor Segmentation 3DAbstract
One of the deadliest diseases worldwide is brain tumors. In identifying brain tumors, experts perform a subjective analysis that requires considerable time. Previous research has developed automatic 3D brain tumor segmentation using Deep Learning (DL) approaches such as 3D UNet and 3D ResNet. However, these approaches demand significant computational resources. In resource-constrained settings, key criteria for determining the best architecture include memory consumption, inference speed, and accuracy. Therefore, this study introduces the development of the 3D S-UNet architecture, constructed by combining 3D ShuffleNet-V2 as an encoder and 3D UNet as a decoder. The integration of these 3D data processors allows the architecture to be more precise in identifying brain tumor locations and capture richer feature values compared to 2D data processing. The researchers compare 3D S-UNet with another Lightweight Deep Learning architecture, 3D Mobile-UNet. The results show that 3D S-UNet has a smaller memory consumption, using 0.56GB for the highest allocated memory and 1.71GB for reserved memory. In terms of inference speed, 3D S-UNet is faster compared to the other three architectures, achieving a speed of 135.881 milliseconds. 3D S-UNet demonstrates favorable results with a Whole Tumor (WT) dice score, sensitivity, and specificity of 83%, 85%, and 88%, respectively.
Downloads
References
[1] S. Kedia et al., “Brain Tumor Programs in Asia and Africa: Current Status, Challenges, and Future Perspectives,” World Neurosurg, vol. 175, pp. e1041–e1048, 2023, doi: https://doi.org/10.1016/j.wneu.2023.04.067.
[2] U. Raghavendra et al., “Brain tumor detection and screening using artificial intelligence techniques: Current trends and future perspectives,” Comput Biol Med, vol. 163, p. 107063, 2023, doi: https://doi.org/10.1016/j.compbiomed.2023.107063.
[3] G. Tabatabai et al., “Molecular diagnostics of gliomas: the clinical perspective.,” Acta Neuropathol, vol. 120, no. 5, pp. 585–592, Nov. 2010, doi: 10.1007/s00401-010-0750-6.
[4] Y. T. Udaka and R. J. Packer, “Pediatric Brain Tumors,” Neurol Clin, vol. 36, no. 3, pp. 533–556, 2018, doi: https://doi.org/10.1016/j.ncl.2018.04.009.
[5] M. Aljabri and M. AlGhamdi, “A review on the use of deep learning for medical images segmentation,” Neurocomputing, vol. 506, pp. 311–335, 2022, doi: https://doi.org/10.1016/j.neucom.2022.07.070.
[6] G. Litjens et al., “A survey on deep learning in medical image analysis,” Med Image Anal, vol. 42, pp. 60–88, 2017, doi: https://doi.org/10.1016/j.media.2017.07.005.
[7] M. H. Hesamian, W. Jia, X. He, and P. Kennedy, “Deep Learning Techniques for Medical Image Segmentation: Achievements and Challenges,” J Digit Imaging, vol. 32, no. 4, pp. 582–596, 2019, doi: 10.1007/s10278-019-00227-x.
[8] K. Kamnitsas et al., “Efficient multi-scale 3D CNN with fully connected CRF for accurate brain lesion segmentation,” Med Image Anal, vol. 36, pp. 61–78, 2017, doi: https://doi.org/10.1016/j.media.2016.10.004.
A. Beers et al., “Sequential 3d u-nets for biologically-informed brain tumor segmentation,” arXiv preprint arXiv:1709.02967, 2017.
[9] L. Dai, T. Li, H. Shu, L. Zhong, H. Shen, and H. Zhu, “Automatic Brain Tumor Segmentation with Domain Adaptation,” in BrainLes@MICCAI, 2018. [Online]. Available: https://api.semanticscholar.org/CorpusID:59943040
A. Myronenko, “3D MRI brain tumor segmentation using autoencoder regularization,” in BrainLes@MICCAI, 2018.
[10] H. Xu, H. Xie, Y. Liu, C. Cheng, C. Niu, and Y. Zhang, “Deep Cascaded Attention Network for Multi-task Brain Tumor Segmentation,” in International Conference on Medical Image Computing and Computer-Assisted Intervention, 2019. [Online]. Available: https://api.semanticscholar.org/CorpusID:204028162
[11] M. Aly and A. S. Alotaibi, “EMU-Net: Automatic Brain Tumor Segmentation and Classification Using Efficient Modified U-Net,” Computers, Materials and Continua, vol. 77, no. 1, pp. 557–582, 2023, doi: https://doi.org/10.32604/cmc.2023.042493.
[12] W. Zafar et al., “Enhanced TumorNet: Leveraging YOLOv8s and U-net for superior brain tumor detection and segmentation utilizing MRI scans,” Results in Engineering, vol. 24, p. 102994, 2024, doi: https://doi.org/10.1016/j.rineng.2024.102994.
[13] C.-H. Wang, K.-Y. Huang, Y. Yao, J.-C. Chen, H.-H. Shuai, and W.-H. Cheng, “Lightweight Deep Learning: An Overview,” IEEE Consumer Electronics Magazine, pp. 1–12, 2022, doi: 10.1109/MCE.2022.3181759.
[14] S. Chattopadhyay, A. Dey, P. K. Singh, and R. Sarkar, “DRDA-Net: Dense residual dual-shuffle attention network for breast cancer classification using histopathological images,” Comput Biol Med, vol. 145, p. 105437, 2022, doi: https://doi.org/10.1016/j.compbiomed.2022.105437.
[15] D. Das, D. R. Nayak, S. V Bhandary, and U. R. Acharya, “CDAM-Net: Channel shuffle dual attention based multi-scale CNN for efficient glaucoma detection using fundus images,” Eng Appl Artif Intell, vol. 133, p. 108454, 2024, doi: https://doi.org/10.1016/j.engappai.2024.108454.
[16] N. Kim, S. Lee, J. Kim, S. Y. Choi, and S.-M. Park, “Shuffled ECA-Net for stress detection from multimodal wearable sensor data,” Comput Biol Med, vol. 183, p. 109217, 2024, doi: https://doi.org/10.1016/j.compbiomed.2024.109217.
[17] S. Sangui, T. Iqbal, P. C. Chandra, S. K. Ghosh, and A. Ghosh, “3D MRI Segmentation using U-Net Architecture for the detection of Brain Tumor,” Procedia Comput Sci, vol. 218, pp. 542–553, 2023, doi: https://doi.org/10.1016/j.procs.2023.01.036.
[18] M. Pendse, V. Thangarasa, V. Chiley, R. Holmdahl, J. Hestness, and D. DeCoste, “Memory efficient 3d u-net with reversible mobile inverted bottlenecks for brain tumor segmentation,” in International MICCAI Brainlesion Workshop, Springer, 2020, pp. 388–397.
[19] S. Bakas et al., “Identifying the best machine learning algorithms for brain tumor segmentation, progression assessment, and overall survival prediction in the BRATS challenge,” arXiv preprint arXiv:1811.02629, 2018.
[20] S. Bakas et al., “Advancing The Cancer Genome Atlas glioma MRI collections with expert segmentation labels and radiomic features,” Sci Data, vol. 4, no. 1, p. 170117, 2017, doi: 10.1038/sdata.2017.117
Downloads
Published
How to Cite
Issue
Section
License
Copyright (c) 2025 M Sadewa Wicaksana Wibowo, Muhammad Shodiq, Bety Qorry Aina, Angga Lisdiyanto
This work is licensed under a Creative Commons Attribution 4.0 International License.
Starting with Volume 3, Issue 1 (2025), all articles published in ITIJ will be licensed under a Creative Commons Attribution (CC BY) license. This new license replaces our previous Creative Commons Attribution-NonCommercial-NoDerivatives (CC BY-NC-ND) license.
