Abstract:Accurate segmentation of skin lesions is a challenging task because the task is highly influenced by factors such as location, shape and scale. In recent years, Convolutional Neural Networks (CNNs) have achieved advanced performance in automated medical image segmentation. However, existing CNNs have problems such as inability to highlight relevant features and preserve local features, which limit their application in clinical decision-making. This paper proposes a CNN with an added attention mechanism (EA-Net) for more accurate medical image segmentation.EA-Net is based on the U-Net network model framework. Specifically, we added a pixel-level attention module (PA) to the encoder section to preserve the local features of the image during downsampling, making the feature maps input to the decoder more relevant to the ground-truth. At the same time, we added a spatial multi-scale attention module (SA) after the decoding process to increase the spatial weight of the feature maps that are more relevant to the ground-truth, thereby reducing the gap between the output results and the ground-truth. We conducted extensive segmentation experiments on skin lesion images from the ISIC 2017 and ISIC 2018 datasets. The results demonstrate that, when compared to U-Net, our proposed EA-Net achieves an average Dice score improvement of 1.94% and 5.38% for skin lesion tissue segmentation on the ISIC 2017 and ISIC 2018 datasets, respectively. The IoU also increases by 2.69% and 8.31%, and the ASSD decreases by 0.3783 pix and 0.5432 pix, indicating superior segmentation performance. EA-Net can achieve better segmentation results when the original image of skin lesions has an obscure boundary and the segmentation area contains interference factors, which proves that the addition of attention mechanism in the encoder and the application of comprehensive attention mechanism can improve the performance of neural network in the field of skin lesions image segmentation.

Dense and Shuffle Attention U-Net for Automatic Skin Lesion Segmentation.

Attention-Guided Network with Densely Connected Convolution for Skin Lesion Segmentation

Skin lesion segmentation with attention-based SC-Conv U-Net and feature map distortion

SEACU-Net: Attentive ConvLSTM U-Net with squeeze-and-excitation layer for skin lesion segmentation

A Robust Segmentation Method Based on Improved U-Net

Channel Attention Separable Convolution Network for Skin Lesion Segmentation

[Image segmentation of skin lesions based on dense atrous spatial pyramid pooling and attention mechanism]

Automated Segmentation of Skin Lesion Based on Pyramid Attention Network.

GAU-Net: U-Net Variant for Segmentation of Skin Lesion

Improved U-Net Based on Contour Attention for Efficient Segmentation of Skin Lesion

DI-UNet: dual-branch interactive U-Net for skin cancer image segmentation

GSCEU-Net: an End-to-End Lightweight Skin Lesion Segmentation Model with Feature Fusion Based on U-Net Enhancements

EA-Net: Research on Skin Lesion Segmentation Method Based on U-Net.

Channel Context and Dual-Domain Attention Based U-Net for Skin Lesion Attributes Segmentation

AS-Net: Attention Synergy Network for skin lesion segmentation

ELA-Net: An Efficient Lightweight Attention Network for Skin Lesion Segmentation

MALUNet: A Multi-Attention and Light-weight UNet for Skin Lesion Segmentation

LSSF-Net: Lightweight Segmentation with Self-Awareness, Spatial Attention, and Focal Modulation

EIU-Net: Enhanced feature extraction and improved skip connections in U-Net for skin lesion segmentation

USL-Net: Uncertainty Self-Learning Network for Unsupervised Skin Lesion Segmentation

Anti-Aliasing Attention U-net Model for Skin Lesion Segmentation