A Novel Perspective for Multi-Modal Multi-Label Skin Lesion Classification
The efficacy of deep learning-based Computer-Aided Diagnosis (CAD) methods for skin diseases relies on analyzing multiple data modalities (i.e., clinical+dermoscopic images, and patient metadata) and addressing the challenges of multi-label classification. Current approaches tend to rely on limited multi-modal techniques and treat the multi-label problem as a multiple multi-class problem, overlooking issues related to imbalanced learning and multi-label correlation. This paper introduces the innovative Skin Lesion Classifier, utilizing a Multi-modal Multilabel TransFormer-based model (SkinM2Former). For multi-modal analysis, we introduce the Tri-Modal Cross-attention Transformer (TMCT) that fuses the three image and metadata modalities at various feature levels of a transformer encoder. For multi-label classification, we introduce a multi-head attention (MHA) module to learn multi-label correlations, complemented by an optimisation that handles multi-label and imbalanced learning problems. SkinM2Former achieves a mean average accuracy of 77.27% and a mean diagnostic accuracy of 77.85% on the public Derm7pt dataset, outperforming state-of-the-art (SOTA) methods.
Authors
- Yutong Xie (ORCID: https://orcid.org/0000-0002-6644-1250)
- Gustavo Carneiro (ORCID: https://orcid.org/0000-0002-5571-6220)
- Jodie Avery (ORCID: https://orcid.org/0000-0002-8857-9162)
- M. Louise Hull (ORCID: https://orcid.org/0000-0003-1813-3971)
- Yuan Zhang (ORCID: https://orcid.org/0000-0002-4174-0641)
- Wang Hu (ORCID: https://orcid.org/0000-0002-5319-7463)
Institutions
- University of Surrey (GB)
- Mohamed bin Zayed University of Artificial Intelligence (AE)
- The University of Adelaide (AU)
Publication Details
- Published
- 2025-02-26
- DOI
- https://doi.org/10.1109/wacv61041.2025.00350
- Citations
- 5
- Primary Topic
- Cutaneous Melanoma Detection and Management
- Type
- article
- Field-Weighted Citation Impact
- 5.23