Saved in:
Bibliographic Details
Main Authors: Gerolymatos, Stavros, Brubaker, J. Kyle, Schuetz, Martin J. A., Gusev, Vladimir V.
Format: Preprint
Published: 2026
Subjects:
Online Access:https://arxiv.org/abs/2604.23921
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866911625266593792
author Gerolymatos, Stavros
Brubaker, J. Kyle
Schuetz, Martin J. A.
Gusev, Vladimir V.
author_facet Gerolymatos, Stavros
Brubaker, J. Kyle
Schuetz, Martin J. A.
Gusev, Vladimir V.
contents Crystalline materials are widely used in technological applications, yet their discovery remains a significant challenge. As their properties are driven by structure, crystal structure prediction (CSP) methods play a central role in computational approaches aiming to accelerate this process. Previously, CSP has been approached from a combinatorial optimization perspective, with the core challenge of allocating atoms on a fine grid of predefined discrete positions within a unit cell while minimizing their interaction energy. Exact mathematical optimization methods provide guaranteed solutions, but they become computationally expensive for large-scale instances, where the atomic configuration space grows rapidly, particularly in the absence of additional symmetry constraints. In this work, we introduce a neural combinatorial optimization approach to the atom allocation challenge and, subsequently, CSP, based on graph neural networks (GNNs), which can effectively sample from the distribution of feasible structures in an unsupervised manner. We leverage expander graphs to construct computational graphs over discrete positions that capture both short- and long-range interactions between atoms, and employ the Gumbel-Sinkhorn approach to enforce the desired stoichiometry of the generated structures. We demonstrate that our method outperforms classical heuristic approaches and is competitive with a commercial optimization solver across a range of chemical compositions. This enables the use of ever-expanding GPU infrastructure to tackle the inherent combinatorial challenges of CSP, paving the way for scaling beyond current capabilities.
format Preprint
id arxiv_https___arxiv_org_abs_2604_23921
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Crystal structure prediction using graph neural combinatorial optimization
Gerolymatos, Stavros
Brubaker, J. Kyle
Schuetz, Martin J. A.
Gusev, Vladimir V.
Machine Learning
Artificial Intelligence
Crystalline materials are widely used in technological applications, yet their discovery remains a significant challenge. As their properties are driven by structure, crystal structure prediction (CSP) methods play a central role in computational approaches aiming to accelerate this process. Previously, CSP has been approached from a combinatorial optimization perspective, with the core challenge of allocating atoms on a fine grid of predefined discrete positions within a unit cell while minimizing their interaction energy. Exact mathematical optimization methods provide guaranteed solutions, but they become computationally expensive for large-scale instances, where the atomic configuration space grows rapidly, particularly in the absence of additional symmetry constraints. In this work, we introduce a neural combinatorial optimization approach to the atom allocation challenge and, subsequently, CSP, based on graph neural networks (GNNs), which can effectively sample from the distribution of feasible structures in an unsupervised manner. We leverage expander graphs to construct computational graphs over discrete positions that capture both short- and long-range interactions between atoms, and employ the Gumbel-Sinkhorn approach to enforce the desired stoichiometry of the generated structures. We demonstrate that our method outperforms classical heuristic approaches and is competitive with a commercial optimization solver across a range of chemical compositions. This enables the use of ever-expanding GPU infrastructure to tackle the inherent combinatorial challenges of CSP, paving the way for scaling beyond current capabilities.
title Crystal structure prediction using graph neural combinatorial optimization
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2604.23921