Brain Tumor Classification and Segmentation in MR Images Using EfficientNet and U-Net++ Models.
Authors
Affiliations (1)
Affiliations (1)
- Department of Information Technology, College of Computer and Information Sciences, King Saud University, P.O. Box 51178, Riyadh 11543, Saudi Arabia.
Abstract
<b>Background/Objectives</b><b>:</b> Brain tumor analysis using magnetic resonance imaging (MRI) remains a challenging task due to tumor heterogeneity, complex anatomical structures, and reliance on expert interpretation. Although deep learning approaches have shown promising results in medical image analysis, many existing studies focus on either tumor classification or segmentation independently, limiting their applicability in comprehensive automated brain tumor analysis workflows. This study proposes an integrated dual-task deep learning framework for automated brain tumor classification and segmentation using MRI scans. The framework aims to provide complementary diagnostic support by combining tumor-type prediction and tumor boundary delineation within an integrated workflow. <b>Methods:</b> The proposed framework utilizes EfficientNet-based convolutional neural networks for multi-class brain tumor classification and U-Net++ architectures with EfficientNet encoders for tumor segmentation. Experiments were conducted using the BRISC2025 dataset, consisting primarily of 6000 T1-weighted 2D MRI slices collected from axial, coronal, and sagittal planes. Standard preprocessing, augmentation, transfer learning, and selective fine-tuning strategies were applied. Multiple architectures were systematically evaluated using evaluation metrics. <b>Results:</b> EfficientNet-B1 achieved a classification accuracy of 99.70% with near-perfect precision, recall, and F1-scores across glioma, meningioma, pituitary tumor, and no-tumor classes. For segmentation, U-Net++ with an EfficientNet-B1 encoder achieved a Dice score of 0.9055, an IoU score of 0.8442, and an HD95 value of 12.21 pixels on the held-out test set. The proposed framework demonstrated robust performance in detecting small and low-contrast tumor regions while maintaining strong generalization performance across diverse MRI samples. <b>Conclusions:</b> The proposed integrated framework demonstrated strong performance in both brain tumor classification and segmentation tasks, effectively detecting small and low-contrast tumor regions while maintaining good generalization across diverse MRI samples. These findings suggest that the framework may serve as a reliable decision-support tool for automated brain tumor analysis in clinical practice.