Optimizing Brain Tumor Segmentation with MedNeXt: BraTS 2024 SSA and Pediatrics

Sarim Hashmi,Juan Lugo,Abdelrahman Elsayed,Dinesh Saggurthi,Mohammed Elseiagy,Alikhan Nurkamal,Jaskaran Walia,Fadillah Adamsyah Maani,Mohammad Yaqub

2024-11-27

Abstract:Identifying key pathological features in brain MRIs is crucial for the long-term survival of glioma patients. However, manual segmentation is time-consuming, requiring expert intervention and is susceptible to human error. Therefore, significant research has been devoted to developing machine learning methods that can accurately segment tumors in 3D multimodal brain MRI scans. Despite their progress, state-of-the-art models are often limited by the data they are trained on, raising concerns about their reliability when applied to diverse populations that may introduce distribution shifts. Such shifts can stem from lower quality MRI technology (e.g., in sub-Saharan Africa) or variations in patient demographics (e.g., children). The BraTS-2024 challenge provides a platform to address these issues. This study presents our methodology for segmenting tumors in the BraTS-2024 SSA and Pediatric Tumors tasks using MedNeXt, comprehensive model ensembling, and thorough postprocessing. Our approach demonstrated strong performance on the unseen validation set, achieving an average Dice Similarity Coefficient (DSC) of 0.896 on the BraTS-2024 SSA dataset and an average DSC of 0.830 on the BraTS Pediatric Tumor dataset. Additionally, our method achieved an average Hausdorff Distance (HD95) of 14.682 on the BraTS-2024 SSA dataset and an average HD95 of 37.508 on the BraTS Pediatric dataset. Our GitHub repository can be accessed here: Project Repository : <a class="link-external link-https" href="https://github.com/python-arch/BioMbz-Optimizing-Brain-Tumor-Segmentation-with-MedNeXt-BraTS-2024-SSA-and-Pediatrics" rel="external noopener nofollow">this https URL</a>

Image and Video Processing,Computer Vision and Pattern Recognition

What problem does this paper attempt to address?

The problem that this paper attempts to solve is the accuracy of brain tumor segmentation in different populations. Specifically, the paper focuses on improving the performance of automatic brain tumor segmentation in two specific tasks: BraTS 2024 SSA (Sub - Saharan Africa) and pediatric brain tumor segmentation tasks. These problems include:

1. **Limitations of Manual Segmentation**: Manual segmentation of brain tumors is time - consuming and requires expert intervention, and is prone to human errors. This is especially prominent in countries with limited resources, where the medical systems are overburdened and there is a shortage of qualified professionals.

2. **Impact of Data Distribution Differences**: Existing state - of - the - art models are usually limited by the sources of training data. When applied to different groups (such as areas using lower - quality MRI techniques or pediatric patients), distribution shift may occur, affecting the reliability of the models.

3. **Tumor Segmentation Challenges in Specific Populations**:
- **Sub - Saharan Africa**: MRI techniques in this region may be inferior, resulting in poor image quality, which in turn affects the accuracy of tumor segmentation.
- **Pediatric Patients**: The brain anatomy of children is different from that of adults, and specialized models are required to handle these differences.

To address these challenges, the paper proposes a deep - learning method based on the MedNeXt architecture and improves the segmentation performance through comprehensive model integration and thorough post - processing techniques. Experimental results show that this method performs well on unseen validation sets, achieving high Dice Similarity Coefficient (DSC) and Hausdorff Distance (HD95).