Transfer learning approach to Classify the X-ray image that corresponds to corona disease Using ResNet50 pretrained by ChexNet
This work addresses the need for fast and accurate diagnosis of COVID-19 to save patients and prevent spread, but it is incremental as it applies existing transfer learning methods to a new dataset.
The authors tackled the problem of diagnosing COVID-19 from chest X-ray images by training a ResNet50 model pretrained on ImageNet and CheXNet, using the imbalanced CoronaHack dataset for binary and multi-class classification, and comparing Focal loss with Cross entropy loss, achieving results that help classify COVID-19 cases.
Coronavirus adversely has affected people worldwide. There are common symptoms between the Covid19 virus disease and other respiratory diseases like pneumonia or Influenza. Therefore, diagnosing it fast is crucial not only to save patients but also to prevent it from spreading. One of the most reliant methods of diagnosis is through X-ray images of a lung. With the help of deep learning approaches, we can teach the deep model to learn the condition of an affected lung. Therefore, it can classify the new sample as if it is a Covid19 infected patient or not. In this project, we train a deep model based on ResNet50 pretrained by ImageNet dataset and CheXNet dataset. Based on the imbalanced CoronaHack Chest X-Ray dataset introducing by Kaggle we applied both binary and multi-class classification. Also, we compare the results when using Focal loss and Cross entropy loss.