摘要 — 最近,深度学习方法,特别是卷积神经网络 (CNN),在计算机视觉领域取得了巨大突破。此外,大规模注释数据集是成功训练过程的关键。然而,在医学领域获取这样的数据集是一个巨大的挑战。为此,我们提出了一种使用循环一致性生成对抗网络 (GAN) 生成合成医学图像的数据增强方法。我们添加半监督注意模块来生成具有令人信服的细节的图像。我们将肿瘤图像和正常图像视为两个领域。所提出的基于 GAN 的模型可以从正常图像生成肿瘤图像,反过来,它也可以从肿瘤图像生成正常图像。此外,我们表明生成的医学图像可用于提高 ResNet18 在医学图像分类中的性能。我们的模型应用于三个有限的肿瘤 MRI 图像数据集。我们首先在有限的数据集上生成 MRI 图像,然后训练三种流行的分类模型以获得最佳的肿瘤分类模型。最后,我们使用经典的数据增强方法使用真实图像训练分类模型,使用合成图像训练分类模型。这些训练模型之间的分类结果表明,与经典的数据增强方法相比,所提出的 SAG-GAN 数据增强方法可以提高准确率和 AUC。我们相信所提出的数据增强方法可以应用于其他医学图像领域,并提高计算机辅助诊断的准确性。索引词——生成对抗网络 (GAN)、数据增强、注意模块、医学图像处理
主要关键词