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

一个通用的原子系统特征化框架

A Universal Framework for Featurization of Atomistic Systems

Georgia Institute of Technology(佐治亚理工学院)

Xiangyun Lei, Andrew J. Medford

摘要:

分子动力学模拟在许多科学领域是一种非常宝贵的工具。然而,无处不在的经典力场不能描述反应系统,量子分子动力学的计算要求太高,无法处理大型系统或长时间尺度。基于物理学或机器学习的反作用力场可用于弥合时间和长度尺度上的差距,但这些力场需要大量努力来构建,并且对给定的化学成分和应用具有高度特异性。机器学习模型的一个重要限制是使用特定于元素的特征,导致模型随元素数量的扩展性很差。这项工作介绍了高斯多极(GMP)featurization方案,利用物理相关的多极原子周围的电子密度的扩展,以产生特征向量,插值元素类型之间,并具有固定的尺寸,无论元素的数量存在。我们将GMP与神经网络相结合,将其与MD 17数据集广泛使用的Behler-Parinello对称函数进行直接比较,发现它具有更高的准确性和计算效率。此外,我们证明了基于GMP的模型可以实现QM 9数据集的化学准确性,即使外推到新元素,它们的准确性仍然是合理的。最后,我们针对开放催化项目(OCP)数据集测试了基于GMP的模型,显示出与图卷积深度学习模型相当的性能。结果表明,该特征化方案填补了构建高效和可转移的机器学习力场的关键空白。

英文摘要:

Molecular dynamics simulations are an invaluable tool in numerous scientific fields. However, the ubiquitous classical force fields cannot describe reactive systems, and quantum molecular dynamics are too computationally demanding to treat large systems or long timescales. Reactive force fields based on physics or machine learning can be used to bridge the gap in time and length scales, but these force fields require substantial effort to construct and are highly specific to a given chemical composition and application. A significant limitation of machine learning models is the use of element-specific features, leading to models that scale poorly with the number of elements. This work introduces the Gaussian multipole (GMP) featurization scheme that utilizes physically-relevant multipole expansions of the electron density around atoms to yield feature vectors that interpolate between element types and have a fixed dimension regardless of the number of elements present. We combine GMP with neural networks to directly compare it to the widely used Behler-Parinello symmetry functions for the MD17 dataset, revealing that it exhibits improved accuracy and computational efficiency. Further, we demonstrate that GMP-based models can achieve chemical accuracy for the QM9 dataset, and their accuracy remains reasonable even when extrapolating to new elements. Finally, we test GMP-based models for the Open Catalysis Project (OCP) dataset, revealing comparable performance to graph convolutional deep learning models. The results indicate that this featurization scheme fills a critical gap in the construction of efficient and transferable machine-learned force fields.

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

↑