用虚拟粒子衍射图提升晶体性质预测精度
Beyond Structure: Invariant Crystal Property Prediction with Pseudo-Particle Ray Diffraction
- 结合图结构与虚拟粒子衍射图表示晶体
- 在多个数据集上达到当前最佳性能
- 适合材料科学与机器学习交叉研究者
晶体性质预测受量子力学支配,传统密度泛函理论对大规模多体系统计算成本过高。尽管机器学习模型可高效近似,但其性能高度依赖原子表示方式。现有基于图的方法虽逐步引入更多结构信息,却因感受野有限和局部编码难以捕捉长程原子相互作用,导致不同晶体被映射为相同表示,影响预测准确率。为此,我们提出PRDNet,融合独特的倒空间衍射信息与图表示。通过数据驱动的伪粒子生成合成衍射图,增强对元素及环境变化的敏感性。PRDNet确保对晶格对称性的完全不变性。在Materials Project、JARVIS-DFT和MatBench数据集上进行大量实验,结果表明该模型达到当前最优性能。代码已开源:https://github.com/Bin-Cao/PRDNet。
原文摘要 · Abstract (English)
Crystal property prediction, governed by quantum mechanical principles, is computationally prohibitive to solve exactly for large many-body systems using traditional density functional theory. While machine learning models have emerged as efficient approximations for large-scale applications, their performance is strongly influenced by the choice of atomic representation. Although modern graph-based approaches have progressively incorporated more structural information, they often fail to capture long-range atomic interactions due to finite receptive fields and local encoding schemes. This limitation leads to distinct crystals being mapped to identical representations, hindering accurate property prediction. To address this, we introduce PRDNet that leverages unique reciprocal-space diffraction besides graph representations. To enhance sensitivity to elemental and environmental variations, we employ a data-driven pseudo-particle to generate a synthetic diffraction pattern. PRDNet ensures full invariance to crystallographic symmetries. Extensive experiments are conducted on Materials Project, JARVIS-DFT, and MatBench, demonstrating that the proposed model achieves state-of-the-art performance. The code is openly available at https://github.com/Bin-Cao/PRDNet.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。