Journal
MATHEMATICAL BIOSCIENCES AND ENGINEERING
Volume 20, Issue 12, Pages 20510-20527Publisher
AMER INST MATHEMATICAL SCIENCES-AIMS
DOI: 10.3934/mbe.2023907
Keywords
brain tumor segmentation; transformer; attention mechanism; U-Net
Categories
Ask authors/readers for more resources
This paper proposes a novel mixed attention U-Net model, called MAU-Net, for MRI brain tumor segmentation. The model combines spatial-channel attention and self-attention, achieving significant improvements in local and global information modeling.
Computer-aided brain tumor segmentation using magnetic resonance imaging (MRI) is of great significance for the clinical diagnosis and treatment of patients. Recently, U-Net has received widespread attention as a milestone in automatic brain tumor segmentation. Following its merits and motivated by the success of the attention mechanism, this work proposed a novel mixed attention U-Net model, i.e., MAU-Net, which integrated the spatial-channel attention and self-attention into a single U-Net architecture for MRI brain tumor segmentation. Specifically, MAU-Net embeds Shuffle Attention using spatial-channel attention after each convolutional block in the encoder stage to enhance local details of brain tumor images. Meanwhile, considering the superior capability of self-attention in modeling long-distance dependencies, an enhanced Transformer module is introduced at the bottleneck to improve the interactive learning ability of global information of brain tumor images. MAU-Net achieves enhancing tumor, whole tumor and tumor core segmentation Dice values of 77.88/77.47, 90.15/90.00 and 81.09/81.63% on the brain tumor segmentation (BraTS) 2019/2020 validation datasets, and it outperforms the baseline by 1.15 and 0.93% on average, respectively. Besides, MAU-Net also demonstrates good competitiveness compared with representative methods.
Authors
I am an author on this paper
Click your name to claim this paper and add it to your profile.
Reviews
Recommended
No Data Available