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

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.