Deep ensemble approach for pathogen classification in large-scale images using patch-based training and hyper-parameter optimization

BMC Bioinformatics

Research

Muhammad Usman Ghani Khan¹  Fareed Ahmad²  Ahsen Tahir³  Farhan Masud⁴ 

[1] Department of Computer Science, University of Engineering and Technology, G.T. Road, 54890, Lahore, Punjab, Pakistan;National Center of Artificial Intelligence, Al-Khawarizmi Institute of Computer Science, UET, Lahore, Pakistan;Department of Computer Science, University of Engineering and Technology, G.T. Road, 54890, Lahore, Punjab, Pakistan;Quality Operations Laboratory, Institute of Microbiology, University of Veterinary and Animal Sciences, Outfall road, 54000, Lahore, Punjab, Pakistan;Department of Electrical Engineering, University of Engineering and Technology, G.T. road, 54890, Lahore, Punjab, Pakistan;Department of Statistics and Computer Science, Faculty of Life Sciences Business Management, University of Veterinary and Animal Sciences, Outfall Road, 54000, Lahore, Punjab, Pakistan;

关键词: Pathogen classification;    Deep learning models;    Ensemble learning;    Image patching;    Feature fusion;    Tuning hyper-parameter;   

DOI  :  10.1186/s12859-023-05398-7

 received in 2023-03-09, accepted in 2023-06-23,  发布年份 2023

来源: Springer

【 摘 要 】

Pathogenic bacteria present a major threat to human health, causing various infections and illnesses, and in some cases, even death. The accurate identification of these bacteria is crucial, but it can be challenging due to the similarities between different species and genera. This is where automated classification using convolutional neural network (CNN) models can help, as it can provide more accurate, authentic, and standardized results.In this study, we aimed to create a larger and balanced dataset by image patching and applied different variations of CNN models, including training from scratch, fine-tuning, and weight adjustment, and data augmentation through random rotation, reflection, and translation. The results showed that the best results were achieved through augmentation and fine-tuning of deep models. We also modified existing architectures, such as InceptionV3 and MobileNetV2, to better capture complex features. The robustness of the proposed ensemble model was evaluated using two data splits (7:2:1 and 6:2:2) to see how performance changed as the training data was increased from 10 to 20%. In both cases, the model exhibited exceptional performance. For the 7:2:1 split, the model achieved an accuracy of 99.91%, F-Score of 98.95%, precision of 98.98%, recall of 98.96%, and MCC of 98.92%. For the 6:2:2 split, the model yielded an accuracy of 99.94%, F-Score of 99.28%, precision of 99.31%, recall of 98.96%, and MCC of 99.26%. This demonstrates that automatic classification using the ensemble model can be a valuable tool for diagnostic staff and microbiologists in accurately identifying pathogenic bacteria, which in turn can help control epidemics and minimize their social and economic impact.

【 授权许可】

CC BY   
© The Author(s) 2023

【 图 表 】

Fig. 2

Fig. 5

Fig. 2

Fig. 3

Fig. 8

Fig. 10

Fig. 3

12936_2023_4510_Article_IEq8.gif

Fig. 3

Fig. 2

12936_2023_4510_Article_IEq4.gif

Fig. 4

12903_2023_3210_Article_IEq4.gif

Fig. 3

Fig. 1

Fig. 5

41239_2023_389_Article_IEq72.gif

Fig. 4

Fig. 5

41239_2023_389_Article_IEq68.gif

Fig. 3

Fig. 1

Fig. 8

Fig. 3

40691_2023_344_Figd_HTML.gif

Fig. 3

Fig. 4

Fig. 2

Fig. 9

Fig. 4

41239_2023_389_Article_IEq56.gif

Fig. 8

Fig. 2

Fig. 1

Fig. 3

Fig. 5

Fig. 4

Fig. 2

Fig. 2

Fig. 4

Fig. 1

Fig. 3

Fig. 5

Fig. 1

Fig. 2

Fig. 3

Fig. 1

Fig. 3

Fig. 1

Fig. 3

Fig. 4

41239_2023_389_Article_IEq13.gif

Fig. 1

Fig. 1

Fig. 3

Fig. 2

41239_2023_389_Article_IEq8.gif

Fig. 2

Fig. 2

Fig. 1

40708_2023_197_Article_IEq87.gif

40708_2023_197_Article_IEq86.gif

40708_2023_197_Article_IEq84.gif

Fig. 1

40708_2023_197_Article_IEq81.gif

40708_2023_197_Article_IEq83.gif

40708_2023_197_Article_IEq82.gif

Fig. 4

40708_2023_197_Article_IEq79.gif

40708_2023_197_Article_IEq75.gif

Fig. 2

40708_2023_197_Article_IEq55.gif

40708_2023_197_Article_IEq51.gif

40798_2023_598_Article_IEq27.gif

40798_2023_598_Article_IEq8.gif