论文标题
与平均教师 +转移学习gan生成现实的Covid19 X射线
Generating Realistic COVID19 X-rays with a Mean Teacher + Transfer Learning GAN
论文作者
论文摘要
Covid-19是一种新型的传染病,截至2020年8月,全球造成了80万多人死亡。快速测试的需求是高优先级,包括X射线图像分类在内的替代测试策略是一个有希望的研究领域。但是,目前,COVID19 X射线图像的公共数据集具有较低的数据量,这使得开发准确的图像分类器变得具有挑战性。最近的几篇论文利用了生成的对抗网络(GAN),以增加培训数据量。但是现实的合成COVID19 X射线仍然具有挑战性。我们提出了一种新型的平均老师 +转移gan(MTT-GAN),该gan(MTT-GAN)生成Covid19胸部X射线高品质的图像。为了创建一个更准确的gan,我们从Kaggle肺炎X射线数据集中采用转移学习,这是一个高度相关的数据源数量级,比公共COVID19数据集大。此外,我们采用平均教师算法作为提高培训稳定性的限制。我们的定性分析表明,MTT-GAN生成的X射线图像非常优于基线gan,并且在视觉上与真实的X射线相当。尽管经过董事会认证的放射科医生可以将MTT-GAN假货与Real Covid19 X射线区分开。定量分析表明,与基线GAN相比,MTT-GAN极大地提高了二进制CoVID19分类器以及多级肺炎分类器的准确性。与最近报道的类似二进制和多级CoVID19筛选任务的文献结果相比,我们的分类准确性是有利的。
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.