Deep clustering via center-oriented margin free-triplet loss for skin lesion detection in highly ımbalanced datasets

Date
2022-06-29
Editor(s)
Advisor
Supervisor
Co-Advisor
Co-Supervisor
Instructor
Source Title
IEEE Journal of Biomedical and Health Informatics
Print ISSN
Electronic ISSN
Publisher
Institute of Electrical and Electronics Engineers Inc.
Volume
26
Issue
9
Pages
4679 - 4690
Language
English
Journal Title
Journal ISSN
Volume Title
Series
Abstract

Melanoma is a fatal skin cancer that is curable and has dramatically increasing survival rate when diagnosed at early stages. Learning-based methods hold significant promise for the detection of melanoma from dermoscopic images. However, since melanoma is a rare disease, existing databases of skin lesions predominantly contain highly imbalanced numbers of benign versus malignant samples. In turn, this imbalance introduces substantial bias in classification models due to the statistical dominance of the majority class. To address this issue, we introduce a deep clustering approach based on the latent-space embedding of dermoscopic images. Clustering is achieved using a novel center-oriented margin-free triplet loss (COM-Triplet) enforced on image embeddings from a convolutional neural network backbone. The proposed method aims to form maximally-separated cluster centers as opposed to minimizing classification error, so it is less sensitive to class imbalance. To avoid the need for labeled data, we further propose to implement COM-Triplet based on pseudo-labels generated by a Gaussian mixture model (GMM). Comprehensive experiments show that deep clustering with COM-Triplet loss outperforms clustering with triplet loss, and competing classifiers in both supervised and unsupervised settings. © 2013 IEEE.

Course
Other identifiers
Book Title
Citation
Published Version (Please cite this version)