Computer Science ›› 2023, Vol. 50 ›› Issue (4): 96-102.doi: 10.11896/jsjkx.220300054

• Computer Graphics & Multimedia • Previous Articles     Next Articles

Skin Lesion Segmentation Combining Boundary Enhancement and Multi-scale Attention

BAI Xuefei1, JIN Zhichao1, WANG Wenjian1,2, MA Yanan1   

  1. 1 School of Computer and Information Technology,Shanxi University,Taiyuan 030006,China
    2 Key Laboratory of Computational Intelligence and Chinese Information Processing of Ministry of Education(Shanxi University),Taiyuan 030006,China
  • Received:2022-03-07 Revised:2022-08-16 Online:2023-04-15 Published:2023-04-06
  • About author:BAI Xuefei,born in 1980,Ph.D,asso-ciate professor,is a member of China Computer Federation.Her main research interests include image proces-sing and machine learning.
    WANG Wenjian,born in 1968,Ph.D,professor,is a member of China Computer Federation.Her main research interests include image processing,machine learning and computing intelligence.
  • Supported by:
    National Natural Science Foundation of China(61703252,62076154,62276161,U21A20513) and Key Research and Development Program of Shanxi Province(202102150401013).

Abstract: In view of the various types of skin lesions in shape,color and size,which pose a huge challenge to the accurate segmentation of skin lesions,a skin lesion segmentation network that combines boundary enhancement and multi-scale attention is proposed(BEMA U-Net).It consists of two modules,one is called spatial multi-scale attention module,which is used to extract spatial global features,and the other is called boundary enhancement module,which is used to enhance the edge features of the lesion area.BEMA U-Net adds the two modules to the U-Net network with encoding and decoding structure,which can effectively suppress the interference of background noise in the image of lesions and enhance the edge details of lesions.In addition,the mixed loss function is designed,Dice loss and Boundary loss are combined,and the dynamic weight adjustment of the mixed loss function is realized in the training process,so that the network could carry out multiple supervision on the extraction of the overall features and edge details of the pathological images,and the problems of hair interference and edge blur in the segmentation of skin pathological images are alleviated.Experimental results on ISIC2017 and ISIC2018 public data sets show that the proposed algorithm has better segmentation effect for skin lesions with continuous edges and clear contours.

Key words: Skin lesion segmentation, Spatial multi-scale attention, Global feature, Boundary enhancement, U-Net

CLC Number: 

  • TP391
