由于医学成像技术的限制、标注过程耗时昂贵，以及隐私安全问题，使得医疗数据集增扩成为相关研究领域的热点问题。传统的数据集增扩方式主要包括样本采样和仿射变换两种方法。样本采样技术主要是改变数据集的原始分布，包括降采样、过采样或二者结合，以扩大数据集的图像总量。另一种传统方法是仿射变换，它是一种二维坐标到二维坐标之间的线性变换。通过仿射变换，可以实现图像的旋转、镜像、拉伸等操作，从而增加图像数据集的总量。除了样本采样和仿射变换外，还有一些其他方法，如扣取、遮罩、缩放、加噪等处理方式，其原理与仿射变换相同，通过改变原图的表现形式来改变其特征矩阵，从而迫使网络从语义上去理解图片，达到对数据集的增扩目的。

目前，国内外已经在图像补全、图像修复、图像翻译、图像上色、视频生成等应用领域开展生成对抗网络算法研究的工作 [4] ，特别是图像修复领域已经取得了不少研究成果 [5] [6] [7] [8] [9] 。近年来，GAN技术在医疗领域也取得了不少研究成果。例如，文献 [10] [11] [12] 利用生成对抗网络技术，将含有噪声的医疗图像重建为清晰图像，为医护人员做出更准确的诊断提供了更好的基础。文献 [13] [14] [15] 利用生成对抗网络技术，对医疗影像重建进行了有效的研究探索。但使用生成对抗网络进行医疗图像增扩的研究相对比较缺乏，因此使用生成对抗网络技术对肺结节影像增扩具有较好的研究意义。

生成对抗网络(GAN)主要原理是通过对输入随机噪声的处理，生成器不断生成虚假的数据，判别器的任务则是判别生成器所生成数据的真假，二者在相互对抗的过程中互相学习逐渐变强。最终，通过这种博弈，生成器可输出与真实数据相似度极高的新数据。生成对抗网络基本结构如 图1 所示。

由RADFORD等 [16] 提出的DCGAN (Deep Convolutional Generative Adversarial Network)，是一种改进的生成对抗网络模型架构。它开创性的将CNN与GAN网络相结合，显著提高了性能。相比于普通的GAN，DCGAN在学习医疗影像数据病灶特征方面更具适应性和丰富性。它能够捕获医疗影像隐含的特征分布，并合成全新的数据样本。

本文使用的DCGAN网络的生成网络结构如 图2 所示，其可提高GAN生成图片的稳定性、以及生成质量和种类丰富度。虽然改进的DCGAN性能上有所提升，但是并未解决GAN训练过程中的稳定性问题，在训练时仍需平衡生成器与鉴别器的训练进程 [17] 。

由于原始GAN网络存在会导致梯度不稳定的问题，WGAN (Wasserstein Generative Adversarial Network) [18] 改进了原始GAN网络的损失函数，并采用沃瑟斯坦(Wassertein)距离来优化GAN结构的损失函数，从根本上解决了原始GAN网络训练过程中梯度消失的问题。但由于WGAN网络使用的Wassertein距离需要满足Lipschitz连续性限制条件，采用了权重裁剪的方式，这会导致模型建模能力弱化，容易出现梯度消失以及梯度爆炸问题。改进WGAN-GP [19] 通过梯度惩罚的方式改善了WGAN网络的训练稳定性，避免了梯度消失和爆炸问题，并提高了模型的建模能力，使得生成器可以更好地学习真实数据分布的特征。

本文使用数据集为LIDC-IDRI (The Lung Image Database Consortium)公开数据集 [20] ，选取了其中300例Dicom格式的病例，并经过预处理后得到3050张256 × 256格式大小的肺结节影像数据，其中2729张作为训练数据集，321张作为测试数据集，LIDC数据集中原始肺结节图像如 图3 所示。训练前统一设置每次训练输入肺结节数据数量(Batchsize)为16张，训练轮数(Epoch)为500轮，随后将肺结节影像数据分别输入DCGAN和WGAN-GP网络进行训练。训练完成后，综合评估生成器损失(G_Loss)与鉴别器损失(D_Loss)，保存最佳生成器网络的模型文件。DCGAN网络与WGAN-GP网络生成肺结节数据图像分别如 图4 、 图5 所示。

