Jan, 2024

混合串联回归框架的晶态材料研究中调和协方差性和表达性的深哈密顿回归

TL;DR深度学习用于材料研究中的哈密顿回归需要满足协方差定律,其中在保证网络表达能力的前提下实现 SO (3) 等变性仍然是个困难的挑战。为了解决协方差 - 表达能力两难问题,我们提出了一个混合框架,利用两个级联回归阶段。第一个阶段,利用一个理论上保证协变的神经网络对 3D 原子系统的对称性建模,产生理论上协变的特征和基线哈密顿预测,辅助第二阶段在学习协变性方面。与此同时,第二阶段由我们提出的非线性 3D 图 Transformer 网络驱动,用于对 3D 原子系统进行结构建模,将第一阶段的输出细化为具有更好表达能力的哈密顿精细预测。理论上协变但不可避免地表达能力较差的模型与高度表达能力的非线性网络的组合,能够实现精确、可推广的预测,并在坐标变换下保持稳健的协方差。我们的方法在电子结构计算的哈密顿预测方面达到了最先进的性能,在五个晶体材料数据库上进行的实验证实了这一点。