IDEAS home Printed from https://ideas.repec.org/a/gam/jmathe/v9y2021i4p434-d503826.html
   My bibliography  Save this article

Reliable Learning with PDE-Based CNNs and DenseNets for Detecting COVID-19, Pneumonia, and Tuberculosis from Chest X-Ray Images

Author

Listed:
  • Anca Nicoleta Marginean

    (Computer Science Department, Technical University of Cluj-Napoca, 400114 Cluj-Napoca, Romania)

  • Delia Doris Muntean

    (County Clinical Emergency Hospital of Cluj-Napoca, 400006 Cluj-Napoca, Romania
    Department of Radiology, Iuliu HaĊ£ieganu University of Medicine and Pharmacy, 400006 Cluj-Napoca, Romania)

  • George Adrian Muntean

    (County Clinical Emergency Hospital of Cluj-Napoca, 400006 Cluj-Napoca, Romania
    Department of Ophthalmology, Iuliu HaĊ£ieganu University of Medicine and Pharmacy, 400006 Cluj-Napoca, Romania)

  • Adelina Priscu

    (Department of Internal Medicine, Indiana University Health Ball Memorial Hospital, Muncie, IN 47303, USA)

  • Adrian Groza

    (Computer Science Department, Technical University of Cluj-Napoca, 400114 Cluj-Napoca, Romania)

  • Radu Razvan Slavescu

    (Computer Science Department, Technical University of Cluj-Napoca, 400114 Cluj-Napoca, Romania)

  • Calin Lucian Timbus

    (Department of Mathematics, Technical University of Cluj-Napoca, 400114 Cluj-Napoca, Romania)

  • Gabriel Zeno Munteanu

    (County Clinical Emergency Hospital of Cluj-Napoca, 400006 Cluj-Napoca, Romania)

  • Cezar Octavian Morosanu

    (University Hospital Bristol, Bristol BS2 8HW, UK)

  • Maria Margareta Cosnarovici

    (The Oncology Institute Prof. Dr. Ion Chiricuta, 400015 Cluj-Napoca, Romania)

  • Camelia-M. Pintea

    (Department of Mathematics and Informatics, Technical University of Cluj-Napoca, 400114 Cluj-Napoca, Romania)

Abstract

It has recently been shown that the interpretation by partial differential equations (PDEs) of a class of convolutional neural networks (CNNs) supports definition of architectures such as parabolic and hyperbolic networks. These networks have provable properties regarding the stability against the perturbations of the input features. Aiming for robustness, we tackle the problem of detecting changes in chest X-ray images that may be suggestive of COVID-19 with parabolic and hyperbolic CNNs and with domain-specific transfer learning. To this end, we compile public data on patients diagnosed with COVID-19, pneumonia, and tuberculosis, along with normal chest X-ray images. The negative impact of the small number of COVID-19 images is reduced by applying transfer learning in several ways. For the parabolic and hyperbolic networks, we pretrain the networks on normal and pneumonia images and further use the obtained weights as the initializers for the networks to discriminate between COVID-19, pneumonia, tuberculosis, and normal aspects. For DenseNets, we apply transfer learning twice. First, the ImageNet pretrained weights are used to train on the CheXpert dataset, which includes 14 common radiological observations (e.g., lung opacity, cardiomegaly, fracture, support devices). Then, the weights are used to initialize the network which detects COVID-19 and the three other classes. The resulting networks are compared in terms of how well they adapt to the small number of COVID-19 images. According to our quantitative and qualitative analysis, the resulting networks are more reliable compared to those obtained by direct training on the targeted dataset.

Suggested Citation

  • Anca Nicoleta Marginean & Delia Doris Muntean & George Adrian Muntean & Adelina Priscu & Adrian Groza & Radu Razvan Slavescu & Calin Lucian Timbus & Gabriel Zeno Munteanu & Cezar Octavian Morosanu & M, 2021. "Reliable Learning with PDE-Based CNNs and DenseNets for Detecting COVID-19, Pneumonia, and Tuberculosis from Chest X-Ray Images," Mathematics, MDPI, vol. 9(4), pages 1-20, February.
  • Handle: RePEc:gam:jmathe:v:9:y:2021:i:4:p:434-:d:503826
    as

    Download full text from publisher

    File URL: https://www.mdpi.com/2227-7390/9/4/434/pdf
    Download Restriction: no

    File URL: https://www.mdpi.com/2227-7390/9/4/434/
    Download Restriction: no
    ---><---

    Corrections

    All material on this site has been provided by the respective publishers and authors. You can help correct errors and omissions. When requesting a correction, please mention this item's handle: RePEc:gam:jmathe:v:9:y:2021:i:4:p:434-:d:503826. See general information about how to correct material in RePEc.

    If you have authored this item and are not yet registered with RePEc, we encourage you to do it here. This allows to link your profile to this item. It also allows you to accept potential citations to this item that we are uncertain about.

    We have no bibliographic references for this item. You can help adding them by using this form .

    If you know of missing items citing this one, you can help us creating those links by adding the relevant references in the same way as above, for each refering item. If you are a registered author of this item, you may also want to check the "citations" tab in your RePEc Author Service profile, as there may be some citations waiting for confirmation.

    For technical questions regarding this item, or to correct its authors, title, abstract, bibliographic or download information, contact: MDPI Indexing Manager (email available below). General contact details of provider: https://www.mdpi.com .

    Please note that corrections may take a couple of weeks to filter through the various RePEc services.

    IDEAS is a RePEc service. RePEc uses bibliographic data supplied by the respective publishers.