Comparison of Fine-tuned CNN Architectures for COVID-19 Infection Diagnosis

  • Jonathan Jonathan
  • Moeljono Widjaja
  • Alethea Suryadibrata UMN

Abstract

SARS-CoV-2 (COVID-19) virus spread quickly worldwide affects a variety of industries. The government took preventive steps to control the infection, such as diagnosing the human's lung by taking an X-Ray to see if the lungs were infected with COVID-19 or not. Using several pre-trained Convolutional Neural Network models as the basic model, this research deconstructs the comparison of fine-tuned architecture to identify which pre-trained model delivers the best outcomes in diagnosis by applying machine learning. Comparison is conducted using two scenarios that use batch sizes 64 and 32. Accuracy and f1 score are two evaluation metrics used to justify the model's good performance because the images in the real world, especially for positive classes, are scarce. According to the study, EfficientNetB0 outperforms other pre-trained models, namely ResNet50V2 and Xception, which achieved an accuracy of 0.895 and f1 score of 0.8871.

Downloads

Download data is not yet available.
Published
2024-07-11
How to Cite
Jonathan, J., Widjaja, M., & Suryadibrata, A. (2024). Comparison of Fine-tuned CNN Architectures for COVID-19 Infection Diagnosis. Ultimatics : Jurnal Teknik Informatika, 16(1), 63-68. https://doi.org/https://doi.org/10.31937/ti.v16i1.3652