Journal of Machine and Computing


Transfer Learning Based Weighted Deep Learning Ensemble Model for Medical Image Classification



Journal of Machine and Computing

Received On : 10 August 2023

Revised On : 12 December 2023

Accepted On : 08 June 2024

Published On : 05 July 2024

Volume 04, Issue 03

Pages : 661-668


Abstract


Malignant melanoma is a well-known and deadly form of cancer that originates from epidermal melanocytes in humans. Early detection of such diseases, including various forms of cancer, is necessary for speeding up diagnosis and enhancing patient outcomes. A novel transfer learning-based ensemble-deep learning model was presented for diagnosing diseases at a preliminary stage. Data augmentation was used to increase the dataset, and integration of Inception-v3, DenseNet-121, and ResNet-50 techniques, along with an ensemble method, was employed to overcome the scarcity of labeled datasets and increase the accuracy as well as make the model more robust. The proposed system was trained and tested employing the International Skin Imaging Collaboration (ISIC) dataset. The suggested ensemble model gained the best performance, producing 98% accuracy, 98% area under the curve, 98% precision, and 98% F1 score. The proposed model outperformed the existing state-of-the-art models in disease classification. Furthermore, the proposed model will be beneficial for medical diagnosis and reduce the incidence of various diseases.


Keywords


Ensemble Model, Skin Disease, Classification, Deep Learning, Transfer Learning.


  1. J. Amin et al., “Integrated design of deep features fusion for localization and classification of skin cancer,” Pattern Recognition Letters, vol. 131, pp. 63–70, Mar. 2020, doi: 10.1016/j.patrec.2019.11.042.
  2. D. Bisla, A. Choromanska, R. S. Berman, J. A. Stein, and D. Polsky, “Towards Automated Melanoma Detection With Deep Learning: Data Purification and Augmentation,” 2019 IEEE/CVF Conference on Computer Vision and Pattern Recognition Workshops (CVPRW), Jun. 2019, doi: 10.1109/cvprw.2019.00330.
  3. H. M. Balaha and A. E.-S. Hassan, “Skin cancer diagnosis based on deep transfer learning and sparrow search algorithm,” Neural Computing and Applications, vol. 35, no. 1, pp. 815–853, Sep. 2022, doi: 10.1007/s00521-022-07762-9.
  4. V. L. Cohen et al., “Staging uveal melanoma with whole-body positron-emission tomography/computed tomography and abdominal ultrasound: Low incidence of metastatic disease, high incidence of second primary cancers,” Middle East African Journal of Ophthalmology, vol. 25, no. 2, p. 91, 2018, doi: 10.4103/meajo.meajo_96_18.
  5. A. Dahou, A. O. Aseeri, A. Mabrouk, R. A. Ibrahim, M. A. Al-Betar, and M. A. Elaziz, “Optimal Skin Cancer Detection Model Using Transfer Learning and Dynamic-Opposite Hunger Games Search,” Diagnostics, vol. 13, no. 9, p. 1579, Apr. 2023, doi: 10.3390/diagnostics13091579.
  6. G. Frisso et al., “Functional Studies and In Silico Analyses to Evaluate Non-Coding Variants in Inherited Cardiomyopathies,” International Journal of Molecular Sciences, vol. 17, no. 11, p. 1883, Nov. 2016, doi: 10.3390/ijms17111883.
  7. T. H. Johansen et al., “Recent advances in hyperspectral imaging for melanoma detection,” WIREs Computational Statistics, vol. 12, no. 1, Apr. 2019, doi: 10.1002/wics.1465.
  8. A. Julian, B. N. Narendra, A. Chiranjeevi, and A. V. C. Reddy, “Prediction of Brain Tumor Classification by using CNN,” 2023 International Conference on Computer Communication and Informatics (ICCCI), Jan. 2023, doi: 10.1109/iccci56745.2023.10128408.
  9. M. A. Khan, M. Sharif, T. Akram, S. A. C. Bukhari, and R. S. Nayak, “Developed Newton-Raphson based deep features selection framework for skin lesion recognition,” Pattern Recognition Letters, vol. 129, pp. 293–303, Jan. 2020, doi: 10.1016/j.patrec.2019.11.034.
  10. T. Kränke, K. Tripolt-Droschl, L. Röd, R. Hofmann-Wellenhof, M. Koppitz, and M. Tripolt, “New AI-algorithms on smartphones to detect skin cancer in a clinical setting—A validation study,” Plos One, Vol. 18, no. 2, p. e0280670, Feb. 2023, doi: 10.1371/journal.pone.0280670.
  11. V. Vidya Lakshmi and J. S. Leena Jasmine, “A Hybrid Artificial Intelligence Model for Skin Cancer Diagnosis,” Computer Systems Science and Engineering, vol. 37, no. 2, pp. 233–245, 2021, doi: 10.32604/csse.2021.015700.
  12. S. O. Manoj, K. R. Abirami, A. Victor, and M. Arya, “Automatic Detection and Categorization of Skin Lesions for Early Diagnosis of Skin Cancer Using YOLO-v3 - DCNN Architecture,” Image Analysis & Stereology, vol. 42, no. 2, pp. 101–117, Jul. 2023, doi: 10.5566/ias.2773.
  13. T. Mazhar et al., “The Role of Machine Learning and Deep Learning Approaches for the Detection of Skin Cancer,” Healthcare, vol. 11, no. 3, p. 415, Feb. 2023, doi: 10.3390/healthcare11030415.
  14. A. Mahbod, G. Schaefer, C. Wang, R. Ecker, and I. Ellinge, “Skin Lesion Classification Using Hybrid Deep Neural Networks,” ICASSP 2019 - 2019 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP), May 2019, doi: 10.1109/icassp.2019.8683352.
  15. P. Natha And R. Pothuraju, “Skin Cancer Detection Using Machine Learning Classification Models,” Nov. 2023, doi: 10.20944/preprints202311.0248.v1.
  16. N. Nigar, A. Wajid, S.Islam, and M.K.Shahzad, “Skin Cancer Classification: A Deep Learning Approach,” Pakistan Journal of Science, vol. 75, no. 02, Jul. 2023, doi: 10.57041/pjs.v75i02.851.
  17. R. O. Ogundokun et al., “Enhancing Skin Cancer Detection and Classification in Dermoscopic Images through Concatenated MobileNetV2 and Xception Models,” Bioengineering, vol. 10, no. 8, p. 979, Aug. 2023, doi: 10.3390/bioengineering10080979.
  18. L. Riaz et al., “A Comprehensive Joint Learning System to Detect Skin Cancer,” IEEE Access, vol. 11, pp. 79434–79444, 2023, doi: 10.1109/access.2023.3297644.
  19. Haldorai, B. L. R, S. Murugan, and M. Balakrishnan, “An Investigation on Different Approaches for Medical Imaging,” EAI/Springer Innovations in Communication and Computing, pp. 57–75, 2024, doi: 10.1007/978-3-031-53972-5_3.
  20. J. S M, M. P, C. Aravindan, and R. Appavu, “Classification of skin cancer from dermoscopic images using deep neural network architectures,” Multimedia Tools and Applications, vol. 82, no. 10, pp. 15763–15778, Oct. 2022, doi: 10.1007/s11042-022-13847-3.
  21. M. Zambrano-Román, J. R. Padilla-Gutiérrez, Y. Valle, J. F. Muñoz-Valle, and E. Valdés-Alvarado, “Non-Melanoma Skin Cancer: A Genetic Update and Future Perspectives,” Cancers, vol. 14, no. 10, p. 2371, May 2022, doi: 10.3390/cancers14102371.

