1. Medical Big Data Analysis with Attention and Large Margin Loss Model for Skin Lesion Application
- Author
-
Wei Hu, Jing Wu, Yining Li, Hong Guo, Yuan Wen, Xiaoming Liu, and Tianyi Liu
- Subjects
business.industry ,Computer science ,Big data ,020206 networking & telecommunications ,Pattern recognition ,02 engineering and technology ,Theoretical Computer Science ,Lesion ,Hardware and Architecture ,Control and Systems Engineering ,Margin (machine learning) ,Modeling and Simulation ,Signal Processing ,Pattern recognition (psychology) ,0202 electrical engineering, electronic engineering, information engineering ,medicine ,020201 artificial intelligence & image processing ,Artificial intelligence ,medicine.symptom ,business ,Skin lesion ,Information Systems ,Skin imaging - Abstract
Due to melanoma is one of the skin cancers with the highest mortality rate and have a large amount of data during the collection and diagnosis, there is an urgent need to improve the diagnostic efficiency and accuracy. However, there remain problems in analyzing medical big data for skin lesion application, such as the intra-class variation and inter-class similarity in skin lesion images and the lacks of ability to focus on the lesion area affecting the classification results of the model. To address these dilemmas, in this paper, we proposed a novel machine learning-based approach that builds on top of DenseNet. It combines the attention mechanism and large margin loss to enhance the classification accuracy in terms of intra-class compactness and inter-class separability. We evaluated our model on ISIC 2017 (International Skin Imaging Collaboration) dataset, which has achieved 92% of Mean AUC. The experimental results show the effectiveness of our solution outperforms the state-of-the-art significantly in classify skin lesion and can accurately classify malignant melanoma on medical images.
- Published
- 2021