SODA: Detecting COVID-19 in Chest X-Rays With Semi-Supervised Open Set Domain Adaptation

Jieli Zhou, Baoyu Jing, Zeya Wang, Hongyi Xin, Hanghang Tong

Research output: Contribution to journalArticlepeer-review

15 Scopus citations

Abstract

Due to the shortage of COVID-19 viral testing kits, radiology imaging is used to complement the screening process. Deep learning based methods are promising in automatically detecting COVID-19 disease in chest X-ray images. Most of these works first train a Convolutional Neural Network (CNN) on an existing large-scale chest X-ray image dataset and then fine-tune the model on the newly collected COVID-19 chest X-ray dataset, often at a much smaller scale. However, simple fine-tuning may lead to poor performance for the CNN model due to two issues, first the large domain shift present in chest X-ray datasets and second the relatively small scale of the COVID-19 chest X-ray dataset. In an attempt to address these two important issues, we formulate the problem of COVID-19 chest X-ray image classification in a semi-supervised open set domain adaptation setting and propose a novel domain adaptation method, Semi-supervised Open set Domain Adversarial network (SODA). SODA is designed to align the data distributions across different domains in the general domain space and also in the common subspace of source and target data. In our experiments, SODA achieves a leading classification performance compared with recent state-of-the-art models in separating COVID-19 with common pneumonia. We also present initial results showing that SODA can produce better pathology localizations in the chest X-rays.

Original languageEnglish
Pages (from-to)2605-2612
Number of pages8
JournalIEEE/ACM Transactions on Computational Biology and Bioinformatics
Volume19
Issue number5
DOIs
StatePublished - 2022

Bibliographical note

Publisher Copyright:
© 2004-2012 IEEE.

Keywords

  • COVID-19
  • domain adaptation
  • medical image analysis
  • open set domain adaptation
  • semi-supervised learning

ASJC Scopus subject areas

  • Biotechnology
  • Genetics
  • Applied Mathematics

Fingerprint

Dive into the research topics of 'SODA: Detecting COVID-19 in Chest X-Rays With Semi-Supervised Open Set Domain Adaptation'. Together they form a unique fingerprint.

Cite this