TESL-Net: A Transformer-Enhanced CNN for Accurate Skin Lesion Segmentation

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Iqbal, Shahzaib, Zeeshan, Muhammad, Mehmood, Mehwish, Khan, Tariq M., Razzak, Imran
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866910570105536512
author Iqbal, Shahzaib
Zeeshan, Muhammad
Mehmood, Mehwish
Khan, Tariq M.
Razzak, Imran
author_facet Iqbal, Shahzaib
Zeeshan, Muhammad
Mehmood, Mehwish
Khan, Tariq M.
Razzak, Imran
contents Early detection of skin cancer relies on precise segmentation of dermoscopic images of skin lesions. However, this task is challenging due to the irregular shape of the lesion, the lack of sharp borders, and the presence of artefacts such as marker colours and hair follicles. Recent methods for melanoma segmentation are U-Nets and fully connected networks (FCNs). As the depth of these neural network models increases, they can face issues like the vanishing gradient problem and parameter redundancy, potentially leading to a decrease in the Jaccard index of the segmentation model. In this study, we introduced a novel network named TESL-Net for the segmentation of skin lesions. The proposed TESL-Net involves a hybrid network that combines the local features of a CNN encoder-decoder architecture with long-range and temporal dependencies using bi-convolutional long-short-term memory (Bi-ConvLSTM) networks and a Swin transformer. This enables the model to account for the uncertainty of segmentation over time and capture contextual channel relationships in the data. We evaluated the efficacy of TESL-Net in three commonly used datasets (ISIC 2016, ISIC 2017, and ISIC 2018) for the segmentation of skin lesions. The proposed TESL-Net achieves state-of-the-art performance, as evidenced by a significantly elevated Jaccard index demonstrated by empirical results.
format Preprint
id arxiv_https___arxiv_org_abs_2408_09687
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle TESL-Net: A Transformer-Enhanced CNN for Accurate Skin Lesion Segmentation
Iqbal, Shahzaib
Zeeshan, Muhammad
Mehmood, Mehwish
Khan, Tariq M.
Razzak, Imran
Image and Video Processing
Computer Vision and Pattern Recognition
Early detection of skin cancer relies on precise segmentation of dermoscopic images of skin lesions. However, this task is challenging due to the irregular shape of the lesion, the lack of sharp borders, and the presence of artefacts such as marker colours and hair follicles. Recent methods for melanoma segmentation are U-Nets and fully connected networks (FCNs). As the depth of these neural network models increases, they can face issues like the vanishing gradient problem and parameter redundancy, potentially leading to a decrease in the Jaccard index of the segmentation model. In this study, we introduced a novel network named TESL-Net for the segmentation of skin lesions. The proposed TESL-Net involves a hybrid network that combines the local features of a CNN encoder-decoder architecture with long-range and temporal dependencies using bi-convolutional long-short-term memory (Bi-ConvLSTM) networks and a Swin transformer. This enables the model to account for the uncertainty of segmentation over time and capture contextual channel relationships in the data. We evaluated the efficacy of TESL-Net in three commonly used datasets (ISIC 2016, ISIC 2017, and ISIC 2018) for the segmentation of skin lesions. The proposed TESL-Net achieves state-of-the-art performance, as evidenced by a significantly elevated Jaccard index demonstrated by empirical results.
title TESL-Net: A Transformer-Enhanced CNN for Accurate Skin Lesion Segmentation
topic Image and Video Processing
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2408.09687