We propose a Gaussian manifold variational auto-encoder (GM-VAE) whose latent
space consists of a set of Gaussian distributions. It is known that the set of
the univariate Gaussian distributions with the Fisher information metric form a
hyperbolic space, which we call a Gaussian manifold. To learn the VAE endowed
with the Gaussian manifolds, we propose a pseudo-Gaussian manifold normal
distribution based on the Kullback-Leibler divergence, a local approximation of
the squared Fisher-Rao distance, to define a density over the latent space. In
experiments, we demonstrate the efficacy of GM-VAE on two different tasks:
density estimation of image datasets and environment modeling in model-based
reinforcement learning. GM-VAE outperforms the other variants of hyperbolic-
and Euclidean-VAEs on density estimation tasks and shows competitive
performance in model-based reinforcement learning. We observe that our model
provides strong numerical stability, addressing a common limitation reported in
previous hyperbolic-VAEs.

通过使用高斯流形变分自编码器 (GM-VAE) 来提高图像数据集的密度估计和基于模型的强化学习下的环境建模。GM-VAE 在估计密度任务上优于其他变量的双曲线和欧几里得 VAEs，并在基于模型的强化学习中展现出竞争性的性能。