Enhancing skin lesion segmentation with a fusion of convolutional neural networks and transformer models

被引:2
|
作者
Xu, Zhijian [1 ]
Guo, Xingyue [2 ]
Wang, Juan [2 ]
机构
[1] China West Normal Univ, Sch Elect Informat Engn, 1 Shida Rd, Nanchong 637009, Sichuan, Peoples R China
[2] China West Normal Univ, Sch Comp Sci, 1 Shida Rd, Nanchong 637009, Sichuan, Peoples R China
基金
中国国家自然科学基金;
关键词
Deep learning; Convolutional neural networks; Transformer; Skin lesion segmentation; Feature fusion; Dual branch encoder;
D O I
10.1016/j.heliyon.2024.e31395
中图分类号
O [数理科学和化学]; P [天文学、地球科学]; Q [生物科学]; N [自然科学总论];
学科分类号
07 ; 0710 ; 09 ;
摘要
Accurate segmentation is crucial in diagnosing and analyzing skin lesions. However, automatic segmentation of skin lesions is extremely challenging because of their variable sizes, uneven color distributions, irregular shapes, hair occlusions, and blurred boundaries. Owing to the limited range of convolutional networks receptive fields, shallow convolution cannot extract the global features of images and thus has limited segmentation performance. Because medical image datasets are small in scale, the use of excessively deep networks could cause overfitting and increase computational complexity. Although transformer networks can focus on extracting global information, they cannot extract sufficient local information and accurately segment detailed lesion features. In this study, we designed a dual -branch encoder that combines a convolution neural network (CNN) and a transformer. The CNN branch of the encoder comprises four layers, which learn the local features of images through layer -wise downsampling. The transformer branch also comprises four layers, enabling the learning of global image information through attention mechanisms. The feature fusion module in the network integrates local features and global information, emphasizes important channel features through the channel attention mechanism, and filters irrelevant feature expressions. The information exchange between the decoder and encoder is finally achieved through skip connections to supplement the information lost during the sampling process, thereby enhancing segmentation accuracy. The data used in this paper are from four public datasets, including images of melanoma, basal cell tumor, fibroma, and benign nevus. Because of the limited size of the image data, we enhanced them using methods such as random horizontal flipping, random vertical flipping, random brightness enhancement, random contrast enhancement, and rotation. The segmentation accuracy is evaluated through intersection over union and duration, integrity, commitment, and effort indicators, reaching 87.7 % and 93.21 %, 82.05 % and 89.19 %, 86.81 % and 92.72 %, and 92.79 % and 96.21 %, respectively, on the ISIC 2016, ISIC 2017, ISIC 2018, and PH2 datasets, respectively (code: https://github.com/hyjane/CCT-Net).
引用
收藏
页数:10
相关论文
共 50 条
  • [1] Convolutional Neural Networks Applied for Skin Lesion Segmentation
    Araujo, Graziela Silva
    Camara-Chavez, Guillermo
    Oliveira, Roberta B.
    2021 XLVII LATIN AMERICAN COMPUTING CONFERENCE (CLEI 2021), 2021,
  • [2] Automating Vitiligo Skin Lesion Segmentation Using Convolutional Neural Networks
    Low, Makena
    Huang, Victor
    Raina, Priyanka
    2020 IEEE 17TH INTERNATIONAL SYMPOSIUM ON BIOMEDICAL IMAGING (ISBI 2020), 2020, : 1992 - 1995
  • [3] An Ensemble of Fully Convolutional Neural Networks for Automatic Skin Lesion Segmentation
    Kanca, Elif
    Ayas, Selen
    2022 MEDICAL TECHNOLOGIES CONGRESS (TIPTEKNO'22), 2022,
  • [4] Skin lesion segmentation method for dermoscopic images with convolutional neural networks and semantic segmentation
    Thanh, Dang N. H.
    Nguyen Hoang Hai
    Le Minh Hieu
    Tiwari, Prayag
    Prasath, V. B. Surya
    COMPUTER OPTICS, 2021, 45 (01) : 122 - 129
  • [5] Skin-lesion segmentation using boundary-aware segmentation network and classification based on a mixture of convolutional and transformer neural networks
    Amin, Javaria
    Azhar, Marium
    Arshad, Habiba
    Zafar, Amad
    Kim, Seong-Han
    FRONTIERS IN MEDICINE, 2025, 12
  • [6] Skin Lesion Segmentation with Improved Convolutional Neural Network
    Ozturk, Saban
    Ozkaya, Umut
    JOURNAL OF DIGITAL IMAGING, 2020, 33 (04) : 958 - 970
  • [7] Skin Lesion Segmentation with Improved Convolutional Neural Network
    Şaban Öztürk
    Umut Özkaya
    Journal of Digital Imaging, 2020, 33 : 958 - 970
  • [8] Skin Lesion Segmentation Using Deep Convolutional Networks
    Arora, Parul
    Sharma, Nikhil
    Bhatt, Prakhar
    Saxena, Abhishek
    EAI/Springer Innovations in Communication and Computing, 2021, : 111 - 122
  • [9] Pigmented skin lesion segmentation based on random forest and full convolutional neural networks
    Yang Tiejun
    Peng Shan
    Hu Ping
    Huang Lin
    OPTICS IN HEALTH CARE AND BIOMEDICAL OPTICS VIII, 2018, 10820
  • [10] LEVERAGING ADAPTIVE COLOR AUGMENTATION IN CONVOLUTIONAL NEURAL NETWORKS FOR DEEP SKIN LESION SEGMENTATION
    Saha, Anindo
    Prasad, Prem
    Thabit, Abdullah
    2020 IEEE 17TH INTERNATIONAL SYMPOSIUM ON BIOMEDICAL IMAGING (ISBI 2020), 2020, : 2014 - 2017