Balanced Supervised Contrastive Learning for Skin Lesion Classification

Lan Yan,Kenli Li
DOI: https://doi.org/10.1109/smc53992.2023.10393927
2023-01-01
Abstract:Deep neural networks have emerged as an important tool for computer-aided diagnosis. However, deep models for skin lesion classification still face the challenges of intra-class variation and inter-class similarity, as well as data imbalance. To address these challenges, in this paper, we propose a balanced supervised contrastive learning (BSCL) approach for the skin lesion classification task. Our model consists of two branches for supervised contrastive learning and classification, respectively. The introduced supervised contrastive learning branch helps the network to learn more discriminative representations. Moreover, we design both a category -averaging strategy which averages the instances of every class in a mini-batch, and a category-complement strategy which makes all categories to appear in each mini-batch, to balance the influence from different skin lesion categories. Besides, we introduce a multi-weighted classification loss to learn a balanced classifier. Extensive experiments on two benchmarks demonstrate that our approach is able to learn strong feature representations and achieve state-of-the-art skin lesion classification performance.
What problem does this paper attempt to address?