Generating Realistic COVID19 X-rays with a Mean Teacher + Transfer
Learning GAN
- URL: http://arxiv.org/abs/2009.12478v1
- Date: Sat, 26 Sep 2020 00:05:06 GMT
- Title: Generating Realistic COVID19 X-rays with a Mean Teacher + Transfer
Learning GAN
- Authors: Sumeet Menon (1), Joshua Galita (1), David Chapman (1), Aryya
Gangopadhyay (1), Jayalakshmi Mangalagiri (1), Phuong Nguyen (1), Yaacov
Yesha (1), Yelena Yesha (1), Babak Saboury (1 and 2), Michael Morris (1, 2,
and 3) ((1) University of Maryland, Baltimore County, (2) National Institutes
of Health Clinical Center, (3) Networking Health)
- Abstract summary: We present a novel Mean Teacher + Transfer GAN (MTT-GAN) that generates COVID19 chest X-ray images of high quality.
In order to create a more accurate GAN, we employ transfer learning from the Kaggle Pneumonia X-Ray dataset.
Our analysis shows that the MTT-GAN generates X-ray images that are greatly superior to a baseline GAN and visually comparable to real X-rays.
- Score: 0.0
- License: http://arxiv.org/licenses/nonexclusive-distrib/1.0/
- Abstract: COVID-19 is a novel infectious disease responsible for over 800K deaths
worldwide as of August 2020. The need for rapid testing is a high priority and
alternative testing strategies including X-ray image classification are a
promising area of research. However, at present, public datasets for COVID19
x-ray images have low data volumes, making it challenging to develop accurate
image classifiers. Several recent papers have made use of Generative
Adversarial Networks (GANs) in order to increase the training data volumes. But
realistic synthetic COVID19 X-rays remain challenging to generate. We present a
novel Mean Teacher + Transfer GAN (MTT-GAN) that generates COVID19 chest X-ray
images of high quality. In order to create a more accurate GAN, we employ
transfer learning from the Kaggle Pneumonia X-Ray dataset, a highly relevant
data source orders of magnitude larger than public COVID19 datasets.
Furthermore, we employ the Mean Teacher algorithm as a constraint to improve
stability of training. Our qualitative analysis shows that the MTT-GAN
generates X-ray images that are greatly superior to a baseline GAN and visually
comparable to real X-rays. Although board-certified radiologists can
distinguish MTT-GAN fakes from real COVID19 X-rays. Quantitative analysis shows
that MTT-GAN greatly improves the accuracy of both a binary COVID19 classifier
as well as a multi-class Pneumonia classifier as compared to a baseline GAN.
Our classification accuracy is favourable as compared to recently reported
results in the literature for similar binary and multi-class COVID19 screening
tasks.
Related papers
- Performance of GAN-based augmentation for deep learning COVID-19 image
classification [57.1795052451257]
The biggest challenge in the application of deep learning to the medical domain is the availability of training data.
Data augmentation is a typical methodology used in machine learning when confronted with a limited data set.
In this work, a StyleGAN2-ADA model of Generative Adversarial Networks is trained on the limited COVID-19 chest X-ray image set.
arXiv Detail & Related papers (2023-04-18T15:39:58Z) - In-Line Image Transformations for Imbalanced, Multiclass Computer Vision
Classification of Lung Chest X-Rays [91.3755431537592]
This study aims to leverage a body of literature in order to apply image transformations that would serve to balance the lack of COVID-19 LCXR data.
Deep learning techniques such as convolutional neural networks (CNNs) are able to select features that distinguish between healthy and disease states.
This study utilizes a simple CNN architecture for high-performance multiclass LCXR classification at 94 percent accuracy.
arXiv Detail & Related papers (2021-04-06T02:01:43Z) - Many-to-One Distribution Learning and K-Nearest Neighbor Smoothing for
Thoracic Disease Identification [83.6017225363714]
deep learning has become the most powerful computer-aided diagnosis technology for improving disease identification performance.
For chest X-ray imaging, annotating large-scale data requires professional domain knowledge and is time-consuming.
In this paper, we propose many-to-one distribution learning (MODL) and K-nearest neighbor smoothing (KNNS) methods to improve a single model's disease identification performance.
arXiv Detail & Related papers (2021-02-26T02:29:30Z) - Twin Augmented Architectures for Robust Classification of COVID-19 Chest
X-Ray Images [6.127080932156285]
Gold standard for COVID-19 is RT-PCR, testing facilities for which are limited and not always optimally distributed.
We show that popular choices of dataset selection suffer from data homogeneity, leading to misleading results.
We introduce a state-of-the-art technique, termed as Twin Augmentation, for modifying popular pre-trained deep learning models.
arXiv Detail & Related papers (2021-02-16T06:50:17Z) - Chest X-ray Image Phase Features for Improved Diagnosis of COVID-19
Using Convolutional Neural Network [2.752817022620644]
Recent research has shown radiography of COVID-19 patient contains salient information about the COVID-19 virus.
Chest X-ray (CXR) due to its faster imaging time, wide availability, low cost and portability gains much attention.
In this study, we design a novel multi-feature convolutional neural network (CNN) architecture for improved classification of COVID-19 from CXR images.
arXiv Detail & Related papers (2020-11-06T20:26:26Z) - RANDGAN: Randomized Generative Adversarial Network for Detection of
COVID-19 in Chest X-ray [0.0]
COVID-19 spread across the globe at an immense rate has left healthcare systems incapacitated to diagnose and test patients at the needed rate.
Studies have shown promising results for detection of COVID-19 from viral bacterial pneumonia in chest X-rays.
In this study, we propose a randomized generative adversarial network (RANDGAN) that detects images of an unknown class (COVID-19) without the need for labels and training data.
arXiv Detail & Related papers (2020-10-06T15:58:09Z) - Classification of COVID-19 in CT Scans using Multi-Source Transfer
Learning [91.3755431537592]
We propose the use of Multi-Source Transfer Learning to improve upon traditional Transfer Learning for the classification of COVID-19 from CT scans.
With our multi-source fine-tuning approach, our models outperformed baseline models fine-tuned with ImageNet.
Our best performing model was able to achieve an accuracy of 0.893 and a Recall score of 0.897, outperforming its baseline Recall score by 9.3%.
arXiv Detail & Related papers (2020-09-22T11:53:06Z) - Fused Deep Convolutional Neural Network for Precision Diagnosis of
COVID-19 Using Chest X-Ray Images [0.0]
We propose a computer-aided diagnosis (CAD) to accurately classify chest X-ray scans of COVID-19 and normal subjects by fine-tuning several neural networks.
Using k-fold cross-validation and a bagging ensemble, we achieve an accuracy of 99.7% and a sensitivity of 100%.
arXiv Detail & Related papers (2020-09-15T02:27:20Z) - Learning Invariant Feature Representation to Improve Generalization
across Chest X-ray Datasets [55.06983249986729]
We show that a deep learning model performing well when tested on the same dataset as training data starts to perform poorly when it is tested on a dataset from a different source.
By employing an adversarial training strategy, we show that a network can be forced to learn a source-invariant representation.
arXiv Detail & Related papers (2020-08-04T07:41:15Z) - Data Augmentation using Generative Adversarial Networks (GANs) for
GAN-based Detection of Pneumonia and COVID-19 in Chest X-ray Images [0.0]
We propose a new GAN architecture for augmentation of chest X-rays for semi-supervised detection of pneumonia and COVID-19.
We show that the proposed GAN can be used to effectively augment data and improve classification accuracy of disease in chest X-rays for pneumonia and COVID-19.
arXiv Detail & Related papers (2020-06-05T18:30:20Z) - Predicting COVID-19 Pneumonia Severity on Chest X-ray with Deep Learning [57.00601760750389]
We present a severity score prediction model for COVID-19 pneumonia for frontal chest X-ray images.
Such a tool can gauge severity of COVID-19 lung infections that can be used for escalation or de-escalation of care.
arXiv Detail & Related papers (2020-05-24T23:13:16Z)
This list is automatically generated from the titles and abstracts of the papers in this site.
This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.