Timezone: »

Shape your Space: A Gaussian Mixture Regularization Approach to Deterministic Autoencoders
Amrutha Saseendran · Kathrin Skubch · Stefan Falkner · Margret Keuper

Wed Dec 08 12:30 AM -- 02:00 AM (PST) @ Virtual

Variational Autoencoders (VAEs) are powerful probabilistic models to learn representations of complex data distributions. One important limitation of VAEs is the strong prior assumption that latent representations learned by the model follow a simple uni-modal Gaussian distribution. Further, the variational training procedure poses considerable practical challenges. Recently proposed regularized autoencoders offer a deterministic autoencoding framework, that simplifies the original VAE objective and is significantly easier to train. Since these models only provide weak control over the learned latent distribution, they require an ex-post density estimation step to generate samples comparable to those of VAEs. In this paper, we propose a simple and end-to-end trainable deterministic autoencoding framework, that efficiently shapes the latent space of the model during training and utilizes the capacity of expressive multi-modal latent distributions. The proposed training procedure provides direct evidence if the latent distribution adequately captures complex aspects of the encoded data. We show in experiments the expressiveness and sample quality of our model in various challenging continuous and discrete domains. An implementation is available at https://github.com/boschresearch/GMM_DAE.

Author Information

Amrutha Saseendran (Bosch Center for Artificial Intelligence)
Kathrin Skubch (Bosch Center for Artificial Intelligence)
Stefan Falkner (University of Freiburg)
Margret Keuper (Uni-Mannheim)

More from the Same Authors