arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2304.09067eess.IVcs.CVphysics.med-ph

基于GAN的增强在深度学习COVID-19图像分类中的性能

Performance of GAN-based augmentation for deep learning COVID-19 image classification

National Centre for Nuclear Research(国家核研究中心)

Oleksandr Fedoruk, Konrad Klimaszewski, Aleksander Ogonowski, Rafał Możdżonek

AI总结:

针对医疗领域训练数据不足的问题,本研究在有限的COVID-19胸部X射线图像集上训练StyleGAN2-ADA模型进行数据增强,并比较了基于GAN的增强与经典图像增强方法在多类分类任务中的性能,发现经典方法表现优于GAN增强,EfficientNet-B0准确率达90.2%。

摘要:

将深度学习应用于医疗领域面临的最大挑战是训练数据的可用性。当面临有限的数据集时,数据增强是机器学习中使用的典型方法。在经典方法中,使用图像变换,即旋转、裁剪和亮度变化。在这项工作中,一个生成对抗网络的StyleGAN2-ADA模型在有限的COVID-19胸部X射线图像集上进行训练。在评估了生成图像的质量之后,它们被用来增加训练数据集,改善其类间平衡。我们考虑了胸部X射线图像的多类分类问题,包括文献中尚未被彻底探索的COVID-19阳性类。展示了基于迁移学习的COVID-19胸部X射线图像分类结果。比较了几个深度卷积神经网络模型的性能。研究了经典图像增强(即旋转、裁剪和亮度变化)对检测性能的影响。此外,将经典图像增强与基于GAN的增强进行了比较。最准确的模型是EfficientNet-B0,准确率为90.2%,在具有简单类平衡的数据集上训练。对于所考虑的数据集,发现GAN增强方法不如经典方法。

英文摘要:

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 a classical approach image transformations i.e. rotations, cropping and brightness changes are used. In this work, a StyleGAN2-ADA model of Generative Adversarial Networks is trained on the limited COVID-19 chest X-ray image set. After assessing the quality of generated images they are used to increase the training data set improving its balance between classes. We consider the multi-class classification problem of chest X-ray images including the COVID-19 positive class that hasn't been yet thoroughly explored in the literature. Results of transfer learning-based classification of COVID-19 chest X-ray images are presented. The performance of several deep convolutional neural network models is compared. The impact on the detection performance of classical image augmentations i.e. rotations, cropping, and brightness changes are studied. Furthermore, classical image augmentation is compared with GAN-based augmentation. The most accurate model is an EfficientNet-B0 with an accuracy of 90.2 percent, trained on a dataset with a simple class balancing. The GAN augmentation approach is found to be subpar to classical methods for the considered dataset.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