用双模型协同生成晶体结构,速度比传统方法快2000倍
Siamese Foundation Models for Crystal Structure Prediction
- 构建生成器与能量预测器的双塔框架,通过相互优化提升生成质量
- 在3个真实超导体上实现100%结构匹配,原子位置误差仅0.0012
- 适合材料发现领域研究者,尤其关注高效结构预测的团队
从化学组成预测晶体结构是材料发现中的核心挑战,因其复杂的三维几何特性而区别于蛋白质折叠等领域。本文提出基于扩散的晶体通用模型DAO,采用双塔基础模型架构:结构生成器与能量预测器。生成器在大规模稳定与非稳定结构数据集上分两阶段预训练,利用预测器对非稳定构型进行松弛并引导生成采样。在两个经典基准测试中,预训练显著提升多种骨干网络的性能。消融实验表明生成器与预测器间存在相互促进效应。进一步验证于三个实际超导体(Cr₆Os₂、Zr₁₆Rh₈O₄、Zr₁₆Pd₈O₄),传统计算难以触及。对于Cr₆Os₂,DAO在20次生成下达到100%实验参考匹配率,原子位置误差0.0012,单次迭代速度比基于DFT的预测器快2000倍以上。这些结果凸显该方法在推动材料科学研究方面的巨大潜力。
原文摘要 · Abstract (English)
Predicting crystal structures from chemical compositions is a fundamental challenge in materials discovery, complicated by complex 3D geometries that distinguish it from fields like protein folding. Here, we present Diffusion-based Crystal Omni (DAO), a pretrain-finetune framework for crystal structure prediction integrating two Siamese foundation models: a structure generator and an energy predictor. The generator is pretrained via a two-stage pipeline on a vast dataset of stable and unstable structures, leveraging the predictor to relax unstable configurations and guide the generative sampling. Across two well-known benchmarks, pretraining significantly enhances performance across multiple backbone architectures. Ablation studies confirm that the synergy between the generator and predictor mutually benefits both components. We further validate DAO on three real-world superconductors ($\text{Cr}_6\text{Os}_2$, $\text{Zr}_{16}\text{Rh}_8\text{O}_4$, and $\text{Zr}_{16}\text{Pd}_8\text{O}_4$) typically inaccessible to conventional computation. For $\text{Cr}_6\text{Os}_2$, DAO achieves a 100\% match rate with experimental references and an atomic-position error of 0.0012 under 20-shot generation, performing over 2000$\times$ faster per iteration than DFT-based structure predictors. These compelling results collectively highlight the potential of our approach for advancing materials science research.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。