Periodic Materials Generation using Text-Guided Joint Diffusion Model

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Das, Kishalay, Khastagir, Subhojyoti, Goyal, Pawan, Lee, Seung-Cheol, Bhattacharjee, Satadeep, Ganguly, Niloy
Formato: Preprint
Publicado: 2025
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866929738141925376
author Das, Kishalay
Khastagir, Subhojyoti
Goyal, Pawan
Lee, Seung-Cheol
Bhattacharjee, Satadeep
Ganguly, Niloy
author_facet Das, Kishalay
Khastagir, Subhojyoti
Goyal, Pawan
Lee, Seung-Cheol
Bhattacharjee, Satadeep
Ganguly, Niloy
contents Equivariant diffusion models have emerged as the prevailing approach for generating novel crystal materials due to their ability to leverage the physical symmetries of periodic material structures. However, current models do not effectively learn the joint distribution of atom types, fractional coordinates, and lattice structure of the crystal material in a cohesive end-to-end diffusion framework. Also, none of these models work under realistic setups, where users specify the desired characteristics that the generated structures must match. In this work, we introduce TGDMat, a novel text-guided diffusion model designed for 3D periodic material generation. Our approach integrates global structural knowledge through textual descriptions at each denoising step while jointly generating atom coordinates, types, and lattice structure using a periodic-E(3)-equivariant graph neural network (GNN). Extensive experiments using popular datasets on benchmark tasks reveal that TGDMat outperforms existing baseline methods by a good margin. Notably, for the structure prediction task, with just one generated sample, TGDMat outperforms all baseline models, highlighting the importance of text-guided diffusion. Further, in the generation task, TGDMat surpasses all baselines and their text-fusion variants, showcasing the effectiveness of the joint diffusion paradigm. Additionally, incorporating textual knowledge reduces overall training and sampling computational overhead while enhancing generative performance when utilizing real-world textual prompts from experts.
format Preprint
id arxiv_https___arxiv_org_abs_2503_00522
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Periodic Materials Generation using Text-Guided Joint Diffusion Model
Das, Kishalay
Khastagir, Subhojyoti
Goyal, Pawan
Lee, Seung-Cheol
Bhattacharjee, Satadeep
Ganguly, Niloy
Machine Learning
Materials Science
Equivariant diffusion models have emerged as the prevailing approach for generating novel crystal materials due to their ability to leverage the physical symmetries of periodic material structures. However, current models do not effectively learn the joint distribution of atom types, fractional coordinates, and lattice structure of the crystal material in a cohesive end-to-end diffusion framework. Also, none of these models work under realistic setups, where users specify the desired characteristics that the generated structures must match. In this work, we introduce TGDMat, a novel text-guided diffusion model designed for 3D periodic material generation. Our approach integrates global structural knowledge through textual descriptions at each denoising step while jointly generating atom coordinates, types, and lattice structure using a periodic-E(3)-equivariant graph neural network (GNN). Extensive experiments using popular datasets on benchmark tasks reveal that TGDMat outperforms existing baseline methods by a good margin. Notably, for the structure prediction task, with just one generated sample, TGDMat outperforms all baseline models, highlighting the importance of text-guided diffusion. Further, in the generation task, TGDMat surpasses all baselines and their text-fusion variants, showcasing the effectiveness of the joint diffusion paradigm. Additionally, incorporating textual knowledge reduces overall training and sampling computational overhead while enhancing generative performance when utilizing real-world textual prompts from experts.
title Periodic Materials Generation using Text-Guided Joint Diffusion Model
topic Machine Learning
Materials Science
url https://arxiv.org/abs/2503.00522