Abstract:Purpose: The purpose of this study is to develop accurate and automated detection and segmentation methods for brain tumors, given their significant fatality rates, with aggressive malignant tumors like Glioblastoma Multiforme (GBM) having a five-year survival rate as low as 5 to 10%. This underscores the urgent need to improve diagnosis and treatment outcomes through innovative approaches in medical imaging and deep learning techniques. Methods: In this work, we propose a novel approach utilizing the two-headed UNetEfficientNets model for simultaneous segmentation and classification of brain tumors from Magnetic Resonance Imaging (MRI) images. The model combines the strengths of EfficientNets and a modified two-headed Unet model. We utilized a publicly available dataset consisting of 3064 brain MR images classified into three tumor classes: Meningioma, Glioma, and Pituitary. To enhance the training process, we performed 12 types of data augmentation on the training dataset. We evaluated the methodology using six deep learning models, ranging from UNetEfficientNet-B0 to UNetEfficientNet-B5, optimizing the segmentation and classification heads using binary cross entropy (BCE) loss with Dice and BCE with focal loss, respectively. Post-processing techniques such as connected component labeling (CCL) and ensemble models were applied to improve segmentation outcomes. Results: The proposed UNetEfficientNet-B4 model achieved outstanding results, with an accuracy of 99.4% after postprocessing. Additionally, it obtained high scores for DICE (94.03%), precision (98.67%), and recall (99.00%) after post-processing. The ensemble technique further improved segmentation performance, with a global DICE score of 95.70% and Jaccard index of 91.20%. Conclusion: Our study demonstrates the high efficiency and accuracy of the proposed UNetEfficientNet-B4 model in the automatic and parallel detection and segmentation of brain tumors from MRI images. This approach holds promise for improving diagnosis and treatment planning for patients with brain tumors, potentially leading to better outcomes and prognosis.

An UNet-Based Brain Tumor Segmentation Framework Via Optimal Mass Transportation Pre-processing

A data pre-processing with mass-preserving optimal mass transportation for brain tumor segmentation

A Two-Phase Optimal Mass Transportation Technique for 3D Brain Tumor Detection and Segmentation

A novel 2-phase residual U-net algorithm combined with optimal mass transportation for 3D brain tumor detection and segmentation

3D brain tumor segmentation using a two-stage optimal mass transport algorithm

MRI Image Segmentation of Brain Tumor Based on Res-UNet

Research on the magnetic resonance imaging brain tumor segmentation algorithm based on DO‐UNet

MM-UNet: A multimodality brain tumor segmentation network in MRI images

A Deep Multi-Task Learning Framework for Brain Tumor Segmentation

An Optimization Framework for Processing and Transfer Learning for the Brain Tumor Segmentation

ResUNet + + : a comprehensive improved UNet + + framework for volumetric semantic segmentation of brain tumor MR images

Multi-Scale Masked 3-D U-Net For Brain Tumor Segmentation

Multimodal Brain Tumor Segmentation Based on an Intelligent UNET-LSTM Algorithm in Smart Hospitals

Brain Tumor Segmentation Using an Ensemble of 3D U-Nets and Overall Survival Prediction Using Radiomic Features

Two-headed UNetEfficientNets for parallel execution of segmentation and classification of brain tumors: incorporating postprocessing techniques with connected component labelling

Model Ensemble for Brain Tumor Segmentation in Magnetic Resonance Imaging

Multimodal brain tumor image segmentation based on DenseNet

A Deep Learning Framework Integrating MRI Image Preprocessing Methods for Brain Tumor Segmentation and Classification

Multimodal MRI brain tumor segmentation using 3D attention UNet with dense encoder blocks and residual decoder blocks

Automatic Segmentation of MRI of Brain Tumor Using Deep Convolutional Network

Brain tumor magnetic resonance image segmentation by a multiscale contextual attention module combined with a deep residual UNet (MCA-ResUNet)