Single-Shared Network with Prior-Inspired Loss for Parameter-Efficient Multi-Modal Imaging Skin Lesion Classification
Not on the map. Why?
Abstract (arXiv)
In this study, we introduce a multi-modal approach that efficiently integrates multi-scale clinical and dermoscopy features within a single network, thereby substantially reducing model parameters. The proposed method includes three novel fusion schemes. Firstly, unlike current methods that usually employ two individual models for for clinical and dermoscopy modalities, we verified that multimodal feature can be learned by sharing the parameters of encoder while leaving the individual modal-specific classifiers. Secondly, the shared cross-attention module can replace the individual one to efficiently interact between two modalities at multiple layers. Thirdly, different from current methods that equally optimize dermoscopy and clinical branches, inspired by prior knowledge that dermoscopy images play a more significant role than clinical images, we propose a novel biased loss. This loss guides the single-shared network to prioritize dermoscopy information over clinical information, implicitly learning a better joint feature representation for the modal-specific task. Extensive experiments on a well-recognized Seven-Point Checklist (SPC) dataset and a collected dataset demonstrate the effectiveness of our method on both CNN and Transformer structures. Furthermore, our method exhibits superiority in both accuracy and model parameters compared to currently advanced methods.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 13 of its 34 references that are in the map. First posted on arXiv 28 Mar 2024.
Builds on
13 of 34 references.
References · 13
- 2015Deep Residual Learning for Image Recognitioncited by 51,809
- 2014Adam: A Method for Stochastic Optimizationcited by 36,960
- 2021Swin Transformer: Hierarchical Vision Transformer using Shifted Windowscited by 12,082
- 2016Densely Connected Convolutional Networkscited by 6,691
- 2018Averaging Weights Leads to Wider Optima and Better Generalizationcited by 1,194
- 2020PAD-UFES-20: a skin lesion dataset composed of patient data and clinical images collected from smartphonescited by 116
- 2021Exploring the Equivalence of Siamese Self-Supervised Learning via A Unified Gradient Frameworkcited by 49
- 2022Learning Where to Learn in Cross-View Self-Supervised Learningcited by 31
- 2021Learning Deep Multimodal Feature Representation with Asymmetric Multi-layer Fusioncited by 25
- 2021Single Model Deep Learning on Imbalanced Small Datasets for Skin Lesion Classificationcited by 23
- 2022TFormer: A throughout fusion transformer for multi-modal skin lesion diagnosiscited by 15
- 2021Graph-Based Intercategory and Intermodality Network for Multilabel Classification and Melanoma Diagnosis of Skin Lesions in Dermoscopy and Clinical Imagescited by 6
1 more
- 2020Multi-Scale Weight Sharing Network for Image Recognitioncited by 2
Built on by
Not cited yet.