Abstract
Discrete latent spaces in variational autoencoders have been shown toeffectively capture the data distribution for many real-world problems such asnatural language understanding, human intent prediction, and visual scenerepresentation. However, discrete latent spaces need to be sufficiently largeto capture the complexities of real-world data, rendering downstream taskscomputationally challenging. For instance, performing motion planning in ahigh-dimensional latent representation of the environment could be intractable.We consider the problem of sparsifying the discrete latent space of a trainedconditional variational autoencoder, while preserving its learnedmultimodality. As a post hoc latent space reduction technique, we useevidential theory to identify the latent classes that receive direct evidencefrom a particular input condition and filter out those that do not. Experimentson diverse tasks, such as image generation and human behavior prediction,demonstrate the effectiveness of our proposed technique at reducing thediscrete latent sample space size of a model while maintaining its learnedmultimodality.