DeepAI AI Chat
Log In Sign Up

Crystal Diffusion Variational Autoencoder for Periodic Material Generation

by   Tian Xie, et al.

Generating the periodic structure of stable materials is a long-standing challenge for the material design community. This task is difficult because stable materials only exist in a low-dimensional subspace of all possible periodic arrangements of atoms: 1) the coordinates must lie in the local energy minimum defined by quantum mechanics, and 2) global stability also requires the structure to follow the complex, yet specific bonding preferences between different atom types. Existing methods fail to incorporate these factors and often lack proper invariances. We propose a Crystal Diffusion Variational Autoencoder (CDVAE) that captures the physical inductive bias of material stability. By learning from the data distribution of stable materials, the decoder generates materials in a diffusion process that moves atomic coordinates towards a lower energy state and updates atom types to satisfy bonding preferences between neighbors. Our model also explicitly encodes interactions across periodic boundaries and respects permutation, translation, rotation, and periodic invariances. We significantly outperform past methods in three tasks: 1) reconstructing the input structure, 2) generating valid, diverse, and realistic materials, and 3) generating materials that optimize a specific property. We also provide several standard datasets and evaluation metrics for the broader machine learning community.


page 4

page 6

page 9

page 10

page 11

page 12

page 13

page 16


Data-driven discovery of novel 2D materials by deep generative models

Efficient algorithms to generate candidate crystal structures with good ...

Inverse design of crystals using generalized invertible crystallographic representation

Deep learning has fostered many novel applications in materials informat...

Disentanglement with Hyperspherical Latent Spaces using Diffusion Variational Autoencoders

A disentangled representation of a data set should be capable of recover...

Periodic Graph Transformers for Crystal Material Property Prediction

We consider representation learning on periodic graphs encoding crystal ...

Graph Neural Network for Hamiltonian-Based Material Property Prediction

Development of next-generation electronic devices for applications call ...

Neural Networks Fail to Learn Periodic Functions and How to Fix It

Previous literature offers limited clues on how to learn a periodic func...

Fast predictions of lattice energies by continuous isometry invariants of crystal structures

Crystal Structure Prediction (CSP) aims to discover solid crystalline ma...