Skin Lesion Classification using EfficientNet B0 and B1 via Transfer Learning for Computer Aided Diagnosis

Josh Frederich, Julieta Himawan, Mia Rizkinia

Research output: Contribution to journalConference articlepeer-review

Abstract

Automatic diagnosis of skin cancer from skin lesions using dermoscopy images has still been a challenging task for artificial intelligence, especially in Artificial Neural Networks using deep learning. Currently, previous research on skin lesion dermoscopy classification models have shown competitive performances whose accuracy remains to be improved. Applying and configuring the correct deep learning architecture has been an important factor in making an accurate automatic diagnosis. Hence, in this research, a new classification model was developed by using the latest architectures i.e., EfficientNet-B0 and B1 via transfer learning approach. This research was conducted by using HAM10000 dataset. To avoid overfitting due to the imbalanced data, the data were trained under some scenarios varying five variations of augmentation, class weight, and normalization within each model. The result showed that for EfficientNet-B0, the best model was the one that used only augmentation, which has the accuracy, precision, recall, and F1-scores of 91%, 76%, 68%, and 71%, respectively. While for EfficientNet-B1, the best model used augmentation and class weights with accuracy, precision, recall, and F1-score of 89%, 78%, 73%, and 73%, respectively. The best EfficientNet-B1 model can outperform the existing state-of-the-art model with an increase in recall and F1-score by 2% and 12% from the semi-supervised model, respectively. The model can also be integrated with a graphical user interface for dermatologists to use in dermoscopy examinations.

Original languageEnglish
Article number110002
JournalAIP Conference Proceedings
Volume3080
Issue number1
DOIs
Publication statusPublished - 7 Mar 2024
Event15th Asian Congress on Biotechnology in conjunction with the 7th International Symposium on Biomedical Engineering, ACB-ISBE 2022 - Bali, Indonesia
Duration: 2 Oct 20226 Oct 2022

Keywords

  • Automatic diagnosis
  • Deep learning
  • EfficientNet
  • Skin lesions
  • Transfer learning

Fingerprint

Dive into the research topics of 'Skin Lesion Classification using EfficientNet B0 and B1 via Transfer Learning for Computer Aided Diagnosis'. Together they form a unique fingerprint.

Cite this