论文标题

基于生成的对抗网络和使用胸部X射线数据集的微调深度转移学习模型检测冠状病毒(COVID-19)相关肺炎相关的肺炎

Detection of Coronavirus (COVID-19) Associated Pneumonia based on Generative Adversarial Networks and a Fine-Tuned Deep Transfer Learning Model using Chest X-ray Dataset

论文作者

Khalifa, Nour Eldeen M., Taha, Mohamed Hamed N., Hassanien, Aboul Ella, Elghamrawy, Sally

论文摘要

根据世界卫生组织的说法,Covid-19冠状病毒是毁灭性的病毒之一。这种新颖的病毒导致肺炎,这是一种感染,使人类的肺囊炎症。检测到这些炎症的方法之一是使用X射线胸部。在本文中,将介绍基于生成的对抗网络(GAN)的肺炎胸部X射线检测,并为有限数据集提供了微调的深度传输学习。 GAN的使用对提出的模型鲁棒性产生了积极影响,并使其不适合过度拟合问题,并有助于从数据集中生成更多图像。这项研究中使用的数据集由5863个X射线图像组成,其中有两个类别:正常和肺炎。这项研究仅使用10%的数据集来训练数据,并使用GAN生成90%的图像来证明所提出的模型的效率。通过论文,Alexnet,Googlenet,Squeeznet和Resnet18被选为深度转移学习模型,以检测胸部X射线的肺炎。这些模型是根据其架构上的少量层数选择的,这将反映在降低模型的复杂性以及消耗的内存和时间。使用GAN和深层转移模型的组合证明了根据测试精度测量的效率。研究得出的结论是,根据测试精度测量,RESNET18是最合适的深层传输模型,并在使用GAN作为图像增强器时,使用其他性能指标(例如精度,回忆和F1得分)获得了99%的速度。最后,研究结束时与相关工作进行了比较结果,该工作使用了相同的数据集,但该研究仅使用了10%的原始数据集。在测试准确性方面,提出的工作比相关工作取得了优越的结果。

The COVID-19 coronavirus is one of the devastating viruses according to the world health organization. This novel virus leads to pneumonia, which is an infection that inflames the lungs' air sacs of a human. One of the methods to detect those inflames is by using x-rays for the chest. In this paper, a pneumonia chest x-ray detection based on generative adversarial networks (GAN) with a fine-tuned deep transfer learning for a limited dataset will be presented. The use of GAN positively affects the proposed model robustness and made it immune to the overfitting problem and helps in generating more images from the dataset. The dataset used in this research consists of 5863 X-ray images with two categories: Normal and Pneumonia. This research uses only 10% of the dataset for training data and generates 90% of images using GAN to prove the efficiency of the proposed model. Through the paper, AlexNet, GoogLeNet, Squeeznet, and Resnet18 are selected as deep transfer learning models to detect the pneumonia from chest x-rays. Those models are selected based on their small number of layers on their architectures, which will reflect in reducing the complexity of the models and the consumed memory and time. Using a combination of GAN and deep transfer models proved it is efficiency according to testing accuracy measurement. The research concludes that the Resnet18 is the most appropriate deep transfer model according to testing accuracy measurement and achieved 99% with the other performance metrics such as precision, recall, and F1 score while using GAN as an image augmenter. Finally, a comparison result was carried out at the end of the research with related work which used the same dataset except that this research used only 10% of original dataset. The presented work achieved a superior result than the related work in terms of testing accuracy.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源