Semantic segmentation of brain tumor images using attention-based residual light u-net model

Chakrapani
DOI: https://doi.org/10.1007/s11042-024-19224-6
IF: 2.577
2024-04-25
Multimedia Tools and Applications
Abstract:Accurate segmentation of brain tumor regions in MRI images is essential for monitoring tumor growth. In view of this, several automated brain tumor segmentation models are proposed. U-Net is one of the most popular models widely used for image segmentation. Based on the U-Net architecture, complex models are proposed for brain tumor segmentation. However, existing models mainly focus on increasing the depth of the model for better accuracy without much bothering the transmission of spatial information. Attention-based U-Net model can help to extract salient local-level features that can be passed to the decoder part of the network. Considering the above issues, we propose a variant of the U-Net model called the Attention-based Residual Light U-Net model. The proposed model is effective in the sense that it has comparatively less model complexity, a better attention mechanism, and a combined loss function that can handle class imbalance problems. Our experiments on one of the latest BraTS-2021 datasets show that the proposed method can achieve a mean Intersection over Union of 0.8874, 0.8863, 0.8905, and dice score of 0.9415, 0.9387, 0.9115 on the whole tumor, tumor core, and enhancing tumor, respectively.
computer science, information systems, theory & methods,engineering, electrical & electronic, software engineering
What problem does this paper attempt to address?