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

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

Malaria, Image Classification, Convolutional Neural Network, Fine-Tuning, ResNet50

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%.

Aris Muhandisin1
, Yufis Azhar2

University of Muhammadiyah Malang

02-11-2022

Fajar bagus W

PDF

Indonesia

Text