English

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

Image and Video Processing 2021-05-27 v2 Computer Vision and Pattern Recognition

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.

Keywords

Cite

@article{arxiv.2009.11850,
  title  = {ECOVNet: An Ensemble of Deep Convolutional Neural Networks Based on EfficientNet to Detect COVID-19 From Chest X-rays},
  author = {Nihad Karim Chowdhury and Muhammad Ashad Kabir and Md. Muhtadir Rahman and Noortaz Rezoana},
  journal= {arXiv preprint arXiv:2009.11850},
  year   = {2021}
}