Enhancing Brain Tumor MRI Segmentation Accuracy and Efficiency with Optimized U-Net Architecture
DOI:
https://doi.org/10.56532/mjsat.v4i3.302Keywords:
Brain tumor, Image segmentation, U-Net architecture, Image processingAbstract
This study presents an enhanced approach to brain tumor segmentation using an optimized U-Net architecture, focusing on MRI scans. Our research proposes an automated solution that utilizes U-Net to accurately differentiate between tumorous and non-tumorous tissues, addressing the challenges of manual segmentation such as time consumption, accuracy, and inter-observer variability. Our approach to accurately segmenting brain tumors utilizes the BraTS 2019 dataset and involves preprocessing steps that normalize image data. We employ a modified U-Net model that stands out for its depth and integration of multi-inception modules. Our evaluation metrics, including an IoU score of 0.8252 and a low-test loss of approximately 7.075e-05, highlight the high precision of our model in segmenting brain tumors. However, limitations arise from dataset specificity and potential class imbalance, suggesting future work should focus on enhancing generalizability and addressing computational efficiency. Deep learning has been shown to have significant potential in enhancing diagnostic accuracy and treatment planning in neuro-oncology. This, in turn, opens new opportunities for further developments in automated medical image analysis.
References
Sun, J, Chen, W, Peng, S, & Liu, B. (2019). DRRNet: dense residual refine networks for automatic brain tumor segmentation. J Med Syst., 43(7), 221.
Kamnitsas, K, Ledig, C, Newcombe, V. F. J, & et al. (2017). Efficient multi-scale 3D CNN with fully connected CRF for accurate brain lesion segmentation. Medical image analysis, 36, 61-78.
Xu, M., Yin, Y., & Ye, J. (2015). A convolutional neural network for modulation classification of digital signals. In 2015 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP) (pp. 2016-2020). IEEE.
Nair, V., & Hinton, G. E. (2010). Rectified linear units improve restricted boltzmann machines. In Proceedings of the 27th international conference on machine learning (ICML-10) (pp. 807-814).
Krizhevsky, A., Sutskever, I., & Hinton, G. E. (2012). Imagenet classification with deep convolutional neural networks. In Advances in neural information processing systems (pp. 1097-1105).
Jony, A. I., & Arnob, A. K. B. (2024). A long short-term memory based approach for detecting cyber attacks in IoT using CIC-IoT2023 dataset. Journal of Edge Computing, https://doi.org/10.55056/jec.648.
Alam, M. M., & Jony, A. I. (2023). Supply Chain Management Techniques Using Big Data for Agro-Based Food Products in Bangladesh. International Journal of Data Science and Big Data Analytics, 3(2), 19-34.
Amin, J, Sharif, M, Yasmin, M, & Fernandes, S. L. (2018). Big data analysis for brain tumor detection: deep convolutional neural networks. Future Gen Comput Syst., 87, 290-297.
Hussain, S, Anwar, S. M., & Majid, M. (2018). Segmentation of glioma tumors in brain using deep convolutional neural network. Neurocomputing, 282, 248-261.
Chang, J, Ye, M., Zhang, X., Huang, D., Wang, P. & Ya, C. (2020). Brain Tumor Segmentation based on 3D Unet with Multi-class Focal Loss. IEEE Access, 8, 118195-118203.
Havaei, M, Davy, A., Warde-Farley, D., Biard, A., Courville, A., Bengio, Y., Pal, C., Jodoin, P. M., & Larochelle, H. (2017). Brain tumor segmentation with deep neural networks, medical image analysis, 35, 18–31.
Shawki, N., Elseify, T., Cap, T., Shah, V., Obeid, I., & Picone, J. (2020). A Deep Learning-Based Real-time Seizure Detection System, IEEE Journal of Translational Engineering in Health and Medicine, 8, 1-11.
Latif, U., Shahid, A. R., Raza, B., Ziauddin, S. & Khan, M. A. (2021). An end- to-end brain tumor segmentation system using multi-inception-UNET, Computer Methods and Programs in Biomedicine, 203, 105864.
Cahall, D. E, Rasool, G, Bouaynaya, N. C, & Fathallah-Shaykh, H. M. (2019). Inception modules enhance brain tumor segmentation. Front Comput Neurosci., 13, 44.
Jony, A. I. (2016). Applications of real-time big data analytics. International Journal of Computer Applications, 144(5), 1-5.
Szegedy, C, Liu, W, Jia, Y, & et al. (2015). Going deeper with convolutions. Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, Boston, MA, USA: IEEE; pp. 1-9.
Liu, D, Zhang, H, Zhao, M, Yu, X, Yao, S, Zhou, W. (2018). Brain tumor segmentation based on dilated convolution refine networks. 16th International Conference on Software Engineering Research, Management and Applications (SERA). Kunming, China: IEEE; 2018, June, pp. 113-120.
Li, H, Li, A, & Wang, M. A. (2019). novel end-to-end brain tumor segmentation method using improved fully convolutional networks. Compute Biol Med, 108, 150-160.
Morshed, M. S. J. & Jony, A. I. (2016). Real Time Social Network Data Analysis for Community Detection. International Journal of Computer Applications, 139(6), 1-5.
Latif, U., Shahid, A. R., Raza, B., Ziauddin, S., & Khan, M. A. (2021). An end‐to‐end brain tumor segmentation system using multi‐inception‐UNET. International Journal of Imaging Systems and Technology, 31(4), 1803-1816.
Lisun-Ul-Islam, M., Rahat, M. R. H., Esha, S., Faiyaz, A., & Jony, A. I. (2023). Hourly Air Quality Prediction in Dhaka City Using Time Series Forecasting Techniques: Deep Learning Perspectives. Tuijin Jishu/Journal of Propulsion Technology, 44(5), 568-579.
Myronenko, A. (2018). 3D MRI brain tumor segmentation using autoencoder regularization. International MICCAI Brainlesion Workshop. Cham, Switzerland: Springer; 2018, September: 311-320.
Chen L, Bentley P, Mori K, Misawa K, Fujiwara M, Rueckert D. (2018). Drinet for medical image segmentation. IEEE Trans Med Imaging, 37(11), 2453-2462
Ronneberger, O., Fischer, P., & Brox, T. (2015). U-Net: Convolutional networks for biomedical image segmentation. In International Conference on Medical Image Computing and Computer-Assisted Intervention (pp. 234-241). Springer.
Jony, A. I., & Arnob, A. K. B. (2024). Deep Learning Paradigms for Breast Cancer Diagnosis: A Comparative Study on Wisconsin Diagnostic Dataset. Malaysian Journal of Science and Advanced Technology, 4(2), 109-117.
Tanvir, K., Jony, A. I., Haq, M. K., Nazera, F., Dass, M., & Raju, V. (2023). Clinical Insights Through Xception: A Multiclass Classification of Ocular Pathologies. Tuijin Jishu/Journal of Propulsion Technology, 44(04), 5876-5885.
Shovon, M. S. H., Islam, M. J., Nabil, M. N. A. K., Molla, M. M., Jony, A. I., & Mridha, M. F. (2022). Strategies for enhancing the multi-stage classification performances of her2 breast cancer from hematoxylin and eosin images. Diagnostics, 12(11), 2825.
Aghalari, M., Aghagolzadeh, A., & Ezoji, M. (2021). Brain tumor image segmentation via asymmetric/symmetric UNet based on two-pathway-residual blocks. Biomedical signal processing and control, 69, 102841.
Latif, U., Shahid, A. R., Raza, B., Ziauddin, S., & Khan, M. A. (2021). An end‐to‐end brain tumor segmentation system using multi‐inception‐UNET. International Journal of Imaging Systems and Technology, 31(4), 1803-1816.
Aledhari, M., & Razzak, R. (2020, December). An adaptive segmentation technique to detect brain tumors using 2D Unet. In 2020 IEEE International Conference on bioinformatics and biomedicine (BIBM) (pp. 2328-2334). IEEE.
Maji, D., Sigedar, P., & Singh, M. (2022). Attention Res-UNet with Guided Decoder for semantic segmentation of brain tumors. Biomedical Signal Processing and Control, 71, 103077.
Agrawal, P., Katal, N., & Hooda, N. (2022). Segmentation and classification of brain tumor using 3D-UNet deep neural networks. International Journal of Cognitive Computing in Engineering, 3, 199-210.
Allah, A. M. G., Sarhan, A. M., & Elshennawy, N. M. (2023). Edge U-Net: Brain tumor segmentation using MRI based on deep U-Net model with boundary information. Expert Systems with Applications, 213, 118833.
Bindu, N. P., & Sastry, P. N. (2023). Automated brain tumor detection and segmentation using modified UNet and ResNet model. Soft Computing, 27(13), 9179-9189.
Akbar, A. S., Fatichah, C., Suciati, N., & Za’in, C. (2024). Yaru3DFPN: a lightweight modified 3D UNet with feature pyramid network and combine thresholding for brain tumor segmentation. Neural Computing and Applications, 1-16.
Downloads
Published
How to Cite
Issue
Section
License
Copyright (c) 2024 Sultanul Arifeen Hamim, Akinul Islam Jony
This work is licensed under a Creative Commons Attribution-NonCommercial 4.0 International License.
