变分AutoEncoder (VAE)采样控制

Question

假设使用mnist数据训练变分自动编码器(VAE)。抽样，从正态分布中提取。我的问题是:假设我只想生成1s，而不生成其他数字。我怎么能做到呢？我采样直到生成一个1，然后继续从那个点附近取样吗？或者有没有一种更有控制的方法来告诉VAE要生成什么数字？谢谢

编辑*

理想的编码器将接受任何输入并生成来自\mathcal{N}(0,1)的示例。如果输入为1、2或3，则采样将全部来自\mathcal{N}(0,1)，那么，解码器如何知道输入是1、2或3？

Answer 1

您可以在从分类分布(https://wiseodd.github.io/techblog/2016/12/17/conditional-vae/)中提取的另一个变量上为潜在变量设置条件。