Self-Supervised Learning to Increase the Performance of Skin Lesion Classification

• Published in: Electronics (Basel) 2020-11, Vol.9 (11), p.1930
• Main Authors: Kwasigroch, Arkadiusz, Grochowski, Michał, Mikołajczyk, Agnieszka
• Format: Article
• Language: English
• Subjects: Algorithms; Artificial neural networks; Classification; Datasets; Deep learning; Machine learning; Medical research; Neural networks; Self-supervised learning; Training
• ISSN: 2079-9292; 2079-9292
• DOI: 10.3390/electronics9111930

To successfully train a deep neural network, a large amount of human-labeled data is required. Unfortunately, in many areas, collecting and labeling data is a difficult and tedious task. Several ways have been developed to mitigate the problem associated with the shortage of data, the most common of which is transfer learning. However, in many cases, the use of transfer learning as the only remedy is insufficient. In this study, we improve deep neural models training and increase the classification accuracy under a scarcity of data by the use of the self-supervised learning technique. Self-supervised learning allows an unlabeled dataset to be used for pretraining the network, as opposed to transfer learning that requires labeled datasets. The pretrained network can be then fine-tuned using the annotated data. Moreover, we investigated the effect of combining the self-supervised learning approach with transfer learning. It is shown that this strategy outperforms network training from scratch or with transfer learning. The tests were conducted on a very important and sensitive application (skin lesion classification), but the presented approach can be applied to a broader family of applications, especially in the medical domain where the scarcity of data is a real problem.