International Journal of Engineering and Manufacturing (IJEM)
IJEM Vol. 14, No. 6, 8 Dec. 2024
Cover page and Table of Contents: PDF (size: 1050KB)
Dilated Convolutional Neural Network with Attention Mechanism for Classification of Malaria Parasites
PDF (1050KB), PP.11-26
Views: 0 Downloads: 0
Author(s)
Suleiman Garba
1,*
Muhammad Bashir Abdullahi
1
Sulaimon Adebayo Bashir
1
Abisoye Opeyemi Aderike
1
Index Terms
Attention Mechanism, Batch Size, Classification, Dilated CNN, Malaria, Parasite
Abstract
Malaria remains a pervasive global health challenge, affecting millions of lives daily. Traditional diagnostic methods, involving manual blood smear examination, are time-consuming and prone to errors, especially in large-scale testing. Although promising, automated detection techniques often fail to capture the intricate spatial features of malaria parasites leading to inconsistent performance. In order to close these gaps, this work suggest an improved technique that combines a Self-Attention Mechanism and a Dilated Convolutional Neural Network (D-CNN) to allow the model to effectively and precisely classify malaria parasites as infected or uninfected. Both local and global spatial information are captured by dilated convolutions, and crucial features are given priority by the attention mechanism for accurate detection in complex images. We also examine batch size variation and find that it plays a crucial role in maximizing generalization, accuracy, and resource efficiency. A batch size of 64 produced superior results after testing six different sizes, yielding an AUC of 99.12%, F1-Score of 96, precision of 97.63%, recall of 93.99%, and accuracy of 96.08%. This batch size balances efficient gradient updates and stabilization, reducing overfitting and improving generalization, especially on complex medical datasets. Our approach was benchmarked against existing competitors using the same publicly available malaria dataset, demonstrating a 2-3% improvement in AUC and precision over state-of-the-art models, such as traditional CNNs and machine learning methods. This highlights its superior ability to minimize false positives and negatives, particularly in complex diagnostic cases. These advancements enhance the reliability of large-scale diagnostic systems, improve clinical decision-making, and address key challenges in automated malaria detection.
Cite This Paper
Suleiman Garba, Muhammad Bashir Abdullahi, Sulaimon Adebayo Bashir, Abisoye Opeyemi Aderike, "Dilated Convolutional Neural Network with Attention Mechanism for Classification of Malaria Parasites", International Journal of Engineering and Manufacturing (IJEM), Vol.14, No.6, pp. 11-26, 2024. DOI:10.5815/ijem.2024.06.02
Reference
[1]WHO (World Health Organization). Malaria microscopy quality assurance manual – Version. 2. WHO, page 140, 2016.
[2]Acevedo, A., Alférez, S., Merino, A., Puigví, L., & Rodellar, J. (2019). Recognition of peripheral blood cell images using convolutional neural networks. Computer methods and programs in biomedicine, 180, 105020.
[3]Rosado, L., Da Costa, J. M. C., Elias, D., & Cardoso, J. S. (2017). Mobile-based analysis of malaria-infected thin blood smears: automated species and life cycle stage determination. Sensors, 17(10), 2167.
[4]Bain, B. J. (2005). Diagnosis from the blood smear. New England Journal of Medicine, 353(5), 498-507.
[5]He, K., Zhang, X., Ren, S., & Sun, J. (2016). Deep residual learning for image recognition. In Proceedings of the IEEE conference on computer vision and pattern recognition (pp. 770-778).
[6]Baker, N., Lu, H., Erlikhman, G., & Kellman, P. J. (2020). Local features and global shape information in object classification by deep convolutional neural networks. Vision research, 172, 46-61.
[7]Liu, Y. H. (2018, September). Feature extraction and image recognition with convolutional neural networks. In Journal of Physics: Conference Series (Vol. 1087, p. 062032). IOP Publishing.
[8]Purnama, I. K. E., Rahmanti, F. Z., & Purnomo, M. H. (2013, November). Malaria parasite identification on thick blood film using genetic programming. In 2013 3rd International Conference on Instrumentation, Communications, Information Technology and Biomedical Engineering (ICICI-BME) (pp. 194-198). IEEE.
[9]Bibin, D., Nair, M. S., & Punitha, P. (2017). Malaria parasite detection from peripheral blood smear images using deep belief networks. IEEE Access, 5, 9099-9108.
[10]Rajaraman, S., Antani, S. K., Poostchi, M., Silamut, K., Hossain, M. A., Maude, R. J., ... & Thoma, G. R. (2018). Pre-trained convolutional neural networks as feature extractors toward improved malaria parasite detection in thin blood smear images. PeerJ, 6, e4568.
[11]Sriporn, K., Tsai, C. F., Tsai, C. E., & Wang, P. (2020). Analyzing malaria disease using effective deep learning approach. Diagnostics, 10(10), 744.
[12]Goni, M. O. F., Mondal, M. N. I., Islam, S. R., Nahiduzzaman, M., Islam, M. R., Anower, M. S., & Kwak, K. S. (2023). Diagnosis of Malaria Using Double Hidden Layer Extreme Learning Machine Algorithm With CNN Feature Extraction and Parasite Inflator. IEEE Access, 11, 4117-4130.
[13]Khan, A., Gupta, K. D., Venugopal, D., & Kumar, N. (2020, July). Cidmp: Completely interpretable detection of malaria parasite in red blood cells using lower-dimensional feature space. In 2020 International Joint Conference on Neural Networks (IJCNN) (pp. 1-8). IEEE.
[14]Fuhad, K. F., Tuba, J. F., Sarker, M. R. A., Momen, S., Mohammed, N., & Rahman, T. (2020). Deep learning-based automatic malaria parasite detection from blood smear and its smartphone-based application. Diagnostics, 10(5), 329.
[15]Islam, M. R., Nahiduzzaman, M., Goni, M. O. F., Sayeed, A., Anower, M. S., Ahsan, M., & Haider, J. (2022). Explainable transformer-based deep learning model for the detection of malaria parasites from blood cell images. Sensors, 22(12), 4358.
[16]Mohanty, I., Pattanaik, P. A., & Swarnkar, T. (2019). Automatic detection of malaria parasites using unsupervised techniques. In Proceedings of the International Conference on ISMAC in Computational Vision and Bio-Engineering 2018 (ISMAC-CVB) (pp. 41-49). Springer International Publishing.
[17]Dong, Y., Jiang, Z., Shen, H., Pan, W. D., Williams, L. A., Reddy, V. V., ... & Bryan, A. W. (2017, February). Evaluations of deep convolutional neural networks for automatic identification of malaria infected cells. In 2017 IEEE EMBS international conference on biomedical & health informatics (BHI) (pp. 101-104). IEEE..
[18]Roy, K., Sharmin, S., Mukta, R. M., & Sen, A. (2018). Detection of malaria parasite in Giemsa blood sample using image processing. International Journal of Computer Science and Information Technology, 10(1), 55-65.
[19] Li, D., & Ma, Z. (2022). Residual attention learning network and SVM for malaria parasite detection. Multimedia Tools and Applications, 81(8), 10935-10960.
[20]Manning, K., Zhai, X., & Yu, W. (2022). Image analysis and machine learning-based malaria assessment system. Digital Communications and Networks, 8(2), 132-142.
[21]Reddy, A. S. B., & Juliet, D. S. (2019, April). Transfer learning with ResNet-50 for malaria cell-image classification. In 2019 International Conference on Communication and Signal Processing (ICCSP) (pp. 0945-0949). IEEE.
[22]Krishnadas, P., & Sampathila, N. (2021, July). Automated detection of malaria implemented by deep learning in PyTorch. In 2021 IEEE International Conference on Electronics, Computing and Communication Technologies (CONECCT) (pp. 01-05). IEEE.
[23]Taha, B., & Liza, F. R. (2021, December). Automatic identification of malaria-infected cells using deep convolutional neural network. In 2021 24th International Conference on Computer and Information Technology (ICCIT) (pp. 1-5). IEEE..
[24]Maqsood, A., Farid, M. S., Khan, M. H., & Grzegorzek, M. (2021). Deep malaria parasite detection in thin blood smear microscopic images. Applied Sciences, 11(5), 2284.
[25]Shal, A., & Gupta, R. (2022, January). A comparative study on malaria cell detection using computer vision. In 2022 12th International Conference on Cloud Computing, Data Science & Engineering (Confluence) (pp. 548-552). IEEE.
[26]Gummadi, S. D., Ghosh, A., & Vootla, Y. (2022, June). Transfer Learning based Classification of Plasmodium Falciparum Parasitic Blood Smear Images. In 2022 10th International Symposium on Digital Forensics and Security (ISDFS) (pp. 1-5). IEEE.
[27]Qayyum, A. B. A., Islam, T., & Haque, M. A. (2019, November). Malaria diagnosis with dilated convolutional neural network based image analysis. In 2019 IEEE International Conference on Biomedical Engineering, Computer and Information Technology for Health (BECITHCON) (pp. 68-72). IEEE.
[28]Alharbi, A. H., Aravinda, C. V., Shetty, J., Jabarulla, M. Y., Sudeepa, K. B., & Singh, S. K. (2022). Computational models-based detection of peripheral malarial parasites in blood smears. Contrast Media & Molecular Imaging, 2022.
[29]Quan, Q., Wang, J., & Liu, L. (2020). An effective convolutional neural network for classifying red blood cells in malaria diseases. Interdisciplinary Sciences: Computational Life Sciences, 12, 217-225.
[30]Fu, M., Wu, K., Li, Y., Luo, L., Huang, W., & Zhang, Q. (2023). An intelligent detection method for plasmodium based on self-supervised learning and attention mechanism. Frontiers in Medicine, 10.