Pytorch实现之GIEGAN（生成器信息增强GAN）训练自己的数据集

简介

简介：在训练数据样本之前首先利用VAE来推断潜在空间中不同类的分布，用于后续的训练，并使用它来初始化GAN。与ACGAN和BAGAN不同的是，提出的GIEGAN有一个分类器结构，这个分类器主要判断生成的图像或者样本图像属于哪个类，而鉴别器仅判断图像是来自于生成器还是真实样本。

论文题目：Generator Information Enhancement Generative Adversarial Networks for Alleviating Data Imbalance Problems（生成器信息增强生成对抗网络缓解数据不平衡问题）

会议：2022 7th International Conference on Intelligent Computing and Signal Processing (ICSP)

摘要：在现实数据集中，大量的类分布是不平衡的，严重影响了深度学习方法的分类能力。数据增强可以通过为少数类创建数据来重新平衡数据集来解决此类问题。然而，如何从给定的有限训练数据的真实分布中生成数据仍然是一个挑战。为了克服这一挑战，我们提出了一种新颖有效的合成过采样方法，生成器信息增强生成对抗网络（GIE-GAN）。其生成信息在两个方面得到增强。首先，生成器可以通过使用变分自编码器进行初始化来获得所有类的公共知识。其次，在训练过程中，在GAN中加入一个能够为生成器提