摘要
几何图神经网络(GNN)已成为建模分子几何的强大工具。然而,在有效捕获大型分子系统中的长程相互作用方面,它们遇到了局限性。为了解决这一挑战,我们引入了神经 P$^3$M,这是一种通用的几何 GNN 增强器,通过将网格点与原子一起纳入并以可训练的方式重新构想传统数学运算,扩展了其功能范围。神经 P$^3$M 在各种分子系统中表现出灵活性,并在预测能量和力方面表现出非凡的准确性,在 MD22 数据集等基准测试中优于其他方法。它还与各种架构集成,在 OE62 数据集上实现了平均 22% 的改进。