[1]JENKINS R W,FISHER D E.Treatment of advance melanoma in 2020 and beyond treatment in review[J].Journal of Investigative Dermatology,2020,141(1):23-31.
[2]MEMON A,BANNISTER P,ROGERS I,et al.Changing epidemiology and age-specific incidence of cutaneous malignant melanoma in england:an analysis of the national cancer registration data by age,gender and anatomical site,1981-2018[J].The Lancet Regional Health-Europe,2021,2:10024-10036.
[3]VIALE P H.The American cancer society’s facts & figures:2020 edition[J].Journal of the Advanced Practitioner in Oncology,2020,11(2):135-136.
[4]PERUCH F,BOGO F,BONAZZA M,et al.Simpler,faster,more accurate melanocytic lesion segmentation through meds[J].IEEE Transactions on Biomedical Engineering,2014,61(2):557-565.
[5]CELEBI M,IVATOMI H,SCHAEFER G,et al.Lesion border detection in dermoscopy images[J].Computerized Medical Imaging and Graphics,2009,33(2):148-153.
[6]WONG A,SCHARCANSKI J,FIEGUTH P.Automatic skin lesion segmentation via iterative stochastic region merging[J].IEEE Transactions on Information Technology in Biomedicine,2011,15(6):929-936.
[7]LONG J,SHELHAMER E,DARRELL T.et al.Fully convolutional networks for semantic segmentation[J].IEEE Transactions on Pattern Analysis and Machine Intelligence,2015,39(4):640-651.
[8]YUAN Y D,CHAO M,LO Y C.Automatic skin lesion segmen-tation using deep fully convolutional networks with jaccard distance[J].IEEE Transactions on Medical Imaging,2017,36(9):1876-1886.
[9]JIANG Y,CAO S M,TAO S X,et al.Skin lesion segmentation based on multi-scale attention convolutional neural network[J].IEEE Access,2020,8:122811-122825.
[10]RONNEBERGER O,FISCHER P,BROX T.U-net:convolu-tional networks for biomedical image segmentation[C]//Medical Image Computing and Computer-Assisted Intervention(MICCAI 2015).Cham:Springer,2015:234-241.
[11]SARKER M,RASHWAN H,AKRAM F,et al.SLSDeep:skin lesion segmentation based on dilated residual and pyramid pooling networks[C]//International Conference on Medical Image Computing and Computer-Assisted Intervention(MICCIA 2018).Cham:Springer,2018:21-29.
[12]AZAD R,ASADI-AGHBOLAGHI M,FATHY M,et al.Bi-directional convlstm u-net with densley connected convolutions[C]//2019 IEEE/CVF International Conference on Computer Vision Workshop(ICCVW).Piscataway:IEEE,2019:406-415.
[13]ASADI-AGHBOLAGHI M,AZAD R,FATHY M,et al.Multi-level context gating of embedded collective knowledge for medical image segmentation[J].arXiv:2003.05056,2020.
[14]TONG X,WEI J,SUN B,et al.ASCU-Net:attention gate,spatial and channel attention u-net for skin lesion segmentation[J].Diagnostics,2021,11(3):501.
[15]JHA D,RIEGLER M A,JOHANSEN D,et al.DoubleU-Net:a deep convolutional neural network for medical image segmentation[C]//2020 IEEE 33rd International Symposium on Compu-ter-Based Medical Systems(CBMS).Piscataway:IEEE,2020:558-564.
[16]ALI A,LI J,O’SHEA S,et al.A deep learning based approach to skin lesion border extraction with a novel edge detector in dermoscopy images[C]//The 2019 International Joint Confe-rence on Neural Networks(IJCNN).Piscataway:IEEE,2019:1-7.
[17]CHAKKARAVARTHY A,CHANDRASEKAR A.Anatomicalregion segmentation method from dermoscopy images of pigmented skin lesions[J].International Journal of Imaging Systems and Technology,2020,30(3):636-652.
[18]IOFFE S,SZEGEDY C.Batch normalization:accelerating deep network training by reducing internal covariate shift[C]//The 32nd International Conference on machine Learning.SAN DIEGO:JMLR,2015:448-456.
[19]QAMAR S,AHMAD P,SHEN L.Dense encoder-decoder-based architecture for skin lesion segmentation[J].Cognitive Computation,2021,13(2):583-594.
[20]QIAO Z,CUI Z,NIU X,et al.Image segmentation with pyramid dilated convolution based on resnet and u-net[C]//24th International Conference on Neural Information Processing(ICO-NIP).Cham:Springer,2017:364-372.
[21]ZHANG L,GAO H J,ZHANG J,et al.Optimization of the convolutional neural networks for automatic detection of skin can-cer[J].Open Medicine,2020,15(1):27-37.
[22]MILLETARI F,NAVAB N,AHMADI S A.V-Net:fully convolutional neural networks for volumetric medical image segmentation[C]//2016 Fourth International Conference on 3D Vision(3DV).Piscataway:IEEE,2016:565-571.
[23]BOKHOVKIN A,BURNAEV E.Boundary loss for remotesensing imagery semantic segmentation[C]//International Symposium on Neural Networks(ISNN).Cham:Springer,2019:388-401.
[24]SIMONYAN K,ZISSERMAN A.Very deep convolutional networks for large-scale image recognition[C]//International Conference on Learning Representations(ICLR 2015).ICLR,2015:1-14.
[25]JIA D,WEI D,SOCHER R,et al.ImageNet:a large-scale hierarchical image database[C]//2009 IEEE Conference on Computer Vision and Pattern Recognition(CVPR).Piscataway:IEEE,2009:248-255.
[26]CODELLA N,GUTMAN D,CELEBI M E,et al.Skin lesion analysis toward melanoma detection: a challenge at the 2017 international symposium on biomedical imaging (ISBI),hosted by the international skin imaging collaboration (ISIC)[C]//15th IEEE International Symposium on Biomedical Imaging (ISBI).Piscataway:IEEE,2018:168-172.
[27]CODELLA N,ROTEMBERG V,TSCHANDL P,et al.Skin lesion analysis toward melanoma detection 2018:a challenge hosted by the international skin imaging collaboration (ISIC)[J].arXiv:1902.03368,2019.
[28]ABADI M,BARHAM P,CHEN J,et al.TensorFlow:a system for large -scale machine learning[C]//12th USENIX Sympo-sium on Operating Systems Design and Implementation(OSDI).San Diego:USENIX ASSOC,2016:265-283.
[29]TAQI A M,AWAD A,AL-AZZO F.The impact of multi-optimizers and data augmentation on tensorflow convolutional neural network performance[C]//IEEE Conference on Multimedia Information Processing & Retrieval(MIPR).Piscataway:IEEE,2018:140-145.
[1] BAI Xuefei, MA Yanan, WANG Wenjian. Segmentation Method of Edge-guided Breast Ultrasound Images Based on Feature Fusion [J]. Computer Science, 2023, 50(3): 199-207.
[2] GAO Xin-yue, TIAN Han-min. Droplet Segmentation Method Based on Improved U-Net Network [J]. Computer Science, 2022, 49(4): 227-232.
[3] ZHAO Kai, AN Wei-chao, ZHANG Xiao-yu, WANG Bin, ZHANG Shan, XIANG Jie. Intracerebral Hemorrhage Image Segmentation and Classification Based on Multi-taskLearning of Shared Shallow Parameters [J]. Computer Science, 2022, 49(4): 203-208.
[4] HE Xiao-hui, QIU Fang-bing, CHENG Xi-jie, TIAN Zhi-hui, ZHOU Guang-sheng. High-resolution Image Building Target Detection Based on Edge Feature Fusion [J]. Computer Science, 2021, 48(9): 140-145.
[5] ZHANG Xiao-yu, WANG Bin, AN Wei-chao, YAN Ting, XIANG Jie. Glioma Segmentation Network Based on 3D U-Net++ with Fusion Loss Function [J]. Computer Science, 2021, 48(9): 187-193.
[6] QU Zhong, XIE Yi. Concrete Pavement Crack Detection Algorithm Based on Full U-net [J]. Computer Science, 2021, 48(4): 187-191.
[7] PENG Lei, ZHANG Hui. U-net for Pavement Crack Detection [J]. Computer Science, 2021, 48(11A): 616-619.
[8] LI Hua-ji, CHENG Jiang-hua, LIU Tong, CHENG Bang, ZHAO Kang-cheng. Low-light Image Enhancement Method Based on U-net++ Network [J]. Computer Science, 2021, 48(11A): 278-282.
[9] WANG Xin, ZHANG Hao-yu, LING Cheng. Semantic Segmentation of SAR Remote Sensing Image Based on U-Net Optimization [J]. Computer Science, 2021, 48(11A): 376-381.
[10] YANG Chun-de, JIA Zhu, LI Xin-wei. Study on ECG Signal Recognition and Classification Based on U-Net++ [J]. Computer Science, 2021, 48(10): 121-126.
[11] KANG Yan, CHEN Tie, LI Hao, YANG Bing, ZHANG Ya-chuan, BU Rong-jing. Urban Traffic Flow Completion with Multi-view Attention Mechanism [J]. Computer Science, 2021, 48(10): 177-184.
[12] TANG Yi-xing, LIU Xue-liang, HU She-jiao. Multi-orientation Partitioned Network for Person Re-identification [J]. Computer Science, 2021, 48(10): 204-211.
[13] GAO Qiang, GAO Jing-yang, ZHAO Di. GNNI U-net:Precise Segmentation Neural Network of Left Ventricular Contours for MRI Images Based on Group Normalization and Nearest Interpolation [J]. Computer Science, 2020, 47(8): 213-220.
Viewed
Full text


Abstract

Cited

  Shared   
  Discussed   
No Suggested Reading articles found!