Hybrid Adaptive Attention U-net for Breast lesion segmentation
AAU-net: An Adaptive Attention U-net for Breast Lesions Segmentation in Ultrasound Images
Considering the complexity of ultrasound images, we develop an adaptiveattention u-net (aau-net) to segment breast lesions automatically and stably from ultrasound images.Specifically, we introduce a hybrid adaptive attention module, which mainly consists of a channel self-attention block and a spatial self-attention block, to replace the traditional convolution operation.The design of the hybrid adaptive attention module can guide the network to adaptively select more robust representation in channel and space dimensions to cope with more complexbreast lesions segmentation.Extensive experiments with several state-of-the-art deep learning segmentation methods on three public breastultrasound datasets show that our method has better performance on breast lesions segmentation.Furthermore, robustness analysis and external experiments demonstrate that our proposed aau-net has better generalization performance on the segmentation of breast lesions.
