Learnable Equivariant Representations of Stochastic Heat Bath Models via the Atomic Cluster Expansion (ACE)


Matthias Sachs

School of Mathematics, University of Birmingham, United Kingdom

Rigorously derived dynamics of coarse-grained particle systems via the Mori-Zwanzig projection formalism take the form of a (generalized) Langevin equation with, in general, configuration-dependent friction and diffusion tensors. In this talk, I will introduce a class of equivariant representations of tensor-valued functions based on the Atomic Cluster Expansion (ACE) framework that allows for efficient learning of such configuration-dependent friction and diffusion tensors from data. Besides satisfying the correct equivariance properties with respect to the Euclidean group E(3), the resulting heat bath models satisfy a fluctuation-dissipation relation. Moreover, such models can be extended to include additional symmetries, such as momentum conservation, to preserve the hydrodynamic properties of the particle system.