Automatic skin lesion segmentation using attention residual U-Net with improved encoder-decoder architecture
Rajdeep Kaur,Sukhjeet Kaur
DOI: https://doi.org/10.1007/s11042-024-18895-5
IF: 2.577
2024-03-20
Multimedia Tools and Applications
Abstract:The automatic segmentation of skin lesions in dermoscopic images is a challenging task due to the presence of artifacts, small lesion sizes, and low contrast between lesions and non-lesion regions. Deep learning models such as U-Net have been used for accurate segmentation in general. Still, their success rate is limited in the case of dermoscopic images due to these challenges. In this paper, we propose a U-Net-based model for efficient and effective segmentation of skin lesions in dermoscopic images. The proposed model, called Attention Residual U-Net with a modified decoder (ARU-Net-MD), employs an encoder-decoder architecture with residual learning, attention gates, and a modified decoder with a combined loss function to achieve higher accuracy for the semantic segmentation of dermoscopic images. Residual learning allows for an efficient model with fewer parameters, while attention gates highlight important features, and the modified decoder with a combined loss function assists the learning process and enables the model to learn more semantic information during training. We evaluated our model on four publicly available datasets, PH 2 , ISIC 2016, ISIC 2017, and ISIC 2018, and observed an accuracy of 0.96, 0.97, 0.95, and 0.96, respectively, outperforming other state-of-the-art skin lesion segmentation models.
computer science, information systems, theory & methods,engineering, electrical & electronic, software engineering