MMFNet: A Multi-modal and Multi-attention Fusion Network for Multi-task Skin Disease Classification
Skin diseases rank among the most prevalent ailments in humans, which underscores the critical importance of early detection and diagnosis. Considering clinical practice, the integration of information from various modalities holds significant potential to enhance diagnostic accuracy and precision. In this study, we propose a multi-modal and multi-attention fusion network for multi-task skin disease classification. We initially devise the Shifted Window Cross-attention Fusion (SWCF) module based on the shifted window self-attention mechanism, leveraging the advantages of the attention mechanism to learn the correlations between different modalities and fully integrate multi-modal features. Subsequently, taking into account the heterogeneity of the data, we employ the Heterogeneous Data Cross-attention Fusion (HDCF) module using the idea of information decoupling and separation processing to integrate image and text data. Additionally, we suggest the Dynamic Loss Weight Allocation (DLWA) method for multi-task learning to refine the training procedure. We confirm the superiority of the proposed method on the publicly available multi-modal skin lesion dataset, Derm7pt. The average accuracy of multi-modal skin disease classification is 79.14%, surpassing the current state-of-the-art methods.
Authors
- Ming Gao (ORCID: https://orcid.org/0000-0003-0723-1197)
- Xinlei Zhu (ORCID: https://orcid.org/0000-0002-3087-9912)
- Chuang Wang (ORCID: https://orcid.org/0009-0002-6872-557X)
- Pengjiang Qian (ORCID: https://orcid.org/0000-0002-5596-3694)
- E. Y. K. Ng (ORCID: https://orcid.org/0000-0002-5701-1080)
- Li Liu (ORCID: https://orcid.org/0000-0003-3022-936X)
- Weiwei Cai
Institutions
- Jiangnan University (CN)
- Nanyang Technological University (SG)
Publication Details
- Published
- 2024-12-03
- DOI
- https://doi.org/10.1109/bibm62325.2024.10822724
- Citations
- 1
- Primary Topic
- Cutaneous Melanoma Detection and Management
- Type
- article
- Field-Weighted Citation Impact
- 0.28
Funders
- National Natural Science Foundation of China