E(n)-Equivariant Cartesian Tensor Passing Potential (2402.15286v1)

Abstract: Machine learning potential (MLP) has been a popular topic in recent years for its potential to replace expensive first-principles calculations in some large systems. Meanwhile, message passing networks have gained significant attention due to their remarkable accuracy, and a wave of message passing networks based on Cartesian coordinates has emerged. However, the information of the node in these models is limited to scalars, vectors, and tensors. In this work, we proposed High-order Tensor Passing Potential (HotPP), an E(n) equivariant message passing neural network that extends the node embedding and message to an arbitrary order tensor. By performing some basic equivariant operations, high order tensors can be coupled very simply and thus the model can make direct predictions of high-order tensors such as dipole moments and polarizabilities without any modifications. Compared to high order tensor models based on spherical vectors, this network is simpler and can achieve comparable accuracy with much fewer parameters. The tests in several datasets demonstrate HotPP is a promising new approach that warrants further investigation.