Attentional mechanisms for multiscale aggregate U-Net brain tumor image segmentation

Haoxin Yuan

doi:10.1117/12.3029144

6 May 2024 Attentional mechanisms for multiscale aggregate U-Net brain tumor image segmentation

Haoxin Yuan

Proceedings Volume 13107, Fourth International Conference on Sensors and Information Technology (ICSI 2024); 131070B (2024) https://doi.org/10.1117/12.3029144
Event: Fourth International Conference on Sensors and Information Technology (ICSI 2024), 2024, Xiamen, China

Abstract

In response to the need for brain tumor image segmentation, which involves segmenting it into three different regions based on the degree of tumor lesions, this paper addresses several issues with the U-Net model, such as limited feature extraction accuracy and unclear region segmentation. To tackle these challenges, we propose a multi-scale aggregation U-Net segmentation algorithm that incorporates an attention mechanism. The proposed algorithm follows a two-step process: first, it conducts feature extraction and up-sampling after each layer's down-sampling in the U-Net architecture, and then iteratively aggregates the feature map with the up-sampled map. Second, we introduce residual blocks in the encoding stage to address the issue of gradient vanishing during downsampling. Finally, we incorporate channel attention mechanisms across layers in the decoding stage to direct the network's focus on the relevant regions, thus improving the segmentation accuracy of lesion regions. We conducted simulation experiments using the BraTs2021 training set, achieving Dice coefficients of 0.927, 0.861, and 0.867 for the segmentation of the whole tumor, core tumor, and enhanced tumor images, respectively. These results surpass those of most existing brain tumor segmentation models.

(2024) Published by SPIE. Downloading of the abstract is permitted for personal use only.

Citation Download Citation

Haoxin Yuan "Attentional mechanisms for multiscale aggregate U-Net brain tumor image segmentation", Proc. SPIE 13107, Fourth International Conference on Sensors and Information Technology (ICSI 2024), 131070B (6 May 2024); https://doi.org/10.1117/12.3029144

ACCESS THE FULL ARTICLE

INSTITUTIONAL
Select your institution to access the SPIE Digital Library.

SELECT YOUR INSTITUTION

PERSONAL
Sign in with your SPIE account to access your personal subscriptions or to use specific features such as save to my library, sign up for alerts, save searches, etc.

PERSONAL SIGN IN

No SPIE Account? Create one

PURCHASE THIS CONTENT

SUBSCRIBE TO DIGITAL LIBRARY

50 downloads per 1-year subscription

Members: $195

Non-members: $335 ADD TO CART

25 downloads per 1 - year subscription

Members: $145

Non-members: $250 ADD TO CART

PURCHASE SINGLE ARTICLE

Includes PDF, HTML & Video, when available