Open Materials Generation with Stochastic Interpolants

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Hoellmer, Philipp, Egg, Thomas, Martirossyan, Maya M., Fuemmeler, Eric, Shui, Zeren, Gupta, Amit, Prakash, Pawan, Roitberg, Adrian, Liu, Mingjie, Karypis, George, Transtrum, Mark, Hennig, Richard G., Tadmor, Ellad B., Martiniani, Stefano
Format: Preprint
Veröffentlicht: 2025
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866913938185125888
author Hoellmer, Philipp
Egg, Thomas
Martirossyan, Maya M.
Fuemmeler, Eric
Shui, Zeren
Gupta, Amit
Prakash, Pawan
Roitberg, Adrian
Liu, Mingjie
Karypis, George
Transtrum, Mark
Hennig, Richard G.
Tadmor, Ellad B.
Martiniani, Stefano
author_facet Hoellmer, Philipp
Egg, Thomas
Martirossyan, Maya M.
Fuemmeler, Eric
Shui, Zeren
Gupta, Amit
Prakash, Pawan
Roitberg, Adrian
Liu, Mingjie
Karypis, George
Transtrum, Mark
Hennig, Richard G.
Tadmor, Ellad B.
Martiniani, Stefano
contents The discovery of new materials is essential for enabling technological advancements. Computational approaches for predicting novel materials must effectively learn the manifold of stable crystal structures within an infinite design space. We introduce Open Materials Generation (OMatG), a unifying framework for the generative design and discovery of inorganic crystalline materials. OMatG employs stochastic interpolants (SI) to bridge an arbitrary base distribution to the target distribution of inorganic crystals via a broad class of tunable stochastic processes, encompassing both diffusion models and flow matching as special cases. In this work, we adapt the SI framework by integrating an equivariant graph representation of crystal structures and extending it to account for periodic boundary conditions in unit cell representations. Additionally, we couple the SI flow over spatial coordinates and lattice vectors with discrete flow matching for atomic species. We benchmark OMatG's performance on two tasks: Crystal Structure Prediction (CSP) for specified compositions, and 'de novo' generation (DNG) aimed at discovering stable, novel, and unique structures. In our ground-up implementation of OMatG, we refine and extend both CSP and DNG metrics compared to previous works. OMatG establishes a new state of the art in generative modeling for materials discovery, outperforming purely flow-based and diffusion-based implementations. These results underscore the importance of designing flexible deep learning frameworks to accelerate progress in materials science. The OMatG code is available at https://github.com/FERMat-ML/OMatG.
format Preprint
id arxiv_https___arxiv_org_abs_2502_02582
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Open Materials Generation with Stochastic Interpolants
Hoellmer, Philipp
Egg, Thomas
Martirossyan, Maya M.
Fuemmeler, Eric
Shui, Zeren
Gupta, Amit
Prakash, Pawan
Roitberg, Adrian
Liu, Mingjie
Karypis, George
Transtrum, Mark
Hennig, Richard G.
Tadmor, Ellad B.
Martiniani, Stefano
Machine Learning
Materials Science
The discovery of new materials is essential for enabling technological advancements. Computational approaches for predicting novel materials must effectively learn the manifold of stable crystal structures within an infinite design space. We introduce Open Materials Generation (OMatG), a unifying framework for the generative design and discovery of inorganic crystalline materials. OMatG employs stochastic interpolants (SI) to bridge an arbitrary base distribution to the target distribution of inorganic crystals via a broad class of tunable stochastic processes, encompassing both diffusion models and flow matching as special cases. In this work, we adapt the SI framework by integrating an equivariant graph representation of crystal structures and extending it to account for periodic boundary conditions in unit cell representations. Additionally, we couple the SI flow over spatial coordinates and lattice vectors with discrete flow matching for atomic species. We benchmark OMatG's performance on two tasks: Crystal Structure Prediction (CSP) for specified compositions, and 'de novo' generation (DNG) aimed at discovering stable, novel, and unique structures. In our ground-up implementation of OMatG, we refine and extend both CSP and DNG metrics compared to previous works. OMatG establishes a new state of the art in generative modeling for materials discovery, outperforming purely flow-based and diffusion-based implementations. These results underscore the importance of designing flexible deep learning frameworks to accelerate progress in materials science. The OMatG code is available at https://github.com/FERMat-ML/OMatG.
title Open Materials Generation with Stochastic Interpolants
topic Machine Learning
Materials Science
url https://arxiv.org/abs/2502.02582