Improving COVID-19 CXR Detection with Synthetic Data Augmentation

2021/12/14 by Schaudt, Daniel, Kloth, Christopher, Spaete, Christian +3
#Computer Vision and Pattern Recognition (cs.CV) #FOS: Computer and information sciences #FOS: Electrical engineering #Image and Video Processing (eess.IV) #Machine Learning (cs.LG) #electronic engineering #information engineering

paper · doi:10.48550/arxiv.2112.07529

Abstract

Since the beginning of the COVID-19 pandemic, researchers have developed deep learning models to classify COVID-19 induced pneumonia. As with many medical imaging tasks, the quality and quantity of the available data is often limited. In this work we train a deep learning model on publicly available COVID-19 image data and evaluate the model on local hospital chest X-ray data. The data has been reviewed and labeled by two radiologists to ensure a high quality estimation of the generalization capabilities of the model. Furthermore, we are using a Generative Adversarial Network to generate synthetic X-ray images based on this data. Our results show that using those synthetic images for data augmentation can improve the model's performance significantly. This can be a promising approach for many sparse data domains.

Related

• Using Deep Convolutional Neural Networks to Diagnose COVID-19 From Chest X-Ray Images
• Deep Learning COVID-19 Features on CXR using Limited Training Data Sets
• Iteratively Pruned Deep Learning Ensembles for COVID-19 Detection in Chest X-rays
• Fast and accurate detection of Covid-19-related pneumonia from chest X-ray images with novel deep learning model
• COVID-CXNet: Detecting COVID-19 in Frontal Chest X-ray Images using Deep Learning