Advancing Brain Tumor Segmentation via Attention-based 3D U-Net Architecture and Digital Image Processing

TL;DR

This paper introduces an attention-enhanced 3D U-Net architecture combined with digital image processing to improve brain tumor segmentation accuracy on MRI scans, addressing challenges like irregular tumor shapes and data imbalance.

Contribution

The study proposes integrating attention mechanisms into 3D U-Net and using digital image processing for tumor detection to enhance segmentation performance and robustness.

Findings

Achieved a dice score of 0.975 on BraTS 2020 dataset

Outperformed existing models in specificity and sensitivity

Demonstrated improved delineation of irregular tumor boundaries

Abstract

In the realm of medical diagnostics, rapid advancements in Artificial Intelligence (AI) have significantly yielded remarkable improvements in brain tumor segmentation. Encoder-Decoder architectures, such as U-Net, have played a transformative role by effectively extracting meaningful representations in 3D brain tumor segmentation from Magnetic resonance imaging (MRI) scans. However, standard U-Net models encounter challenges in accurately delineating tumor regions, especially when dealing with irregular shapes and ambiguous boundaries. Additionally, training robust segmentation models on high-resolution MRI data, such as the BraTS datasets, necessitates high computational resources and often faces challenges associated with class imbalance. This study proposes the integration of the attention mechanism into the 3D U-Net model, enabling the model to capture intricate details and prioritize informative regions during the segmentation process. Additionally, a tumor detection algorithm based on digital image processing techniques is utilized to address the issue of imbalanced training data and mitigate bias. This study aims to enhance the performance of brain tumor segmentation, ultimately improving the reliability of diagnosis. The proposed model is thoroughly evaluated and assessed on the BraTS 2020 dataset using various performance metrics to accomplish this goal. The obtained results indicate that the model outperformed related studies, exhibiting dice of 0.975, specificity of 0.988, and sensitivity of 0.995, indicating the efficacy of the proposed model in improving brain tumor segmentation, offering valuable insights for reliable diagnosis in clinical settings.