Variational Autoencoder Toolkit

Conditional Translationally Invariant VAE

In this chapter, we introduce the conditional translational Variational Autoencoder (CtVAE), which incorporates translational vectors (txt_x, tyt_y) and conditional variables, such as cation type, into the latent space. The CtVAE captures positional shifts alongside intrinsic variations, enabling the analysis of system-specific physical properties. This framework provides a structured approach to studying translation-related phenomena while maintaining the distinct physical and chemical characteristics of the components.

Source:CtVAE
(a) The latent manifold generated from the CtVAE demonstrates A and B site atoms reconstructed structure across the latent space. (b) The latent space distribution with KDE highlights clusters.

Figure 1:(a) The latent manifold generated from the CtVAE demonstrates A and B site atoms reconstructed structure across the latent space. (b) The latent space distribution with KDE highlights clusters.

The clear separations and clustering in the latent space reflect the incorporation of physical constraints and rotational invariances.ß

Source:CtVAE
Latent maps generated from the CtVAE.

Figure 2:Latent maps generated from the CtVAE.

Both z1z_1, z2z_2, and txt_x, txyt_xy appear to capture information about the domain structures. However, this raises the question: are we utilizing an excessive number of latent variables to represent the system?