Metaheuristics optimization-based ensemble of deep neural networks for Mpox disease detection

被引:24
作者
Asif, Sohaib [1 ]
Zhao, Ming [1 ]
Tang, Fengxiao [1 ]
Zhu, Yusen [2 ]
Zhao, Baokang [3 ]
机构
[1] Cent South Univ, Sch Comp Sci & Engn, Changsha, Peoples R China
[2] Hunan Univ, Sch Math, Changsha, Peoples R China
[3] Natl Univ Def Technol, Sch Comp Sci, Changsha, Peoples R China
基金
中国国家自然科学基金;
关键词
Ensemble learning; Metaheuristic optimization; Mpox disease; Deep learning; Disease diagnosis;
D O I
10.1016/j.neunet.2023.08.035
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
The rising number of cases of human Mpox has emerged as a major global concern due to the daily increase of cases in several countries. The disease presents various skin symptoms in infected individuals, making it crucial to promptly identify and isolate them to prevent widespread community transmission. Rapid determination and isolation of infected individuals are therefore essential to curb the spread of the disease. Most research in the detection of Mpox disease has utilized convolutional neural network (CNN) models and ensemble methods. However, to the best of our knowledge, none have utilized a meta-heuristic-based ensemble approach. To address this gap, we propose a novel metaheuristics optimization-based weighted average ensemble model (MO-WAE) for detecting Mpox disease. We first train three transfer learning (TL)-based CNNs (DenseNet201, MobileNet, and DenseNet169) by adding additional layers to improve their classification strength. Next, we use a weighted average ensemble technique to fuse the predictions from each individual model, and the particle swarm optimization (PSO) algorithm is utilized to assign optimized weights to each model during the ensembling process. By using this approach, we obtain more accurate predictions than individual models. To gain a better understanding of the regions indicating the onset of Mpox, we performed a Gradient Class Activation Mapping (Grad-CAM) analysis to explain our model's predictions. Our proposed MO-WAE ensemble model was evaluated on a publicly available Mpox dataset and achieved an impressive accuracy of 97.78%. This outperforms state-of-the-art (SOTA) methods on the same dataset, thereby providing further evidence of the efficacy of our proposed model.
引用
收藏
页码:342 / 359
页数:18
相关论文
共 57 条
[31]  
Kundu D., 2022, 2022 25 INT C COMP I
[32]   Deep Learning at Chest Radiography: Automated Classification of Pulmonary Tuberculosis by Using Convolutional Neural Networks [J].
Lakhani, Paras ;
Sundaram, Baskaran .
RADIOLOGY, 2017, 284 (02) :574-582
[33]   Extreme learning machine ensemble model for time series forecasting boosted by PSO: Application to an electric consumption problem [J].
Larrea, Mikel ;
Porto, Alain ;
Irigoyen, Eloy ;
Javier Barragan, Antonio ;
Manuel Andujar, Jose .
NEUROCOMPUTING, 2021, 452 :465-472
[34]  
Lewin S, 2010, PRINCIPLES OF GENDER-SPECIFIC MEDICINE, 2ND EDITION, P497, DOI 10.1016/B978-0-12-374271-1.00045-9
[35]  
Liashchynskyi P, 2019, Arxiv, DOI [arXiv:1912.06059, 10.48550/arXiv.1912.06059]
[36]   Tuning PID and PIλDδ Controllers Using Particle Swarm Optimization Algorithm via El-Khazali's Approach [J].
Momani, Shaher ;
El-Khazali, Reyad ;
Batiha, Iqbal M. .
PROCEEDINGS OF THE 45TH INTERNATIONAL CONFERENCE ON APPLICATION OF MATHEMATICS IN ENGINEERING AND ECONOMICS (AMEE'19), 2019, 2172
[37]   A PSO-based weighting method for linear combination of neural networks [J].
Nabavi-Kerizi, S. H. ;
Abadi, M. ;
Kabir, E. .
COMPUTERS & ELECTRICAL ENGINEERING, 2010, 36 (05) :886-894
[38]  
Ali SN, 2022, Arxiv, DOI arXiv:2207.03342
[39]  
NASER M.Z., 2021, Architecture, Structures and Construction
[40]  
Polikar R, 2012, ENSEMBLE MACHINE LEARNING: METHODS AND APPLICATIONS, P1, DOI 10.1007/978-1-4419-9326-7_1