AttU-NET: attention U-net for brain tumor segmentation

Wang, Sihan, Li, Lei and Zhuang, Xiahai (2022) AttU-NET: attention U-net for brain tumor segmentation. Crimi, Alessandro and Bakas, Spyridon (eds.) In Brainlesion: Glioma, Multiple Sclerosis, Stroke and Traumatic Brain Injuries: 7th International Workshop, BrainLes 2021, Held in Conjunction with MICCAI 2021, Virtual Event, September 27, 2021, Revised Selected Papers, Part II. vol. 12963, Springer Cham. pp. 302-311 . (doi:10.1007/978-3-031-09002-8_27).

Abstract

Tumor delineation is critical for the precise diagnosis and treatment of glioma patients. Since manual segmentation is time-consuming and tedious, automatic segmentation is desired. With the advent of convolution neural network (CNN), tremendous CNN models have been proposed for medical image segmentation. However, the small size of kernel limits the shape of the receptive view, omitting the global information. To utilize the intrinsic features of brain anatomical structure, we propose a modified U-Net with an attention block (AttU-Net) to tract the complementary information from the whole image. The proposed attention block can be easily added to any segmentation backbones, which improved the Dice score by 5%. We evaluated our approach on the dataset of BraTS 2021 challenge and achieved promising performance on this dataset. The Dice scores of enhancing tumor, tumor core, and whole tumor segmentation are 0.793, 0.819, and 0.879, respectively.

This record has no associated files available for download.

Contributors

Author: Lei Li

Download statistics