Vision transformer-based fuzzy semi-supervised multimodal learning for skin lesion image classification
摘要
Skin lesion identification and treatment is a critical and challenging task in the healthcare sector. Usually, several CNN classifiers have been applied to identify skin diseases using only simple visual information. However, in this context, recent research has focused on multimodalities. Since different modalities contain unique information their combination represents the underlying concepts more clearly, increasing the accuracy of the model. This study proposes a multimodal architecture integrating two methods for fusing image and metadata features. One is, Vision Transformers (ViT) for extracting visual characteristics from images, and the other is an MLP (Multilayer perceptron) for the metadata information. Following pre-training on massive volumes of data, the ViT surpasses cutting-edge convolutional networks in various benchmarks while using fewer computational resources to train and achieve better performance. Furthermore, to avoid the problems of discovering a high volume of labeled data, the fuzziness-based semi-supervised deep learning (FSSDL) method is applied. Using low-quality fuzzy samples for retraining classifiers has been found to enhance the performance of the classifier. The integrated model was applied to the PAD-UFES_20 dataset, containing both image and metadata. In order to strengthen this research, the dataset was classified for binary and multiclass classification. The model’s performance was assessed using various performance metrics (e.g.