Kinetic Langevin Diffusion for Crystalline Materials Generation

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Cornet, François, Bergamin, Federico, Bhowmik, Arghya, Lastra, Juan Maria Garcia, Frellsen, Jes, Schmidt, Mikkel N.
Formato: Preprint
Publicado: 2025
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866915372041502720
author Cornet, François
Bergamin, Federico
Bhowmik, Arghya
Lastra, Juan Maria Garcia
Frellsen, Jes
Schmidt, Mikkel N.
author_facet Cornet, François
Bergamin, Federico
Bhowmik, Arghya
Lastra, Juan Maria Garcia
Frellsen, Jes
Schmidt, Mikkel N.
contents Generative modeling of crystalline materials using diffusion models presents a series of challenges: the data distribution is characterized by inherent symmetries and involves multiple modalities, with some defined on specific manifolds. Notably, the treatment of fractional coordinates representing atomic positions in the unit cell requires careful consideration, as they lie on a hypertorus. In this work, we introduce Kinetic Langevin Diffusion for Materials (KLDM), a novel diffusion model for crystalline materials generation, where the key innovation resides in the modeling of the coordinates. Instead of resorting to Riemannian diffusion on the hypertorus directly, we generalize Trivialized Diffusion Model (TDM) to account for the symmetries inherent to crystals. By coupling coordinates with auxiliary Euclidean variables representing velocities, the diffusion process is now offset to a flat space. This allows us to effectively perform diffusion on the hypertorus while providing a training objective that accounts for the periodic translation symmetry of the true data distribution. We evaluate KLDM on both Crystal Structure Prediction (CSP) and De-novo Generation (DNG) tasks, demonstrating its competitive performance with current state-of-the-art models.
format Preprint
id arxiv_https___arxiv_org_abs_2507_03602
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Kinetic Langevin Diffusion for Crystalline Materials Generation
Cornet, François
Bergamin, Federico
Bhowmik, Arghya
Lastra, Juan Maria Garcia
Frellsen, Jes
Schmidt, Mikkel N.
Machine Learning
Generative modeling of crystalline materials using diffusion models presents a series of challenges: the data distribution is characterized by inherent symmetries and involves multiple modalities, with some defined on specific manifolds. Notably, the treatment of fractional coordinates representing atomic positions in the unit cell requires careful consideration, as they lie on a hypertorus. In this work, we introduce Kinetic Langevin Diffusion for Materials (KLDM), a novel diffusion model for crystalline materials generation, where the key innovation resides in the modeling of the coordinates. Instead of resorting to Riemannian diffusion on the hypertorus directly, we generalize Trivialized Diffusion Model (TDM) to account for the symmetries inherent to crystals. By coupling coordinates with auxiliary Euclidean variables representing velocities, the diffusion process is now offset to a flat space. This allows us to effectively perform diffusion on the hypertorus while providing a training objective that accounts for the periodic translation symmetry of the true data distribution. We evaluate KLDM on both Crystal Structure Prediction (CSP) and De-novo Generation (DNG) tasks, demonstrating its competitive performance with current state-of-the-art models.
title Kinetic Langevin Diffusion for Crystalline Materials Generation
topic Machine Learning
url https://arxiv.org/abs/2507.03602