Machine Learning and Deep Learning Algorithms for Skin Cancer Classification from Dermoscopic Images

We carry out a critical assessment of machine learning and deep learning models for the classification of skin tumors. Machine learning (ML) algorithms tested in this work include logistic regression, linear discriminant analysis, k-nearest neighbors classifier, decision tree classifier and Gaussian...

Celý popis

Uloženo v:
Podrobná bibliografie
Vydáno v:Bioengineering (Basel) Ročník 9; číslo 3; s. 97
Hlavní autoři: Bechelli, Solene, Delhommelle, Jerome
Médium: Journal Article
Jazyk:angličtina
Vydáno: Switzerland MDPI AG 27.02.2022
MDPI
Témata:
Accuracy
Algorithms
Archives & records
Artificial neural networks
Bayesian analysis
Bioengineering
Classification
Classifiers
convolutional neural network
Datasets
Decision analysis
Decision trees
Deep learning
Discriminant analysis
Image classification
Learning algorithms
Machine learning
Melanoma
Model accuracy
Neural networks
Skin cancer
Transfer learning
Tumors
deep learning
skin cancer
image classification
machine learning
convolutional neural network
ISSN:2306-5354, 2306-5354
On-line přístup:Získat plný text
Tagy: Přidat tag
Žádné tagy, Buďte první, kdo vytvoří štítek k tomuto záznamu!
Popis
Shrnutí:We carry out a critical assessment of machine learning and deep learning models for the classification of skin tumors. Machine learning (ML) algorithms tested in this work include logistic regression, linear discriminant analysis, k-nearest neighbors classifier, decision tree classifier and Gaussian naive Bayes, while deep learning (DL) models employed are either based on a custom Convolutional Neural Network model, or leverage transfer learning via the use of pre-trained models (VGG16, Xception and ResNet50). We find that DL models, with accuracies up to 0.88, all outperform ML models. ML models exhibit accuracies below 0.72, which can be increased to up to 0.75 with ensemble learning. To further assess the performance of DL models, we test them on a larger and more imbalanced dataset. Metrics, such as the F-score and accuracy, indicate that, after fine-tuning, pre-trained models perform extremely well for skin tumor classification. This is most notably the case for VGG16, which exhibits an F-score of 0.88 and an accuracy of 0.88 on the smaller database, and metrics of 0.70 and 0.88, respectively, on the larger database.
Bibliografie:ObjectType-Article-1
SourceType-Scholarly Journals-1
ObjectType-Feature-2
content type line 14
content type line 23
ISSN:2306-5354
2306-5354
DOI:10.3390/bioengineering9030097