Skin Disease Classification Based on Multi-level Feature Fusion and Attention Mechanism

Xinhua Liu,Licheng Yang,Xiaolin Ma,Hailan Kuang
DOI: https://doi.org/10.1109/iciba56860.2023.10165024
2023-01-01
Abstract:Up to now, skin cancer is one of the most common cancers with great danger, thus it is very essential to diagnose at an early state to save patients lives. Recent work has applied methods based on convolutional neural networks(CNNs) to classify images of skin lesions for diagnosis. However, most of these methods failed to work well on mobile or embedded devices due to the huge computational cost and memory requirements. Even though lightweight CNNs have been developed for mobile devices, they perform poorly on multi-classification tasks. To solve these shortcomings, we propose a method based on multi-level feature fusion and attention mechanism, MFA-MViT. The effectiveness of MFA-MViT was evaluated on the ISIC-PAD dataset built based on the ISIC2019 and PAD-UFES-20 datasets. The experimental results show that the classification effect is better than that of similar methods within a small amount of computation and parameters.
What problem does this paper attempt to address?