A comparative analysis of deep learning architectures on high variation malaria parasite classification dataset.

Journal: Tissue & cell
PMID:

Abstract

Malaria, one of the leading causes of death in underdeveloped countries, is primarily diagnosed using microscopy. Computer-aided diagnosis of malaria is a challenging task owing to the fine-grained variability in the appearance of some uninfected and infected class. In this paper, we transform a malaria parasite object detection dataset into a classification dataset, making it the largest malaria classification dataset (63,645 cells), and evaluate the performance of several state-of-the-art deep neural network architectures pretrained on both natural and medical images on this new dataset. We provide detailed insights into the variation of the dataset and qualitative analysis of the results produced by the best models. We also evaluate the models using an independent test set to demonstrate the model's ability to generalize in different domains. Finally, we demonstrate the effect of conditional image synthesis on malaria parasite detection. We provide detailed insights into the influence of synthetic images for the class imbalance problem in the malaria diagnosis context.

Authors

  • Aimon Rahman
    Department of Electrical & Computer Engineering, North South University, Bashundhara, Dhaka 1229, Bangladesh. Electronic address: aimon.rahman@northsouth.edu.
  • Hasib Zunair
    Concordia University Montreal QC Canada.
  • Tamanna Rahman Reme
    Department of Electrical & Computer Engineering, North South University, Bashundhara, Dhaka 1229, Bangladesh. Electronic address: rahman.reme@northsouth.edu.
  • M Sohel Rahman
    Department of CSE, BUET, ECE Building, West Palasi, Dhaka 1205, Bangladesh. Electronic address: msrahman@cse.buet.ac.bd.
  • M R C Mahdy
    Department of Electrical and Computer Engineering, North South University, Bashundhara, Dhaka 1229, Bangladesh; Pi Labs Bangladesh LTD, ARA Bhaban,39, Kazi Nazrul Islam Avenue, Kawran Bazar, Dhaka 1215, Bangladesh.