当前位置：首页 > news >正文

生成对抗网络（Generative adversarial network——GAN）

news 来源：原创 2025/8/2 11:09:55

文章目录

1. 前言
- 1.1 判别器和生成器的作用？
- 2.2 个人总结
2. 核心代码示例
- 2.1 训练判别器网络
- 2.2 训练生成器网络
参考文章

1. 前言

生成对抗网络的原文：Generative Adversarial Nets，该论文的精读视频：生成对抗网络GAN开山之作论文精读，该论文解析相关的视频：生成对抗网络GAN原理解析。下面是自己对生成对抗网络的理解。

1.1 判别器和生成器的作用？

生成器（Generator）

（1）作用：生成器的主要任务是生成看起来“真实”的数据（假数据）。它从一个随机噪声向量（通常是高斯分布或均匀分布的随机数）作为输入，经过一系列变换后输出与目标数据分布相似的数据样本。
（2）工作机制：接收随机噪声 z 作为输入；将 z 映射到目标数据空间（例如图像、音频等），生成伪造数据 G(z)；目标是让生成的数据尽可能“欺骗”判别器，使其无法区分生成的数据和真实数据。

判别器（Discriminator）

（1）作用：判别器是一个二分类器，用于判断输入数据是来自真实数据分布还是生成器生成的伪造数据。
（2）工作机制：接收真实数据 x 和生成数据 G(z) 作为输入；输出一个概率值（" 例如0表示“假”，1表示“真” "），表示输入数据是真实数据的可能性；目标是尽可能准确地分辨出真实数据和伪造数据。

两者之间的关系

（1）对抗过程：生成器和判别器在训练过程中不断博弈。生成器试图生成更逼真的数据以欺骗判别器，而判别器则努力提高自己的辨别能力。
（2）动态平衡：理想情况下，经过充分训练后，生成器能够生成几乎无法区分于真实数据的样本，而判别器对所有输入的判断概率接近 0.5（即无法分辨真假）。

真假数据的输入流程

在这里插入图片描述

2.2 个人总结

（1）对于生成器 G ：随机噪声数据 z 通过生成器 G 得到 G(z)，我们希望 G(z) 经过判别器 D 输出的概率 D[G(z)] 越大越好（预测概率 D[G(z)] ——> 1）。
（2）对于判别器 D：真实数据 x 通过判别器 D 输出概率 D(x)，我们希望概率 D(x) 越大越好（预测概率 D(x) ——> 1）；对于虚假数据 G(z)，我们希望 G(z) 经过判别器 D 输出的概率 D[G(z)] 越小越好（预测概率 D[G(z)] ——> 0）。

2. 核心代码示例

class Discriminator(nn.Module):# 判别器网络...class Generator(nn.Module):# 生成器网络...# 创建生成器，判别器对象
generator = Generator()
discriminator = Discriminator()# 定义生成器，判别器对象相应的优化器
optimizer_G = torch.optim.Adam(generator.parameters(), lr=opt.lr, betas=(opt.b1, opt.b2))
optimizer_D = torch.optim.Adam(discriminator.parameters(), lr=opt.lr, betas=(opt.b1, opt.b2))# 定义二分类的交叉熵损失
criterion = torch.nn.BCELoss()

2.1 训练判别器网络

定义真实数据的label为1，虚假数据的label为0

real_label = Variable(torch.ones(imgs.size(0), 1)).cuda()  ## 定义真实的图片label为1
fake_label = Variable(torch.zeros(imgs.size(0), 1)).cuda()  ## 定义假的图片的label为0

计算真实数据的损失

real_out = discriminator(real_img)  ## 将真实图片放入判别器中, 输出的值越接近1越好
loss_real_D = criterion(real_out, real_label)  ## 得到真实图片的loss

计算虚假数据的损失

z = Variable(torch.randn(imgs.size(0), opt.latent_dim)).cuda()  ## 随机生成一些噪声, 大小为(128, 100)
fake_img = generator(z).detach()  ## 随机噪声放入生成网络中，生成一张假的图片。
fake_out = discriminator(fake_img)  ## 判别器判断假的图片, 输出的值越接近0越好
loss_fake_D = criterion(fake_out, fake_label)  ## 得到假的图片的loss

利用损失函数优化判别器网络

loss_D = loss_real_D + loss_fake_D  ## 损失包括判真损失和判假损失
optimizer_D.zero_grad()  ## 在反向传播之前，先将梯度归0
loss_D.backward()  ## 将误差反向传播
optimizer_D.step()  ## 更新参数

2.2 训练生成器网络

定义虚假数据的label为1

real_label = Variable(torch.ones(imgs.size(0), 1)).cuda()

计算生成器网络的损失

z = Variable(torch.randn(imgs.size(0), opt.latent_dim)).cuda()  ## 得到随机噪声
fake_img = generator(z)  ## 随机噪声输入到生成器中，得到一副假的图片
output = discriminator(fake_img)  ## 经过判别器得到的结果
loss_G = criterion(output, real_label)  ## 得到的假的图片与真实的图片的label的loss

利用损失函数优化生成器网络

optimizer_G.zero_grad()  ## 梯度归0
loss_G.backward()  ## 进行反向传播
optimizer_G.step()  ## step()一般用在反向传播后面,用于更新生成网络的参数

参考文章

生成对抗网络（GAN）
生成对抗网络——GAN原理与代码
GAN（生成对抗网络）的系统全面介绍（醍醐灌顶）
Pytorch生成对抗网络（GAN）官方入门教程