Detection of brain tumor using Hybridized 3D U-Net model on MRI images
Shreeharsha, J.
DOI: https://doi.org/10.1007/s11042-024-20398-2
IF: 2.577
2024-10-23
Multimedia Tools and Applications
Abstract:The automatic detection of brain tumor is an emerging challenge because tumors vary in mass, nature, position, and similarities between the normal and brain lesions. First, the Magnetic Resonance Imaging (MRI) images are acquired from Brain Tumor Segmentation (BraTS) 2018, 2019 and 2020 databases; here, BraTS2018, BraTS2019 and BraTS2020 databases are considered as DB1, DB2 and DB3. Next, the image denoising is carried out by employing a normalization technique to enhance contrast and intensity levels of the acquired MRI images. Then, a hybridized 3D U-Net model is introduced for Brain Tumor Detection (BTD). In the hybridized 3D U-Net model, a ResNet-50 model is incorporated as a feature extractor in the conventional 3D U-Net model. In this particular context, the ResNet-50 model extracts intricate textures and patterns from the denoised 3D brain images, and assists the 3D U-Net model in accurate segmentation and classification of four modalities of brain tumors: T1, T2, Contrast-Enhanced (CE) -T1, and flair. In the segmentation phase, the hybridized 3D U-Net model obtains maximum segmentation results with a dice coefficient of 96.82%, 98.06% and 98.80%, and Jaccard coefficient of 96.64%, 97.25%, and 98.60% on the DB1, DB2, and DB2 databases. Additionally, in the classification phase, the hybridized 3D U-Net model achieves significant classification results with accuracies of 99.32%, 99.40%, and 99.80% on the DB1, DB2, and DB3 databases, correspondingly. The implementation of this research has been uploaded to the GitHub repository at https://github.com/shreeharsha-revanth/Hybrid-3D-UNet/.
computer science, information systems, theory & methods,engineering, electrical & electronic, software engineering