A comparative analysis of eleven neural networks architectures for small datasets of lung images of COVID-19 patients toward improved clinical decisions.

Journal: Computers in biology and medicine
Published Date:

Abstract

The 2019 novel severe acute respiratory syndrome coronavirus 2-SARS-CoV2, commonly known as COVID-19, is a highly infectious disease that has endangered the health of many people around the world. COVID-19, which infects the lungs, is often diagnosed and managed using X-ray or computed tomography (CT) images. For such images, rapid and accurate classification and diagnosis can be performed using deep learning methods that are trained using existing neural network models. However, at present, there is no standardized method or uniform evaluation metric for image classification, which makes it difficult to compare the strengths and weaknesses of different neural network models. This paper used eleven well-known convolutional neural networks, including VGG-16, ResNet-18, ResNet-50, DenseNet-121, DenseNet-169, Inception-v3, Inception-v4, SqueezeNet, MobileNet, ShuffeNet, and EfficientNet-b0, to classify and distinguish COVID-19 and non-COVID-19 lung images. These eleven models were applied to different batch sizes and epoch cases, and their overall performance was compared and discussed. The results of this study can provide decision support in guiding research on processing and analyzing small medical datasets to understand which model choices can yield better outcomes in lung image classification, diagnosis, disease management and patient care.

Authors

  • Yuan Yang
    The Ministry of Education Key Laboratory of Contemporary Design and Integrated Manufacturing Technology, Northwestern Polytechnical University, No. 127, Youyi Road (West), Xi'an 710072, China.
  • Lin Zhang
    Laboratory of Molecular Translational Medicine, Centre for Translational Medicine, Key Laboratory of Birth Defects and Related Diseases of Women and Children, Ministry of Education, Clinical Research Center for Birth Defects of Sichuan Province, West China Second Hospital, Sichuan University, Chengdu, Sichuan, 610041, China. Electronic address: zhanglin@scu.edu.cn.
  • Mingyu Du
    Beijing Advanced Innovation Center for Big Data-Based Precision Medicine, School of Medicine and Engineering, No.37 Xueyuan Road, Haidian District, Beijing, China; Key Laboratory of Big Data-Based Precision Medicine, Ministry of Industry and Information Technology, No.37 Xueyuan Road, Haidian District, Beijing, China; School of Automation Science and Electrical Engineering, Beihang University, No.37 Xueyuan Road, Haidian District, Beijing, China.
  • Jingyu Bo
    School of Economics and Management, Beijing Jiaotong University, No.3, Shangyuan Village, Haidian District, Beijing, China.
  • Haolei Liu
    Beijing Advanced Innovation Center for Big Data-Based Precision Medicine, School of Medicine and Engineering, No.37 Xueyuan Road, Haidian District, Beijing, China; Key Laboratory of Big Data-Based Precision Medicine, Ministry of Industry and Information Technology, No.37 Xueyuan Road, Haidian District, Beijing, China; School of Automation Science and Electrical Engineering, Beihang University, No.37 Xueyuan Road, Haidian District, Beijing, China.
  • Lei Ren
    Department of Biomaterials, College of Materials, Xiamen University, Xiamen 361005, P.R. China.
  • Xiaohe Li
    The Third People's Hospital of Shenzhen, Shenzhen, China.
  • M Jamal Deen
    Department of Electrical and Computer Engineering and the School of Biomedical Engineering, McMaster University, Hamilton, Ontario, L8S 4K1, Canada. Electronic address: jamal@mcmaster.ca.