In modern computational materials science, deep learning has shown the capability to predict interatomic potentials, thereby supporting and accelerating conventional simulations. However, existing models typically sacrifice either accuracy or efficiency. Moreover, lightweight models are highly demanded for offering simulating systems on a considerably larger scale at reduced computational costs. A century ago, Felix Bloch demonstrated how leveraging the equivariance of the translation operation on a crystal lattice (with geometric symmetry) could significantly reduce the computational cost of determining wavefunctions and accurately calculate material properties. Here, we introduce a lightweight equivariant interaction graph neural network (LEIGNN) that can enable accurate and efficient interatomic potential and force predictions in crystals. Rather than relying on higher-order representations, LEIGNN employs a scalar-vector dual representation to encode equivariant features. By extracting both local and global structures from vector representations and learning geometric symmetry information, our model remains lightweight while ensuring prediction accuracy and robustness through the equivariance. Our results show that LEIGNN consistently outperforms the prediction performance of the representative baselines and achieves significant efficiency across diverse datasets, which include catalysts, molecules, and organic isomers. Finally, to further validate the predicted interatomic potentials from our model, we conduct classical molecular dynamics (MD) and ab initio MD simulation across various systems, including solid, liquid, and gas. It is found that LEIGNN can achieve the accuracy of ab initio MD and retain the computational efficiency of classical MD across all examined systems, demonstrating its accuracy, efficiency, and universality.
翻译:在现代计算材料科学中,深度学习已展现出预测原子间势能的能力,从而支持并加速传统模拟。然而,现有模型通常会在准确性或效率上有所牺牲。此外,轻量化模型因能以较低计算成本模拟更大规模系统而备受青睐。一个世纪前,Felix Bloch证明了利用晶格上平移操作的等变性(具有几何对称性)可显著降低波函数确定过程的计算成本,并精确计算材料性质。在此,我们提出一种轻量等变交互图神经网络(LEIGNN),能够实现晶体中原子间势能和力的准确高效预测。LEIGNN不依赖高阶表示,而是采用标量-矢量双表示来编码等变特征。通过从矢量表示中提取局部和全局结构并学习几何对称信息,我们的模型在保持轻量化的同时,通过等变性确保了预测准确性和鲁棒性。结果表明,LEIGNN在多个数据集(包括催化剂、分子和有机异构体)上的预测性能持续优于代表性基线模型,并实现了显著的高效性。最后,为进一步验证模型预测的原子间势能,我们在固体、液体和气体等多种体系中进行了经典分子动力学(MD)和从头算MD模拟。研究发现,LEIGNN在所有测试体系中均能达到从头算MD的准确度,并保持经典MD的计算效率,证明了其准确性、高效性和普适性。