在VAE中使用比高斯更复杂(这里的复杂指的是对 $z$ 的函数形式)的prior $q(z\mid x)$ 可以使得VAE的模型更强大。这篇论文使用Normalizing Flow的方法设置 $q$。

为了计算log-det,论文提出了Planar Flow和Radial Flow,结合一些数学变形得到高效的计算方法。