Melanoma Classification Using Machine Learning and Deep Learning

Tóm tắt

This paper investigates the application of machine learning algorithms, including ResNet18, for enhancing melanoma detection using the ISIC2020 dataset, comprising two classes: benign and malignant. In addition to traditional classification algorithms (SVM, random forest, KNN, and logistic regression), ResNet18, a deep learning convolutional neural network, was employed, showcasing an impressive accuracy of 96.98%. The study focuses on feature selection using the Histogram of Oriented Gradients (HOG) method and applies preprocessing techniques to optimize data quality. HOG features are extracted to represent skin lesions, and all algorithms are trained and evaluated using cross-validation, yielding accuracy rates ranging from 84 % to 96.98%. The study highlights the potential of machine learning, including HOG-based feature selection and ResNet18, for accurate melanoma detection. Integrating these models into clinical practice could significantly improve diagnostic accuracy and aid in treatment decisions. Further research directions involve exploring ensemble techniques, addressing interpretability concerns, and investigating the synergistic potential of combining traditional algorithms with deep learning models. Overall, this study demonstrates the effectiveness of HOG-based feature selection and ResNet18 in enhancing melanoma detection, underscoring the promise of machine learning in advancing patient outcomes.