提出笛卡尔张量框架,让机器学习势函数更自然地处理原子坐标。
A Cartesian-3j Framework for Machine Learning Interatomic Potentials
- 用笛卡尔张量替代球谐张量,直接对齐原子坐标和张量目标
- 在相同架构下对比显示笛卡尔模型精度与球谐相当
- 新模型TACE-v1-OAM-M在材料发现任务中表现领先,适合高效分子模拟
机器学习原子间势函数(MLIPs)显著提升了计算化学中的外推能力。然而,大多数等变模型基于球谐张量(STs)构建,而笛卡尔张量因与原子坐标和张量目标天然契合,却发展不足。本文提出一种不可约笛卡尔张量(ICTs)的笛卡尔框架,引入\texttt{Cartesian-3j}符号与笛卡尔广义克莱布什-戈登系数,作为球谐张量耦合中\texttt{Wigner-3j}符号的直接类比。我们扩展了\texttt{e3nn}库以支持ICT乘积,并基于该框架构建了\texttt{MACE}、\texttt{NequIP}和\texttt{Allegro}的笛卡尔版本,实现首次在固定架构下仅改变张量基的可控比较。实验表明,不可约笛卡尔模型可达到与球谐模型相当的精度,但直接笛卡尔化带来不利的计算与内存开销,推动专用笛卡尔架构设计。利用ICTs与本框架,我们提出\texttt{TACE-v1-OAM-M},在Matbench Discovery任务上表现优于现有球谐模型,具备竞争力。
原文摘要 · Abstract (English)
Machine learning interatomic potentials (MLIPs) have brought substantial gains in the extrapolation capability in computational chemistry. However, most equivariant models are typically built with spherical tensors (STs), while Cartesian tensor formulations remain less developed despite their natural alignment with atomic coordinates and tensorial targets. In this work, we develop a Cartesian framework for irreducible Cartesian tensors (ICTs) by introduce the \texttt{Cartesian-3j} symbol and Cartesian Generalized Clebsch-Gordan Coefficients, which serve as direct analogues of the \texttt{Wigner-3j} symbol and Generalized Clebsch-Gordan coefficients defined for ST coupling. We extend the \texttt{e3nn} library to support ICT product, and use this framework to build Cartesian counterparts of \texttt{MACE}, \texttt{NequIP}, and \texttt{Allegro}, allowing the first controlled comparison where architectures are held fixed and only the tensor basis is changed. Our experiments show that irreducible Cartesian models can achieve accuracy comparable to spherical counterparts, but direct Cartesianization incurs unfavorable compute and memory scaling, motivating dedicated Cartesian architectural choices. Leveraging ICTs and our framework, we introduce \texttt{TACE-v1-OAM-M} and demonstrate that it achieves competitive performance on Matbench Discovery compared to state-of-the-art ST models.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。