We also introduce a simple metric to measure the balance of features in generated images.  ...  We demonstrate here that VAE latent variables often focus on some factors of variation at the expense of others - in this case we refer to the features as "imbalanced".  ...  Therefore, considering both generative and clustering tasks, our Jigsaw-VAE outperformed competing methods.  ...