Melanoma Classification using Machine Learning and Deep Learning

被引:0
作者
Tran Anh Vu [1 ]
Pham Quang Son [1 ]
Dinh Nghia Hiep [1 ]
Hoang Quang Huy [1 ]
Nguyen Phan Kien [1 ]
Pham Thi Viet Huong [2 ]
机构
[1] Hanoi Univ Sci & Technol, Sch Elect & Elect Engn, Hanoi, Vietnam
[2] Vietnam Natl Univ, Int Sch, Hanoi, Vietnam
来源
2023 1ST INTERNATIONAL CONFERENCE ON HEALTH SCIENCE AND TECHNOLOGY, ICHST 2023 | 2023年
关键词
Melanoma; Histogram of Oriented Gradients; Machine learning; Deep learning; CANCER;
D O I
10.1109/ICHST59286.2023.10565364
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
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.
引用
收藏
页数:6
相关论文
共 26 条
[1]  
[Anonymous], Cancer Stat Facts: Melanoma of the Skin
[2]   Acquired melanocytic nevi as risk factor for melanoma development. A comprehensive review of epidemiological data [J].
Bauer, J ;
Garbe, C .
PIGMENT CELL RESEARCH, 2003, 16 (03) :297-306
[3]  
Beck Teresa L, 2017, Prim Care, V44, pe1, DOI [10.1016/j.pop.2016.09.005, 10.1016/j.pop.2016.09.005]
[4]   CANCER STATISTICS, 1991 [J].
BORING, CC ;
SQUIRES, TS ;
TONG, T .
CA-A CANCER JOURNAL FOR CLINICIANS, 1991, 41 (01) :19-36
[5]   Deep neural networks are superior to dermatologists in melanoma image classification [J].
Brinker, Titus J. ;
Hekler, Achim ;
Enk, Alexander H. ;
Berking, Carola ;
Haferkamp, Sebastian ;
Hauschild, Axel ;
Weichenthal, Michael ;
Klode, Joachim ;
Schadendorf, Dirk ;
Holland-Letz, Tim ;
von Kalle, Christof ;
Froehling, Stefan ;
Schilling, Bastian ;
Utikal, Jochen S. .
EUROPEAN JOURNAL OF CANCER, 2019, 119 :11-17
[6]   Melanoma detection from dermoscopy images using Nasnet Mobile with Transfer Learning [J].
Cakmak, Mustafa ;
Tenekeci, Mehmet Emin .
29TH IEEE CONFERENCE ON SIGNAL PROCESSING AND COMMUNICATIONS APPLICATIONS (SIU 2021), 2021,
[7]   k-Nearest Neighbour Classifiers - A Tutorial [J].
Cunningham, Padraig ;
Delany, Sarah Jane .
ACM COMPUTING SURVEYS, 2021, 54 (06)
[8]   Lung Cancer Risk Prediction with Machine Learning Models [J].
Dritsas, Elias ;
Trigka, Maria .
BIG DATA AND COGNITIVE COMPUTING, 2022, 6 (04)
[9]   Dermatologist-level classification of skin cancer with deep neural networks [J].
Esteva, Andre ;
Kuprel, Brett ;
Novoa, Roberto A. ;
Ko, Justin ;
Swetter, Susan M. ;
Blau, Helen M. ;
Thrun, Sebastian .
NATURE, 2017, 542 (7639) :115-+
[10]  
Evgeniou T., 2001, Machine learning and its applications. Advanced lectures, P249