Ouro
  • Docs
  • Blog
Join for freeSign in
  • Home
  • Teams
  • Search
Assets
  • Quests
  • Posts
  • APIs
  • Data
  • Home
  • Teams
  • Search
Assets
  • Quests
  • Posts
  • APIs
  • Data
9mo
199 views

On this page

  • Orb latent space exploration
Loading compatible actions...
Loading comments...

Orb latent space exploration

This is a continued deep-dive into the latent space generated by the Orb model prior to it's MLFF tasks. I have been attempting to train a model on Tc prediction using this latent space as a feature vector. More on that here:

Orb latent space to Tc prediction

post

After reading the MatterSim paper, the authors proposed the idea of using the MLFF's latent space as a direct property prediction feature set. Earlier, and I had been thinking about using a VAE (or s

9mo

In the post, I'll be focusing purely on what we can learn from the latent space and see what sort of patterns emerge related to Tc, material composition and class, structure, etc.

We can visualize the latent space with dimensionality reduction using UMAP and t-SNE. Try this interactive Plotly visualization!

Interactive 3D latent space exploration with Tc

.html file

Using the 256 dimensional latent space output from the Orb model, we visualize the 3DSC(MP) dataset using t-SNE and UMAP. The UMAP projection has been given the target for learning a manifold that keeps similar Tc materials close together.

9mo

Make sure to bring the visualization into fullscreen with the expand button at the top right of the visualization. There are two, side-by-side, visualizations here.

Another version of the visualization with better point labeling:

3D latent space exploration with Tc

.html file

Using the 256 dimensional latent space output from the Orb model, we visualize the 3DSC(MP) dataset using UMAP with direction from Tc labels. Hover a point to see Tc, formula, and Material Project identifier.

9mo

We can see that clusters of similar Tc often relates to materials of the same class, e.g. like cuprates, BaCu families, and other chemical systems.