FlowMM: Generating Materials with Riemannian Flow Matching

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Miller, Benjamin Kurt, Chen, Ricky T. Q., Sriram, Anuroop, Wood, Brandon M
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866929377194803200
author Miller, Benjamin Kurt
Chen, Ricky T. Q.
Sriram, Anuroop
Wood, Brandon M
author_facet Miller, Benjamin Kurt
Chen, Ricky T. Q.
Sriram, Anuroop
Wood, Brandon M
contents Crystalline materials are a fundamental component in next-generation technologies, yet modeling their distribution presents unique computational challenges. Of the plausible arrangements of atoms in a periodic lattice only a vanishingly small percentage are thermodynamically stable, which is a key indicator of the materials that can be experimentally realized. Two fundamental tasks in this area are to (a) predict the stable crystal structure of a known composition of elements and (b) propose novel compositions along with their stable structures. We present FlowMM, a pair of generative models that achieve state-of-the-art performance on both tasks while being more efficient and more flexible than competing methods. We generalize Riemannian Flow Matching to suit the symmetries inherent to crystals: translation, rotation, permutation, and periodic boundary conditions. Our framework enables the freedom to choose the flow base distributions, drastically simplifying the problem of learning crystal structures compared with diffusion models. In addition to standard benchmarks, we validate FlowMM's generated structures with quantum chemistry calculations, demonstrating that it is about 3x more efficient, in terms of integration steps, at finding stable materials compared to previous open methods.
format Preprint
id arxiv_https___arxiv_org_abs_2406_04713
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle FlowMM: Generating Materials with Riemannian Flow Matching
Miller, Benjamin Kurt
Chen, Ricky T. Q.
Sriram, Anuroop
Wood, Brandon M
Machine Learning
Materials Science
Artificial Intelligence
Computational Physics
Crystalline materials are a fundamental component in next-generation technologies, yet modeling their distribution presents unique computational challenges. Of the plausible arrangements of atoms in a periodic lattice only a vanishingly small percentage are thermodynamically stable, which is a key indicator of the materials that can be experimentally realized. Two fundamental tasks in this area are to (a) predict the stable crystal structure of a known composition of elements and (b) propose novel compositions along with their stable structures. We present FlowMM, a pair of generative models that achieve state-of-the-art performance on both tasks while being more efficient and more flexible than competing methods. We generalize Riemannian Flow Matching to suit the symmetries inherent to crystals: translation, rotation, permutation, and periodic boundary conditions. Our framework enables the freedom to choose the flow base distributions, drastically simplifying the problem of learning crystal structures compared with diffusion models. In addition to standard benchmarks, we validate FlowMM's generated structures with quantum chemistry calculations, demonstrating that it is about 3x more efficient, in terms of integration steps, at finding stable materials compared to previous open methods.
title FlowMM: Generating Materials with Riemannian Flow Matching
topic Machine Learning
Materials Science
Artificial Intelligence
Computational Physics
url https://arxiv.org/abs/2406.04713