IS (Inception Score)指标使用图像分类网络Inception V3 Network来评估GAN网络中生成器生成的图片的质量。主要是将模型生成的图片生成的图片输入到预训练的网络Inception V3 Network中，根据分类网络输出的概率分布来评估生成图像的质量。IS但由于输出的IS指标过分依赖于Inception V3 Network，而Inception V3 Network是根据ImageNet数据集训练的，如果评价的数据集与训练分类网络的数据集相差过大时，可能不足以全面评估生成图片的质量，所以我们采用多种指标方式综合评估GAN网络中生成器的质量。计算IS指标的公式如下所示：

$\text{IS}\left(\text{G}\right)=\exp \left({\mathbb{E}}_{x\sim p_g}{D}_{KL}\left(p\left(y|x\right)\parallel p\left(y\right)\right)\right)$(1)

与IS指标性相比，FID (Fréchet Inception Distance) [21] 指标通过比较真实图像分布和生成图像分布之间的距离来综合评估生成图像的质量。同样是根据分类网络Inception V3 Network计算，但测试FID指标时将会同时输入真实图像和生成图像到分类网络中，再从中提取并计算真实图像的分布以及生成对抗网络生成图像分布之间的距离，其中距离越小代表着生成器生成的图像越接近于真实图像的分布，代表着生成的图像质量越高，因此FID越小越好。我们使用式(2)计算图像数据的FID值。

$\text{FID}\left(x,g\right)=\Vert {\mu }_x-{\mu }_g\Vert +\text{Tr}\left({\displaystyle {\sum }_x+}{\displaystyle {\sum }_g-}2{\left({\displaystyle {\sum }_x{\displaystyle {\sum }_g}}\right)}^{\frac{1}{2}}\right)$(2)

使用训练好的生成器模型输出生成的肺结节数据图像，为了保证结果的公平性，DCGAN网络与WGAN-GP网络生成的肺结节影像数量都为321张，与测试的真实图像数据集保持一致。我们首先使用IS与FID指标对生成的图像进行评估，结果如 表1 所示。接下来，使用SSIM (Structural Similarity Index)和PSNR (Peak Signal-to-Noise Ratio)指标来评估生成图像的质量以及与原始图像相似性。结果如 表2 所示。

从 表1 可以得出，WGAN-GP网络在生成肺结节图像的四个评估指标中均表现优于DCGAN网络。从IS和FID指标的角度来看，WGAN-GP网络相较于DCGAN网络表现更优。WGAN-GP网络的IS均值为1.74，高于DCGAN网络的1.57。而FID均值为137.85，低于DCGAN网络的144.41。这表明WGAN-GP网络生成的图像在质量和多样性方面均优于DCGAN网络，且更接近于真实图像的分布。从而表明其生成的图像质量更高。

SSIM评估了生成的图像与真实图像之间的结构相似性，而PSNR评估了信号与噪声之间的比例。从SSIM和PSNR指标来看，WGAN-GP网络在SSIM和PSNR上也表现更好。WGAN-GP网络的MSSIM为0.44，高于DCGAN网络的0.40，而MPSNR为30.24，略高于DCGAN网络的30.03。这表明WGAN-GP网络生成的图像在结构相似性和信噪比方面都更接近于真实图像，具有更高的质量和清晰度，其中包含更多的细节和复杂性。

综合来看，WGAN-GP网络相较于DCGAN网络在生成肺结节影像方面表现出更好的性能。其引入的梯度惩罚机制有效提高了模型的训练稳定性，从而生成了更接近真实数据分布的高质量图像。