Acknowledgements


Author(s) thanks to Dr. Riyazuddin Y MD for this research completion and support.


Funding


No funding was received to assist with the preparation of this manuscript.


Ethics declarations


Conflict of interest

The authors have no conflicts of interest to declare that are relevant to the content of this article.


Availability of data and materials


Data sharing is not applicable to this article as no new data were created or analysed in this study.


Author information


Contributions

All authors have equal contribution in the paper and all authors have read and agreed to the published version of the manuscript.


Corresponding author


Rights and permissions


Open Access This article is licensed under a Creative Commons Attribution NoDerivs is a more restrictive license. It allows you to redistribute the material commercially or non-commercially but the user cannot make any changes whatsoever to the original, i.e. no derivatives of the original work. To view a copy of this license, visit https://creativecommons.org/licenses/by-nc-nd/4.0/


Cite this article


Giddaluru Lalitha and Riyazuddin Y MD, “Transfer Learning Based Weighted Deep Learning Ensemble Model for Medical Image Classification”, Journal of Machine and Computing, pp. 661-668, July 2024. doi: 10.53759/7669/jmc202404063.


Copyright


© 2024 Giddaluru Lalitha and Riyazuddin Y MD. This is an open access article distributed under the terms of the Creative Commons Attribution License, which permits unrestricted use, distribution, and reproduction in any medium, provided the original author and source are credited.