arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2309.04475cond-mat.mtrl-scics.LG
收藏

基于联合等变扩散的晶体结构预测

Crystal Structure Prediction by Joint Equivariant Diffusion

Tsinghua University(清华大学);Institute for AI, Tsinghua University(清华大学人工智能研究院);Institute for AIR, Tsinghua University(清华大学AIR研究院);Gaoling School of Artificial Intelligence, Renmin University of China(中国人民大学高瓴人工智能学院);National Supercomputer Center in Guangzhou(广州国家超级计算中心);School of Computer Science and Engineering, Sun Yat-sen University(中山大学计算机科学与工程学院);Stanford University(斯坦福大学)

Rui Jiao, Wenbing Huang, Peijia Lin, Jiaqi Han, Pin Chen, Yutong Lu, Yang Liu

AI总结:

针对晶体结构预测中对称几何带来的挑战,提出采用分数坐标表示、结合周期性E(3)等变去噪的DiffCSP扩散模型,联合生成晶格与原子坐标,性能优于现有方法且计算成本更低,还可扩展至从头算晶体生成。

摘要:

晶体结构预测(Crystal Structure Prediction, CSP)在诸多科学学科中至关重要。尽管可采用当前主流的生成模型(如扩散模型)解决CSP问题,但由于晶体结构的对称几何特性——平移、旋转和周期性不变性,该任务面临独特挑战。为融合上述对称性,本文提出DiffCSP,这是一种从稳定晶体中学习结构分布的新型扩散模型。具体而言,DiffCSP通过采用周期性E(3)等变去噪模型,联合生成每个晶体的晶格和原子坐标,以更好地建模晶体几何结构。值得注意的是,与相关等变生成方法不同,DiffCSP利用分数坐标而非笛卡尔坐标来表示晶体,显著提升了原子位置的扩散和生成过程。大量实验验证,我们的DiffCSP显著优于现有CSP方法,且与基于DFT的方法相比计算成本低得多。此外,当扩展用于从头算晶体生成时,也观察到DiffCSP的优越性。

英文摘要:

Crystal Structure Prediction (CSP) is crucial in various scientific disciplines. While CSP can be addressed by employing currently-prevailing generative models (e.g. diffusion models), this task encounters unique challenges owing to the symmetric geometry of crystal structures -- the invariance of translation, rotation, and periodicity. To incorporate the above symmetries, this paper proposes DiffCSP, a novel diffusion model to learn the structure distribution from stable crystals. To be specific, DiffCSP jointly generates the lattice and atom coordinates for each crystal by employing a periodic-E(3)-equivariant denoising model, to better model the crystal geometry. Notably, different from related equivariant generative approaches, DiffCSP leverages fractional coordinates other than Cartesian coordinates to represent crystals, remarkably promoting the diffusion and the generation process of atom positions. Extensive experiments verify that our DiffCSP significantly outperforms existing CSP methods, with a much lower computation cost in contrast to DFT-based methods. Moreover, the superiority of DiffCSP is also observed when it is extended for ab initio crystal generation.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