XAI-MRI: An Ensemble Dual-Modality Approach for 3D Brain Tumor Segmentation Using Magnetic Resonance Imaging

Ahmeed Suliman Farhan,Muhammad Khalid,Umar Manzoor
DOI: https://doi.org/10.1101/2024.11.19.624437
2024-11-21
Abstract:Brain tumor segmentation from Magnetic Resonance Images (MRI) presents significant challenges due to the complex nature of brain tumor tissues. This complexity makes distinguishing tumor tissues from healthy tissues difficult, mainly when radiologists perform manual segmentation. Reliable and accurate segmentation is crucial for effective tumor grading and treatment planning. In this paper, we proposed a novel ensemble dual-modality approach for 3D brain tumor segmentation using MRI. Initially, individual U-Net models are trained and evaluated on single MRI modalities (T1, T2, T1ce, and FLAIR) to establish each modality's performance. Subsequently, we trained U-net models using combinations of the best-performing modalities to exploit the complementary information and improve segmentation accuracy. The most effective dual-modality combinations are then integrated into an ensemble model, designed to capture the strengths of each modality combination. Finally, we suggested the ensemble dual-modality by combining the best-performing two pre-trained dual-modalities models to enhance segmentation performance. Experimental results demonstrate the ensemble dual-modality model significantly improved segmentation performance, achieving a Dice Coefficient of 97.73% and a Mean IoU of 60.08% on the BraTS2020 dataset. Our results demonstrate that this ensemble approach outperforms traditional single-modality and dual-modality models, providing a more robust and accurate method for brain tumor segmentation. This study underscores the potential of ensemble dual-modality models to enhance the precision and reliability of MRI-based brain tumor segmentation. Our code publicly available at: https://github.com/Ahmeed-Suliman-Farhan/Ensemble-Dual-Modality-Approach .
Biology
What problem does this paper attempt to address?