Transfer Learning-Based Semi-Supervised Generative Adversarial Network for Malaria Classification

Abstract

Malaria is a lethal disease responsible for thousands of deaths worldwide every year. Manual methods of malaria diagnosis are timeconsuming that require a great deal of human expertise and efforts. Computerbased automated diagnosis of diseases is progressively becoming popular. Although deep learning models show high performance in the medical field, it demands a large volume of data for training which is hard to acquire for medical problems. Similarly, labeling of medical images can be done with the help of medical experts only. Several recent studies have utilized deep learning models to develop efficient malaria diagnostic system, which showed promising results. However, the most common problem with these models is that they need a large amount of data for training. This paper presents a computer-aided malaria diagnosis system that combines a semi-supervised generative adversarial network and transfer learning. The proposed model is trained in a semi-supervised manner and requires less training data than conventional deep learning models. Performance of the proposed model is evaluated on a publicly available dataset of blood smear images (with malariainfected and normal class) and achieved a classification accuracy of 96.6%.