Site-orbit masked twin-branch Transformer for symmetry-constrained crystal generation
Abstract
Generative models for crystal structure prediction (CSP) have expanded the search space for novel materials and improved screening efficiency. However, simultaneously generating stable and novel structures remains challenging because of the complex structural and physical constraints governing crystals. To address this challenge, we propose site-orbit masked twin-branch Transformer model (ST2M), a symmetry-guided discrete--continuous diffusion framework incorporating degree-of-freedom masks and conditioned on Wyckoff patterns. ST2M represents crystal structures as sequences of independent occupied Wyckoff orbits and employs a twin-branch Transformer to jointly denoise discrete and continuous variables. In our experiments, ST2M achieves a structural validity of 96.37\%. For known patterns in dataset, 23.86\% of generated structures simultaneously satisfy stability, uniqueness, and novelty criteria, representing a relative improvement of 68.15\% over the best-performing baseline. First-principles calculations identify a total of 21 previously unreported stable structures among selected candidates with low predicted energies. These results demonstrate that ST2M can generate chemically viable crystal candidates under prescribed crystallographic constraints and generalize to unknown Wyckoff patterns, providing a practical route to accelerated materials discovery.
Related
- SymADiT: Symmetry-Aware Crystal Generation with Latent Flow Matching
- GEODE: Symmetry-Preserving Cartesian Diffusion for Crystal Generation
- Symmetry- and Property-Aware Crystal Generation with Reinforcement Learning for Inverse Materials Design
- Automated mode-aware design of nonlinear photonic crystal cavities for intrinsic second harmonic generation
- 266 nm generation performance of NSBBF crystal improved by elevating the operating temperature
BibTeX
@misc{liu2026siteorbit,
title = {{Site-orbit masked twin-branch Transformer for symmetry-constrained crystal generation}},
author = {Junchi Liu and Haoran Si and Luo Yan and Sergei Tretiak and Liujiang Zhou},
howpublished = {arXiv},
year = {2026},
month = nov,
doi = {10.48550/arxiv.2611.02906},
eprint = {2611.02906},
archivePrefix = {arXiv},
url = {https://arxiv.org/abs/2611.02906},
}
RIS
TY - UNPB
TI - Site-orbit masked twin-branch Transformer for symmetry-constrained crystal generation
AU - Junchi Liu
AU - Haoran Si
AU - Luo Yan
AU - Sergei Tretiak
AU - Liujiang Zhou
PY - 2026
DA - 2026/11/02
PB - arXiv
DO - 10.48550/arxiv.2611.02906
UR - https://arxiv.org/abs/2611.02906
AB - Generative models for crystal structure prediction (CSP) have expanded the search space for novel materials and improved screening efficiency. However, simultaneously generating stable and novel structures remains challenging because of the complex structural and physical constraints governing crystals. To address this challenge, we propose site-orbit masked twin-branch Transformer model (ST2M), a symmetry-guided discrete--continuous diffusion framework incorporating degree-of-freedom masks and conditioned on Wyckoff patterns. ST2M represents crystal structures as sequences of independent occupied Wyckoff orbits and employs a twin-branch Transformer to jointly denoise discrete and continuous variables. In our experiments, ST2M achieves a structural validity of 96.37\%. For known patterns in dataset, 23.86\% of generated structures simultaneously satisfy stability, uniqueness, and novelty criteria, representing a relative improvement of 68.15\% over the best-performing baseline. First-principles calculations identify a total of 21 previously unreported stable structures among selected candidates with low predicted energies. These results demonstrate that ST2M can generate chemically viable crystal candidates under prescribed crystallographic constraints and generalize to unknown Wyckoff patterns, providing a practical route to accelerated materials discovery.
ER -