Unsupervised and semi-supervised learning with Categorical Generative Adversarial Networks assisted by Wasserstein distance for dermoscopy image Classification

Abstract

Melanoma is a curable aggressive skin cancer if detected early. Typically, the diagnosis involves initial screening with subsequent biopsy and histopathological examination if necessary. Computer aided diagnosis offers an objective score that is independent of clinical experience and the potential to lower the workload of a dermatologist. In the recent past, success of deep learning algorithms in the field of general computer vision has motivated successful application of supervised deep learning methods in computer aided melanoma recognition. However, large quantities of labeled images are required to make further improvements on the supervised method. A good annotation generally requires clinical and histological confirmation, which requires significant effort. In an attempt to alleviate this constraint, we propose to use categorical generative adversarial network to automatically learn the feature representation of dermoscopy images in an unsupervised and semi-supervised manner. Thorough experiments on ISIC 2016 skin lesion chal- lenge demonstrate that the proposed feature learning method has achieved an average precision score of 0.424 with only 140 labeled images. Moreover, the proposed method is also capable of generating real-world like dermoscopy images.

References in corpus (5)

• Unsupervised Representation Learning with Deep Convolutional Generative Adversarial Networks
• f-GAN: Training Generative Neural Samplers using Variational Divergence Minimization
• Regularization With Stochastic Transformations and Perturbations for Deep Semi-Supervised Learning
• Semi-Supervised Learning with Generative Adversarial Networks
• An Overview of Melanoma Detection in Dermoscopy Images Using Image Processing and Machine Learning

Cited by in corpus (8)

• Generative Adversarial Network in Medical Imaging: A Review
• Semi-Supervised Deep Learning for Abnormality Classification in Retinal Images
• Counterfactual Explanations for Medical Image Classification and Regression using Diffusion Autoencoder
• General-to-Detailed GAN for Infrequent Class Medical Images
• Preliminary results in using Deep Learning to emulate BLOB, a nuclear interaction model
• Skin disease diagnosis with deep learning: a review
• A multi-stage GAN for multi-organ chest X-ray image generation and segmentation
• SSMD: Semi-Supervised Medical Image Detection with Adaptive Consistency and Heterogeneous Perturbation