[解读] ClusterGAN : Latent Space Clustering in Generative Adversarial Networks

最新推荐文章于 2026-04-07 10:47:25 发布

原创

最新推荐文章于 2026-04-07 10:47:25 发布 · 2k 阅读

收录于

当前文章被以下社区和专栏收录：

链接: https://arxiv.org/abs/1809.03627v2

ClusterGAN 是一个 GAN 的变体, 通过使得生成样本反推的隐变量在隐空间中具有明显的簇的特征, 来提升 GAN 的性能.

方法

本文主要贡献: 1. 利用混合离散和连续型变量的隐变量来创建一个不平滑的集合形状. 2. 提出一种新的反向传播算法, 可以适应离散连续的混合变量, 以及一个确切的逆映射网络, 来得到给定数据对应的隐变量. 3. 联合训练 GAN 和逆映射网络, 使得投影空间的几何距离反映出变量的几何距离.

在这里插入图片描述
上图是 ClusterGAN 的结构, 与标准的 GAN 相比, 它多了编码网络 E, 编码样本来获得隐变量和类别. 生成器的输入除了隐变量 $Z_n$ , 还嵌入了随机生成的 one-hot 编码的类别变量.

优化目标:
$\begin{aligned} \min _{\Theta_{G}, \Theta_{E}} \max _{\Theta_{D}} \underset{x \sim \mathbb{P}_{x}^{r}}{\mathbf{E}} q(\mathcal{D}(x)) & +\underset{z \sim \mathbb{P}_{z}}{\mathbf{E}} q(1-\mathcal{D}(\mathcal{G}(z))) \\ & + \beta_{n} \underset{z \sim \mathbb{P}_{z}}{\mathbf{E}}\left\|z_{n}-\mathcal{E}\left(\mathcal{G}\left(z_{n}\right)\right)\right\|_{2}^{2} \\ & + \beta_{c} \underset{z \sim \mathbb{P}_{z}}{\mathbf{E}} \mathcal{H}\left(z_{c}, \mathcal{E}\left(\mathcal{G}\left(z_{c}\right)\right)\right. \end{aligned}$