ECOVNet: An Ensemble of Deep Convolutional Neural Networks Based on EfficientNet to Detect COVID-19 From Chest X-rays

Abstract

This paper proposed an ensemble of deep convolutional neural networks (CNN) based on EfficientNet, named ECOVNet, to detect COVID-19 using a large chest X-ray data set. At first, the open-access large chest X-ray collection is augmented, and then ImageNet pre-trained weights for EfficientNet is transferred with some customized fine-tuning top layers that are trained, followed by an ensemble of model snapshots to classify chest X-rays corresponding to COVID-19, normal, and pneumonia. The predictions of the model snapshots, which are created during a single training, are combined through two ensemble strategies, i.e., hard ensemble and soft ensemble to ameliorate classification performance and generalization in the related task of classifying chest X-rays.

References in corpus (16)

• Adam: A Method for Stochastic Optimization
• Very Deep Convolutional Networks for Large-Scale Image Recognition
• Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shift
• SqueezeNet: AlexNet-level accuracy with 50x fewer parameters and <0.5MB model size
• Grad-CAM++: Improved Visual Explanations for Deep Convolutional Networks
• Covid-19: Automatic detection from X-Ray images utilizing Transfer Learning with Convolutional Neural Networks
• Squeeze-and-Excitation Networks
• Can AI help in screening Viral and COVID-19 pneumonia?
• CoroNet: A deep neural network for detection and diagnosis of COVID-19 from chest x-ray images
• CheXNet: Radiologist-Level Pneumonia Detection on Chest X-Rays with Deep Learning
• COVID-19 Image Data Collection
• Searching for Activation Functions
• COVID-ResNet: A Deep Learning Framework for Screening of COVID19 from Radiographs
• PDCOVIDNet: A Parallel-Dilated Convolutional Neural Network Architecture for Detecting COVID-19 from Chest X-Ray Images
• Do Explanations Reflect Decisions? A Machine-centric Strategy to Quantify the Performance of Explainability Algorithms
• COVID-CXNet: Detecting COVID-19 in Frontal Chest X-ray Images using Deep Learning

Cited by in corpus (3)

• Cross-dimensional transfer learning in medical image segmentation with deep learning
• Exploring the Effect of Image Enhancement Techniques on COVID-19 Detection using Chest X-rays Images
• A Survey of Machine Learning Techniques for Detecting and Diagnosing COVID-19 from Imaging