Malaria Blood Cell Image Classification using Transfer Learning with Fine-Tune ResNet50 and Data Augmentation

• Published in: Jurnal RESTI (Rekayasa Sistem dan Teknologi Informasi) (Online) Vol. 6; no. 5; pp. 891 - 897
• Main Authors: Aris Muhandisin, Azhar, Yufis
• Format: Journal Article
• Language: English
• Published: Ikatan Ahli Informatika Indonesia 02.11.2022
• Subjects: convolutional neural network; fine-tuning; image classification; malaria; resnet50
• ISSN: 2580-0760; 2580-0760
• DOI: 10.29207/resti.v6i5.4322

Based on the WHO Report related to malaria, it is estimated that there will be 241 million malaria cases and 627,000 deaths from this disease globally in 2020 with the number of deaths increasing yearly. Preventing malaria disease conditions is through early detection. A more quick and precise malaria diagnosis method was required to simplify and reduce the detection process. Medical image classification could be carried out rapidly and precisely using machine learning or deep learning techniques. This research aims to diagnose malaria by classifying images of malaria blood cells using Deep Learning with a Transfer Learning approach. By utilizing various fine-tuning procedures and implementing data augmentation proposed method develops the method from previous studies. Two types of models Frozen ResNet50 and Fine-Tune ResNet50 are being tested. The dataset utilized will be augmented to improve model performance. This study makes use of the "NIH Malaria Cell Images Dataset" a dataset that contains a total of 27,660 image data. It is divided into two classes: parasitized and uninfected. The results are improved from previous research using the fine-tuned VGG16 model with an accuracy of 96% compared to this study using the fine-tuned ResNet50 model which achieved an accuracy score of 98%.