arXiv AI

uFlowCSP: Crystal Structure Prediction using Mean flow generative models

uFlowCSP is a MeanFlow-based crystal structure prediction model that learns the average probability‑flow velocity, enabling it to generate complete crystal structures in one to five network evaluations. It achieves inference speeds 5×–58× faster than diffusion and flow‑matching methods while matching or surpassing their performance, with a chemistry‑ and symmetry‑aware Transformer that uses canonical atom ordering and per‑token chemistry embeddings. On the MP‑20 benchmark, uFlowCSP attains comparable or higher accuracy with far fewer evaluations and significantly lower wall‑clock time, demonstrating improved accuracy per network evaluation.

arXiv Machine Learning
Sep 23

Deep Generative Crystal Structure Prediction: A Benchmark Study and a Controlled Test of Prototype Dependence

The study benchmarks 12 deep generative crystal structure prediction models against the template-based TCSP 2.0 on 180 test structures, finding that template retrieval achieves the highest top‑1 success (68.3%). Most generative predictions overlap with template substitutions, and removing entire stoichiometric prototype families from training reduces accuracy by 50‑78%, indicating strong prototype dependence. Only a small subset of predictions remain after such removal, suggesting limited genuine de‑novo capability.

By Lai Wei, Rongzhi Dong, Ying Feng, Madeline Miklos, Jianjun Hu
arXiv Machine Learning
6d ago

TCSP3.0: Template-Based Crystal Structure Prediction with Coefficient-Preserving Element Substitution and Structural Diversity

TCSP3.0 is an improved template‑based crystal structure prediction method that enforces coefficient‑preserving element mappings, substitutes species simultaneously on parsed structures, and validates each candidate’s composition and cell. It adds periodic‑number retrieval, soft oxidation‑state evidence, radius‑compatibility scoring, structural deduplication, and ORB‑v3 machine‑learning‑potential relaxation and energy ranking, achieving Top‑1 and Top‑20 match rates of 72.2 % and 90.6 % on the T180 benchmark and outperforming previous TCSP2.0 and several deep generative models. The method runs in a median of 28 s per target on two CPU cores, roughly ten times faster than CrystalFlow, with most gains attributed to the correctness fixes and structural deduplication.

By Ying Feng, Jeffrey Hu, Lai Wei, Liang Tao, Madeline Miklos, Dipannoy Das Gupta, Hossein Erfanshekooh, Jianjun Hu
arXiv AI
Sep 30

Where Should Physics Enter a Molecular Crystal Generator?

The paper introduces CrystAF, an all‑atom crystal flow‑map generation model, and evaluates where physics should be incorporated into generative crystal structure models. By applying physics‑informed post‑training, the authors improve molecular validity and crystal packing without altering sampling speed, while inference‑time corrections further refine the structures. The study demonstrates that post‑training and inference‑time physics are complementary, and that the post‑training approach transfers to other generators such as Clari‑M and MolCrystalFlow.

By Haocheng Tang, Junmei Wang, Wengong Jin
arXiv Machine Learning
Aug 28

Packora: Systematic Design for Generative Molecular Crystal Structure Prediction

Packora is a flow-based generative model designed for molecular crystal structure prediction (CSP). It jointly predicts atomic coordinates and lattice parameters from molecular graphs, supporting multi-component and organometallic crystals and allowing conditioning on conformers, stereochemistry, and space-group data. In evaluations inspired by the CCDC CSP blind test, Packora outperforms baselines on generation and ranking benchmarks, achieving superior matched-budget coverage, higher experimental-form recovery, lower ranks, and faster convergence.

By Nayoung Kim, Kiyoung Seong, Sungsoo Ahn
arXiv Machine Learning
Oct 2

Riemannian Flow Models with Reinforcement Learning for Molecular Crystal Structure Prediction

The paper introduces CG-OMatG, an equivariant Riemannian flow-based generative model that predicts molecular crystal structures using a coarse-grained, hierarchical representation. It treats molecules as rigid bodies, performs inter- and intra-molecular message passing, and learns to reconstruct molecule centroids, orientations, and lattice parameters conditioned on chemical species and conformer geometry. The model is trained on OMC25 and CSD datasets, fine-tuned with policy gradient reinforcement learning to favor low-energy structures, and validated against a CSP blind test benchmark using COMPACK packing-similarity analysis.

By Thomas Egg, Harry Winston Sullivan, Maya M. Martirossyan, Philipp H\"ollmer, Cheng Zeng, Adrian Roitberg, Mingjie Liu, Richard Hennig, Sapna Sarupria, Ellad B. Tadmor, Stefano Martiniani
arXiv Machine Learning
Jul 7

FastCSP: Accelerated Molecular Crystal Structure Prediction with Universal Model for Atoms

arXiv:2508. 02641v2 Announce Type: replace-cross Abstract: Molecular crystal structure prediction (CSP) is essential for applications in pharmaceuticals and organic electronics.

By Vahe Gharakhanyan, Yi Yang, Luis Barroso-Luque, Daniel S. Levine, Sushree Jagriti Sahoo, Brandon M. Wood, Kyle Michel, Muhammed Shuaibi, Gregory J. O. Beran, Viachaslau Bernat, Misko Dzamba, Xiang Fu, Meng Gao, Xingyu Liu, Benjamin K. Miller, Keian Noori, Lafe J. Purvis, Tingling Rao, Ammar Rizvi, Matt Uyttendaele, Andrew J. Ouderkirk, Chiara Daraio, C. Lawrence Zitnick, Arman Boromand, Noa Marom, Zachary W. Ulissi, Anuroop Sriram
arXiv Machine Learning
6d ago

CrystalREPA: Transferring Physical Priors from Universal MLIPs to Crystal Generative Models

CrystalREPA is a plug‑and‑play framework that aligns the atom‑wise hidden states of crystal generative encoders with frozen universal machine‑learning interatomic potential (MLIP) representations using an element‑aware contrastive objective. By doing so, it transfers stability‑aware atomistic priors to generative models with minimal training overhead and no extra inference cost, consistently improving thermodynamic stability, structural validity, and fidelity across multiple generative frameworks and benchmark datasets. The study also shows that an MLIP’s suitability as a teacher is better predicted by the distinguishability of its atom‑wise representation space than by its leaderboard accuracy, providing a practical, accuracy‑independent selection criterion. whyItMatters":"The approach demonstrates a practical method to enhance crystal generative models’ ability to produce stable, realistic structures by leveraging existing MLIP knowledge without incurring additional inference costs."

By Chengqian Zhang, Yucheng Jin, Duo Zhang, Tiejun Li, Han Wang