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 条
  • [31] Optimized Convolutional Neural Network Models for Skin Lesion Classification
    Villa-Pulgarin, Juan Pablo
    Ruales-Torres, Anderson Alberto
    Arias-Garzon, Daniel
    Bravo-Ortiz, Mario Alejandro
    Arteaga-Arteaga, Harold Brayan
    Mora-Rubio, Alejandro
    Alzate-Grisales, Jesus Alejandro
    Mercado-Ruiz, Esteban
    Hassaballah, M.
    Orozco-Arias, Simon
    Cardona-Morales, Oscar
    Tabares-Soto, Reinel
    CMC-COMPUTERS MATERIALS & CONTINUA, 2022, 70 (02): : 2131 - 2148
  • [32] Star Shape Prior in Fully Convolutional Networks for Skin Lesion Segmentation
    Mirikharaji, Zahra
    Hamarneh, Ghassan
    MEDICAL IMAGE COMPUTING AND COMPUTER ASSISTED INTERVENTION - MICCAI 2018, PT IV, 2018, 11073 : 737 - 745
  • [33] PCF-Net: Position and context information fusion attention convolutional neural network for skin lesion segmentation
    Jiang, Yun
    Dong, Jinkun
    Zhang, Yuan
    Cheng, Tongtong
    Lin, Xin
    Liang, Jing
    HELIYON, 2023, 9 (03)
  • [34] Efficient UNet fusion of convolutional neural networks and state space models for medical image segmentation
    Meng, Wenjie
    Mu, Aiming
    Wang, Huajun
    DIGITAL SIGNAL PROCESSING, 2025, 158
  • [35] Multiple Sclerosis Lesion Segmentation using Improved Convolutional Neural Networks
    Kazancli, Erol
    Prchkovska, Vesna
    Rodrigues, Paulo
    Villoslada, Pablo
    Igual, Laura
    VISAPP: PROCEEDINGS OF THE 13TH INTERNATIONAL JOINT CONFERENCE ON COMPUTER VISION, IMAGING AND COMPUTER GRAPHICS THEORY AND APPLICATIONS - VOL 4: VISAPP, 2018, : 260 - 269
  • [36] Performance evaluation of lightweight convolutional neural networks on retinal lesion segmentation
    Siebert, M.
    Rostalski, P.
    MEDICAL IMAGING 2022: COMPUTER-AIDED DIAGNOSIS, 2022, 12033
  • [37] Skin Lesion Segmentation Based on Modification of SegNet Neural Networks
    Quoc Cuong Ninh
    Thi-Thao Tran
    Tien Thanh Tran
    Thi Anh Xuan Tran
    Van-Truong Pham
    PROCEEDINGS OF 2019 6TH NATIONAL FOUNDATION FOR SCIENCE AND TECHNOLOGY DEVELOPMENT (NAFOSTED) CONFERENCE ON INFORMATION AND COMPUTER SCIENCE (NICS), 2019, : 575 - 578
  • [38] HUMAN SKIN SEGMENTATION USING FULLY CONVOLUTIONAL NEURAL NETWORKS
    Ma, Chang-Hsian
    Shih, Huang-chia
    2018 IEEE 7TH GLOBAL CONFERENCE ON CONSUMER ELECTRONICS (GCCE 2018), 2018, : 168 - 170
  • [39] SKIN MELANOMA SEGMENTATION USING RECURRENT AND CONVOLUTIONAL NEURAL NETWORKS
    Attia, Mohamed
    Hossny, Mohamed
    Nahavandi, Saeid
    Yazdabadi, Anousha
    2017 IEEE 14TH INTERNATIONAL SYMPOSIUM ON BIOMEDICAL IMAGING (ISBI 2017), 2017, : 292 - 296
  • [40] DermoNet: densely linked convolutional neural network for efficient skin lesion segmentation
    Baghersalimi, Saleh
    Bozorgtabar, Behzad
    Schmid-Saugeon, Philippe
    Ekenel, Hazim Kemal
    Thiran, Jean-Philippe
    EURASIP JOURNAL ON IMAGE AND VIDEO PROCESSING, 2019,