Weighted Cross-Entropy for Unbalanced Data with Application on COVID X-ray images

25Citations
Citations of this article
20Readers
Mendeley users who have this article in their library.

This article is free to access.

Abstract

Since December 2019 the world is infected by COVID-19 or Coronavirus disease, which spreads very quickly, out of control. The high number of precautions for laboratory access, which need to be taken to contain the virus, together with the difficulties in running the gold standard test for COVID-19, result in a practical incapability to make early diagnosis. Recent advances in deep learning algorithms allow efficient implementation of computer-aided diagnosis. This paper investigates on the performance of a very well known residual network, ResNet50, and a lightweight Atrous CNN (ACNN) network using a Weighted Cross-entropy (WCE) loss function, to alleviate imbalance on COVID datasets. As a result, ResNet50 model initialized with pre-trained weights fine-tuned by ImageNet dataset and exploiting WCE achieved the state-of-the-art performance on COVIDXRay-5K test set, with a top balanced accuracy of 99.87%.

Cite

CITATION STYLE

APA

Ozdemir, O., & Sonmez, E. B. (2020). Weighted Cross-Entropy for Unbalanced Data with Application on COVID X-ray images. In Proceedings - 2020 Innovations in Intelligent Systems and Applications Conference, ASYU 2020. Institute of Electrical and Electronics Engineers Inc. https://doi.org/10.1109/ASYU50717.2020.9259848

Register to see more suggestions

Mendeley helps you to discover research relevant for your work.

Already have an account?

Save time finding and organizing research with Mendeley

Sign up for free