University of Leicester
Browse
paper_v36_Clean.pdf (1.92 MB)

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

Download (1.92 MB)
journal contribution
posted on 2021-11-04, 10:40 authored by J Sun, X Li, C Tang, SH Wang, YD Zhang
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.

History

Citation

Knowledge-Based Systems Volume 232, 28 November 2021, 107494

Author affiliation

School of Informatics

Version

  • AM (Accepted Manuscript)

Published in

Knowledge-Based Systems

Volume

232

Pagination

107494

Publisher

Elsevier BV

issn

0950-7051

Acceptance date

2021-09-11

Copyright date

2021

Available date

2022-09-15

Language

en

Usage metrics

    University of Leicester Publications

    Categories

    No categories selected

    Exports

    RefWorks
    BibTeX
    Ref. manager
    Endnote
    DataCite
    NLM
    DC