FlowLLM: Flow Matching for Material Generation with Large Language Models as Base Distributions

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Sriram, Anuroop, Miller, Benjamin Kurt, Chen, Ricky T. Q., Wood, Brandon M.
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866910678817701888
author Sriram, Anuroop
Miller, Benjamin Kurt
Chen, Ricky T. Q.
Wood, Brandon M.
author_facet Sriram, Anuroop
Miller, Benjamin Kurt
Chen, Ricky T. Q.
Wood, Brandon M.
contents Material discovery is a critical area of research with the potential to revolutionize various fields, including carbon capture, renewable energy, and electronics. However, the immense scale of the chemical space makes it challenging to explore all possible materials experimentally. In this paper, we introduce FlowLLM, a novel generative model that combines large language models (LLMs) and Riemannian flow matching (RFM) to design novel crystalline materials. FlowLLM first fine-tunes an LLM to learn an effective base distribution of meta-stable crystals in a text representation. After converting to a graph representation, the RFM model takes samples from the LLM and iteratively refines the coordinates and lattice parameters. Our approach significantly outperforms state-of-the-art methods, increasing the generation rate of stable materials by over three times and increasing the rate for stable, unique, and novel crystals by $\sim50\%$ - a huge improvement on a difficult problem. Additionally, the crystals generated by FlowLLM are much closer to their relaxed state when compared with another leading model, significantly reducing post-hoc computational cost.
format Preprint
id arxiv_https___arxiv_org_abs_2410_23405
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle FlowLLM: Flow Matching for Material Generation with Large Language Models as Base Distributions
Sriram, Anuroop
Miller, Benjamin Kurt
Chen, Ricky T. Q.
Wood, Brandon M.
Machine Learning
Materials Science
Artificial Intelligence
Material discovery is a critical area of research with the potential to revolutionize various fields, including carbon capture, renewable energy, and electronics. However, the immense scale of the chemical space makes it challenging to explore all possible materials experimentally. In this paper, we introduce FlowLLM, a novel generative model that combines large language models (LLMs) and Riemannian flow matching (RFM) to design novel crystalline materials. FlowLLM first fine-tunes an LLM to learn an effective base distribution of meta-stable crystals in a text representation. After converting to a graph representation, the RFM model takes samples from the LLM and iteratively refines the coordinates and lattice parameters. Our approach significantly outperforms state-of-the-art methods, increasing the generation rate of stable materials by over three times and increasing the rate for stable, unique, and novel crystals by $\sim50\%$ - a huge improvement on a difficult problem. Additionally, the crystals generated by FlowLLM are much closer to their relaxed state when compared with another leading model, significantly reducing post-hoc computational cost.
title FlowLLM: Flow Matching for Material Generation with Large Language Models as Base Distributions
topic Machine Learning
Materials Science
Artificial Intelligence
url https://arxiv.org/abs/2410.23405