CDVAE

Crystal Diffusion Variational Autoencoder

Periodic SE(3)-invariant variational autoencoder and Langevin dynamics generator whose latent optimization is guided by a learned property predictor.

Read the paper

The loop

  1. Proposes

    A periodic GNN decoder predicts unit-cell lattice parameters, stoichometry, and denoises atomic coordinates via annealed Langevin dynamics.

  2. Judged by

    A learned property predictor guides latent optimization; validity checks (symmetry, bond distance, charge neutrality) and a separate learned energy model evaluate finished samples.

  3. Steers

    Gradient ascent in latent space steers sampling toward lower formation energy.

Headline result

Generated valid periodic crystal structures across carbon, perovskite, and Materials Project benchmarks with high structural and compositional validity.

What this does not prove

Latent-space property optimization relies on a learned surrogate predictor.

How it generalizes the harness

Uses a periodic SE(3)-equivariant diffusion VAE rather than an LLM, steered by a learned property predictor and evaluated with validity checks and a learned energy model.

Classification

Classification of CDVAE (Crystal Diffusion Variational Autoencoder)
What it proposesCrystal structure
Generative modelPeriodic SE(3)-equivariant diffusion VAE
Algorithmic authorityCrystallographic validity checker and a learned energy model
FeedbackScore
What gets updatedPopulation of candidates
When it runsAt inference time
ValidityApproximate physics (simulation, not lab results)
EvidenceCode released
LineageOriginal system