MFBCNNC: Momentum factor biogeography convolutional neural network for COVID-19 detection via chest X-ray images[Formula presented]

Junding Sun, Xiang Li, Chaosheng Tang*, Shui Hua Wang*, Yu Dong Zhang*

*Corresponding author for this work

Research output: Contribution to journalArticlepeer-review

10 Citations (Scopus)

Abstract

Aim: By October 6, 2020, Coronavirus disease 2019 (COVID-19) was diagnosed worldwide, reaching 3,355,7427 people and 1,037,862 deaths. Detection of COVID-19 and pneumonia by the chest X-ray images is of great significance to control the development of the epidemic situation. The current COVID-19 and pneumonia detection system may suffer from two shortcomings: the selection of hyperparameters in the models is not appropriate, and the generalization ability of the model is poor. Method: To solve the above problems, our team proposed an improved intelligent global optimization algorithm, which is based on the biogeography-based optimization to automatically optimize the hyperparameters value of the models according to different detection objectives. In the optimization progress, after selecting the immigration of suitable index vector and the emigration of suitable index vector, we proposed adding a comparison operation to compare the value of them. According to the different numerical relationships between them, the corresponding operations are performed to improve the migration operation of biogeography-based optimization. The improved algorithm (momentum factor biogeography-based optimization) can better perform the automatic optimization operation. In addition, our team also proposed two frameworks: biogeography convolutional neural network and momentum factor biogeography convolutional neural network. And two methods for detection COVID-19 based on the proposed frameworks. Results: Our method used three convolutional neural networks (LeNet-5, VGG-16, and ResNet-18) as the basic classification models for chest X-ray images detection of COVID-19, Normal, and Pneumonia. The accuracy of LeNet-5, VGG-16, and ResNet-18 is improved by 1.56%, 1.48%, and 0.73% after using biogeography-based optimization to optimize the hyperparameters of the models. The accuracy of LeNet-5, VGG-16, and ResNet-18 is improved by 2.87%, 6.31%, and 1.46% after using the momentum factor biogeography-based optimization to optimize the hyperparameters of the models. Conclusion: Under the same experimental conditions, the performance of the momentum factor biogeography-based optimization is superior to the biogeography-based optimization in optimizing the hyperparameters of the convolutional neural networks. Experimental results show that the momentum factor biogeography-based optimization can improve the detection performance of the state-of-the-art approaches in terms of overall accuracy. In future research, we will continue to use and improve other global optimization algorithms to enhance the application ability of deep learning in medical pathological image detection.

Original languageEnglish
Article number107494
JournalKnowledge-Based Systems
Volume232
DOIs
Publication statusPublished - 28 Nov 2021
Externally publishedYes

Keywords

  • Biogeography-based optimization
  • COVID-19
  • Convolutional neural network
  • Deep learning
  • Pneumonia

Fingerprint

Dive into the research topics of 'MFBCNNC: Momentum factor biogeography convolutional neural network for COVID-19 detection via chest X-ray images[Formula presented]'. Together they form a unique fingerprint.

Cite this