[发明专利]一种多判别器误差反传的对抗网络方法

说明书

技术领域

本发明涉及深度学习神经网络技术领域，具体涉及一种多判别器误差反传的对抗网络方法。

背景技术

生成式对抗网络(Generative Adversarial Network,简称GAN)是由Goodfellow在2014年提出的框架，它基于“博奕论”的思想，构造生成器(generator)和判别器(discriminator)两种模型，前者通过输入(0，1)的均匀噪声或高斯随机噪声生成图像，后者对输入的图像进行判别，确定是来自数据集的图像还是由生成器产生的图像。判别器每完成一次判断，将结果误差返回给生成器。

然而，原始的GAN模型中，判别器的数量只有一个，这就意味着，生成器生成图像真假与否的判决，只取决于这一个判别器。这种情况下所产生的结果是，如果判别器对生成器生成图像的判断有所偏差，则会影响整个网络训练的精确度和速度。

发明内容

本发明的目的是为了解决现有技术中的上述缺陷，构建了一种多判别器误差反传的对抗网络方法，该对抗网络方法能够通过多个判别器共同判断的角度，对整个网络的训练提供更多的可靠性。它改变了以往判别生成图像只通过一个判别器做出决定的局面，创造性地提出了多个判别器协同工作的网络训练算法，能够避免原有GAN模型对生成图像判断的单一性，能够从更加客观的角度去评判生成器的生成图像，从而使整个网络的训练更加稳定。

本发明的目的可以通过采取如下技术方案达到：

一种多判别器误差反传的对抗网络方法，所述动态调整算法包括下列步骤：

S1、构造生成对抗网络GAN模型，模型包含生成器和判别器；

S2、在现有GAN模型的基础上构造多个判别器；

S3、准备好数据集对改进之后的多判别器网络进行训练；

S4、在每次训练的过程中，记录所有判别器的损失函数；

S5、计算所有判别器损失函数的平均值，返回至生成器进行后续的网络训练。

进一步地，所述的步骤S2具体如下：

在原有GAN模型的基础上，复制多个判别器，分别同时接收来自生成器的生成图像和来自数据集的真实图像。

进一步地，所述的步骤S3中准备好数据集对改进之后的多判别器网络进行训练，过程如下：

S31、根据判别器的数量，选定数据集中的同等数量的真实图像；

S32、将互不相同的来自数据集中的真实图像，输入判别器中进行训练。

进一步地，所述的步骤S4中在每次训练的过程中，记录所有判别器的损失函数，过程如下：

S41、将数据集中的图像输入判别器中进行训练；

S42、记录每一个判别器在每次迭代之后的损失函数。

进一步地，所述的步骤S5中，计算所有判别器损失函数的平均值，返回至生成器进行后续的网络训练，过程如下：

S51、计算所有判别器在每次迭代之后的损失函数之和；

S52、根据判别器数量的大小，对求和之后的损失函数求平均操作；

S53、将求平均之后的损失函数返回至生成器中继续进行网络训练。

本发明相对于现有技术具有如下的优点及效果：

1.客观性：本发明在原始GAN模型的基础之上，提出了多个判别器同时接收生成器生成图像的结构，从多个不同的角度对生成图像进行判决，增强了网络训练的客观性；

2.稳定性：本发明所构建的多判别器误差反传的对抗网络算法，能够从多个判别器的角度对生成图像进行判断，相对于原始模型的损失函数，本发明提出的均值损失函数能够根据多个判别器的结构对生成图像进行修正，更具有可行性，从而使整个网络的训练更加稳定。

附图说明

图1是多判别器误差反传的网络结构图。

具体实施方式

为使本发明实施例的目的、技术方案和优点更加清楚，下面将结合本发明实施例中的附图，对本发明实施例中的技术方案进行清楚、完整地描述，显然，所描述的实施例是本发明一部分实施例，而不是全部的实施例。基于本发明中的实施例，本领域普通技术人员在没有做出创造性劳动前提下所获得的所有其他实施例，都属于本发明保护的范围。

实施例

本实施例公开了一种多判别器误差反传的对抗网络方法，具体包括下列步骤：

步骤S1、构造生成对抗网络GAN模型，模型包含生成器和判别器；

步骤S2、在现有GAN模型的基础上构造多个判别器；

步骤S3、准备好数据集对改进之后的多判别器网络进行训练。