Enhancing malaria detection and classification using convolutional neural networks-vision transformer architecture
Abstract Malaria remains a significant global health challenge, particularly in sub-Saharan Africa. Despite advancements in treatment and prevention, malaria continues to cause substantial morbidity and mortality, particularly among vulnerable populations such as children and pregnant women. Although effective, traditional diagnostic methods, such as microscopy, are time-consuming and require skilled personnel prone to human error, leading to delays in diagnosis and treatment. More so, existing machine learning models used in malaria detection and classification have low performance and overfitting issues. This study presents an enhanced malaria detection and classification model using an ensemble of Convolutional Neural Networks (CNN) and Vision Transformers (ViT). The proposed ensemble model, which combines CNN and ViT, outperforms each individual model, achieving an accuracy of 99.64%, precision of 99.23%, recall of 99.75%, F1 score of 99.51%, and a cross-entropy loss of 0.01. The proposed model demonstrated superior performance compared to those reported in the literature. These results highlight the potential of the CNN-ViT ensemble model for accurate and reliable malaria detection, offering a significant improvement over existing methods.
Authors
- Danison Taremwa
- Fredrick Kanobe (ORCID: https://orcid.org/0000-0002-6418-5361)
- Leonard Nkalubo
- Emmanuel Ahishakiye (ORCID: https://orcid.org/0000-0001-6598-2111)
- Bartha Alexandra Nantongo
- Shallon Ahimbisibwe
Institutions
- Kyambogo University (UG)
Publication Details
- Journal
- Discover Applied Sciences
- Published
- 2025-06-06
- DOI
- https://doi.org/10.1007/s42452-025-06704-z
- Citations
- 4
- Primary Topic
- Digital Imaging for Blood Diseases
- Type
- article
- Field-Weighted Citation Impact
- 5.10