arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2201.00802physics.chem-ph
收藏

用等变Transformer在ANI-1x数据集上学习小分子能量和原子间力

Learning Small Molecule Energies and Interatomic Forces with an Equivariant Transformer on the ANI-1x Dataset

Bryce Hedelius, Fabian B. Fuchs, Dennis Della Corte

摘要:

精确预测原子间的相互作用能量和作用力是高质量分子动力学模拟的基础。机器学习算法可用于通过预测从头算质量能量和力来克服经典MD的局限性。SE(3)-等变神经网络允许对空间关系进行推理,并利用旋转和平移对称性。其中一个算法是SE(3)-Transformer,我们将其应用于ANI-1x数据集。我们早期的实验结果表明,通过消融研究,更深的网络-具有额外的SE(3)-Transformer层-可以达到必要的精度,以允许与MD有效集成。然而,SE(3)-Transformer的更快的实现将是必需的,例如Milesi最近发布的加速版本。

英文摘要:

Accurate predictions of interatomic energies and forces are essential for high quality molecular dynamic simulations (MD). Machine learning algorithms can be used to overcome limitations of classical MD by predicting ab initio quality energies and forces. SE(3)-equivariant neural network allow reasoning over spatial relationships and exploiting the rotational and translational symmetries. One such algorithm is the SE(3)-Transformer, which we adapt for the ANI-1x dataset. Our early experimental results indicate through ablation studies that deeper networks - with additional SE(3)-Transformer layers - could reach necessary accuracies to allow effective integration with MD. However, faster implementations of the SE(3)-Transformer will be required, such as the recently published accelerated version by Milesi.

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

↑