Mixed Re-Sampled Class-Imbalanced Semi-Supervised Learning for Skin Lesion Classification

被引:2
作者
Tian, Ye [1 ]
Zhang, Liguo [1 ,2 ]
Shen, Linshan [1 ]
Yin, Guisheng [1 ]
Chen, Lei [3 ]
机构
[1] Harbin Engn Univ, Coll Comp Sci & Technol, Harbin 150001, Peoples R China
[2] Heilongjiang Hengxun Technol Co Ltd, Harbin 150001, Peoples R China
[3] Georgia Southern Univ, Coll Engn & Informat Technol, Statesboro, GA 30458 USA
关键词
Skin lesion classification; class imbalance; semi-supervised learning;
D O I
10.32604/iasc.2021.016314
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Skin cancer is one of the most common types of cancer in the world, melanoma is considered to be the deadliest type among other skin cancers. Quite recently, automated skin lesion classification in dermoscopy images has become a hot and challenging research topic due to its essential way to improve diagnostic performance, thus reducing melanoma deaths. Convolution Neural Networks (CNNs) are at the heart of this promising performance among a variety of supervised classification techniques. However, these successes rely heavily on large amounts of class-balanced clearly labeled samples, which are expensive to obtain for skin lesion classification in the real world. To address this issue, we propose a mixed re-sampled (MRS) class-imbalanced semi-supervised learning method for skin lesion classification, which consists of two phases, re-sampling, and multiple mixing methods. To counter class imbalance problems, a re-sampling method for semi-supervised learning is proposed, and focal loss is introduced to the semisupervised learning to improve the classification performance. To make full use of unlabeled data to improve classification performance, Fmix and Mixup are used to mix labeled data with the pseudo-labeled unlabeled data. Experiments are conducted to demonstrate the effectiveness of the proposed method on class-imbalanced datasets, the results show the effectiveness of our method as compared with other state-of-the-art semi-supervised methods.
引用
收藏
页码:195 / 211
页数:17
相关论文
共 27 条
  • [1] Berthelot D, 2019, ADV NEUR IN, V32
  • [2] Codella NCF, 2018, I S BIOMED IMAGING, P168, DOI 10.1109/ISBI.2018.8363547
  • [3] Combalia M., 2019, BCN20000: Dermoscopic Lesions in the Wild
  • [4] Large Scale Fine-Grained Categorization and Domain-Specific Transfer Learning
    Cui, Yin
    Song, Yang
    Sun, Chen
    Howard, Andrew
    Belongie, Serge
    [J]. 2018 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2018, : 4109 - 4118
  • [5] Dermatologist-level classification of skin cancer with deep neural networks
    Esteva, Andre
    Kuprel, Brett
    Novoa, Roberto A.
    Ko, Justin
    Swetter, Susan M.
    Blau, Helen M.
    Thrun, Sebastian
    [J]. NATURE, 2017, 542 (7639) : 115 - +
  • [6] Skin Lesion Classification Using CNNs With Patch-Based Attention and Diagnosis-Guided Loss Weighting
    Gessert, Nils
    Sentker, Thilo
    Madesta, Frederic
    Schmitz, Ruediger
    Kniep, Helge
    Baltruschat, Ivo
    Werner, Rene
    Schlaefer, Alexander
    [J]. IEEE TRANSACTIONS ON BIOMEDICAL ENGINEERING, 2020, 67 (02) : 495 - 503
  • [7] Harris E, 2002, ARXIV 200212047
  • [8] Learning Deep Representation for Imbalanced Classification
    Huang, Chen
    Li, Yining
    Loy, Chen Change
    Tang, Xiaoou
    [J]. 2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, : 5375 - 5384
  • [9] Depthwise Separable Convolutional Neural Network for Skin Lesion Classification
    Kassani, Sara Hosseinzadeh
    Kassani, Peyman Hosseinzadeh
    Wesolowski, Michal J.
    Schneider, Kevin A.
    Deters, Ralph
    [J]. 2019 IEEE 19TH INTERNATIONAL SYMPOSIUM ON SIGNAL PROCESSING AND INFORMATION TECHNOLOGY (ISSPIT 2019), 2019,
  • [10] Diagnostic accuracy of dermoscopy
    Kittler, H
    Pehamberger, H
    Wolff, K
    Binder, M
    [J]. LANCET ONCOLOGY, 2002, 3 (03) : 159 - 165