An Explainable Artificial Intelligence Integrated System for Automatic Detection of Dengue From Images of Blood Smears Using Transfer Learning

被引:5
作者
Mayrose, Hilda [1 ]
Sampathila, Niranjana [1 ]
Bairy, G. Muralidhar [1 ]
Nayak, Tushar [1 ]
Belurkar, Sushma [2 ]
Saravu, Kavitha [3 ]
机构
[1] Manipal Acad Higher Educ MAHE, Manipal Inst Technol, Dept Biomed Engn, Manipal 576104, India
[2] Manipal Acad Higher Educ MAHE, Kasturba Med Coll, Dept Pathol, Manipal 576104, India
[3] Manipal Acad Higher Educ MAHE, Kasturba Med Coll, Dept Infect Dis, Manipal 576104, India
关键词
Deep learning; dengue fever; digital pathology; explainable artificial intelligence; GradCAM; peripheral blood smear; pre-trained CNN; transfer learning;
D O I
10.1109/ACCESS.2024.3378516
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Dengue fever is a rapidly increasing mosquito-borne ailment spread by the virus DENV in the tropics and subtropics worldwide. It is a significant public health problem and accounts for many deaths globally. Implementing more effective methods that can more accurately detect dengue cases is challenging. The theme of this digital pathology-associated research is automatic dengue detection from peripheral blood smears (PBS) employing deep learning (DL) techniques. In recent years, DL has been significantly employed for automated computer-assisted diagnosis of various diseases from medical images. This paper explores pre-trained convolution neural networks (CNNs) for automatic dengue fever detection. Transfer learning (TL) is executed on three state-of-the-art CNNs - ResNet50, MobileNetV3Small, and MobileNetV3Large, to customize the models for differentiating the dengue-infected blood smears from the healthy ones. The dataset used to design and test the models contains 100x magnified dengue-infected and healthy control digital microscopic PBS images. The models are validated with a 5-fold cross-validation framework and tested on unseen data. An explainable artificial intelligence (XAI) approach, Gradient-weighted Class Activation Mapping (GradCAM), is eventually applied to the models to allow visualization of the precise regions on the smears most instrumental in making the predictions. While all three transferred pre-trained CNN models performed well (above 98% overall classification accuracy), MobileNetV3Small is the recommended model for this classification problem due to its significantly less computationally demanding characteristics. Transferred pre-trained CNN based on MobileNetV3Small yielded Accuracy, Recall, Specificity, Precision, F1 Score, and Area Under the ROC Curve (AUC) of 0.982 +/- 0.011, 0.973 +/- 0.027, 0.99 +/- 0.013, 0.989 +/- 0.015, 0.981 +/- 0.012 and 0.982 +/- 0.012 respectively, averaged over the five folds on the unseen dataset. Promising results show that the developed models have the potential to provide high-quality support to haematologists by expertly performing tedious, repetitive, and time-consuming tasks in hospitals and remote/low-resource settings.
引用
收藏
页码:41750 / 41762
页数:13
相关论文
共 33 条
[1]   Automated detection and classification of leukemia on a subject-independent test dataset using deep transfer learning supported by Grad-CAM visualization [J].
Abhishek, Arjun ;
Jha, Rajib Kumar ;
Sinha, Ruchi ;
Jha, Kamlesh .
BIOMEDICAL SIGNAL PROCESSING AND CONTROL, 2023, 83
[2]   Development of screen- printed carbon electrode-based immunosensors for the electrochemical detection of dengue virus antigen [J].
Alhazmi, Hassan A. ;
Ahsan, Waquar ;
Taha, Manal M. E. ;
Albratty, Mohammed ;
Najmi, Asim ;
Farasani, Abdullah ;
Abdulhaq, Ahmed A. ;
Darwish, Ibrahim A. .
JOURNAL OF KING SAUD UNIVERSITY SCIENCE, 2023, 35 (03)
[3]   Improved Classification of White Blood Cells with the Generative Adversarial Network and Deep Convolutional Neural Network [J].
Almezhghwi, Khaled ;
Serte, Sertan .
COMPUTATIONAL INTELLIGENCE AND NEUROSCIENCE, 2020, 2020
[4]   Review of deep learning: concepts, CNN architectures, challenges, applications, future directions [J].
Alzubaidi, Laith ;
Zhang, Jinglan ;
Humaidi, Amjad J. ;
Al-Dujaili, Ayad ;
Duan, Ye ;
Al-Shamma, Omran ;
Santamaria, J. ;
Fadhel, Mohammed A. ;
Al-Amidie, Muthana ;
Farhan, Laith .
JOURNAL OF BIG DATA, 2021, 8 (01)
[5]  
[Anonymous], 2023, Anal. J., V7, DOI [10.1016/j.dajour.2023.100230, DOI 10.1016/J.DAJOUR.2023.100230]
[6]  
[Anonymous], 2013, Dengue
[7]   Numerical Grad-Cam Based Explainable Convolutional Neural Network for Brain Tumor Diagnosis [J].
Antonio Marmolejo-Saucedo, Jose ;
Kose, Utku .
MOBILE NETWORKS & APPLICATIONS, 2024, 29 (01) :109-118
[8]  
Aziz S, 2020, 2020 INT C EMERGING, P1, DOI DOI 10.1109/ICETST49965.2020.9080727
[9]   Physical interpretation of machine learning-based recognition of defects for the risk management of existing bridge heritage [J].
Cardellicchio, Angelo ;
Ruggieri, Sergio ;
Nettis, Andrea ;
Reno, Vito ;
Uva, Giuseppina .
ENGINEERING FAILURE ANALYSIS, 2023, 149
[10]   A hybrid approach for efficient multi-classification of white blood cells based on transfer learning techniques and traditional machine learning methods [J].
Cengil, Emine ;
Cinar, Ahmet ;
Yildirim, Muhammed .
CONCURRENCY AND COMPUTATION-PRACTICE & EXPERIENCE, 2022, 34 (06)