GSA-DenseNet121-COVID-19: a Hybrid Deep Learning Architecture for the Diagnosis of COVID-19 Disease based on Gravitational Search Optimization Algorithm

Abstract

In this paper, a novel approach called GSA-DenseNet121-COVID-19 based on a hybrid convolutional neural network (CNN) architecture is proposed using an optimization algorithm. The CNN architecture that was used is called DenseNet121 and the optimization algorithm that was used is called the gravitational search algorithm (GSA). The GSA is adapted to determine the best values for the hyperparameters of the DenseNet121 architecture, and to achieve a high level of accuracy in diagnosing COVID-19 disease through chest x-ray image analysis. The obtained results showed that the proposed approach was able to correctly classify 98% of the test set. To test the efficacy of the GSA in setting the optimum values for the hyperparameters of DenseNet121, it was compared to another optimization algorithm called social ski driver (SSD). The comparison results demonstrated the efficacy of the proposed GSA-DenseNet121-COVID-19 and its ability to better diagnose COVID-19 disease than the SSD-DenseNet121 as the second was able to diagnose only 94% of the test set. As well as, the proposed approach was compared to an approach based on a CNN architecture called Inception-v3 and the manual search method for determining the values of the hyperparameters. The results of the comparison showed that the GSA-DenseNet121 was able to beat the other approach, as the second was able to classify only 95% of the test set samples.

arXiv comment29 pages - 7 Figures - 5 Tables

References in corpus (9)

• Adam: A Method for Stochastic Optimization
• Covid-19: Automatic detection from X-Ray images utilizing Transfer Learning with Convolutional Neural Networks
• CoroNet: A deep neural network for detection and diagnosis of COVID-19 from chest x-ray images
• Activation Functions: Comparison of trends in Practice and Research for Deep Learning
• Rapid identification of pathogenic bacteria using Raman spectroscopy and deep learning
• COVID-19 Image Data Collection
• CovidGAN: Data Augmentation Using Auxiliary Classifier GAN for Improved Covid-19 Detection
• The Effects of Hyperparameters on SGD Training of Neural Networks
• Effective time variation of G in a model universe with variable space dimension

Cited by in corpus (13)

• Common pitfalls and recommendations for using machine learning to detect and prognosticate for COVID-19 using chest radiographs and CT scans
• A Review on Deep Learning Techniques for the Diagnosis of Novel Coronavirus (COVID-19)
• Real-Time COVID-19 Diagnosis from X-Ray Images Using Deep CNN and Extreme Learning Machines Stabilized by Chimp Optimization Algorithm
• Hybrid quantum convolutional neural networks model for COVID-19 prediction using chest X-Ray images
• Automated Detection and Forecasting of COVID-19 using Deep Learning Techniques: A Review
• Medical Imaging and Computational Image Analysis in COVID-19 Diagnosis: A Review
• Interpretable CNN-Multilevel Attention Transformer for Rapid Recognition of Pneumonia from Chest X-Ray Images
• Performance of GAN-based augmentation for deep learning COVID-19 image classification
• Exploring the Effect of Image Enhancement Techniques on COVID-19 Detection using Chest X-rays Images
• Additional Look into GAN-based Augmentation for Deep Learning COVID-19 Image Classification
• Probabilistic combination of eigenlungs-based classifiers for COVID-19 diagnosis in chest CT images
• Tiled sparse coding in eigenspaces for the COVID-19 diagnosis in chest X-ray images
• A Survey of Machine Learning Techniques for Detecting and Diagnosing COVID-19 from Imaging