Abstract:Many learning tasks, including learning potential energy surfaces from ab initio calculations, involve global spatial symmetries and permutational symmetry between atoms or general particles. Equivariant graph neural networks are a standard approach to such problems, with one of the most successful methods employing tensor products between various tensors that transform under the spatial group. However, as the number of different tensors and the complexity of relationships between them increase, maintaining parsimony and equivariance becomes increasingly challenging. In this paper, we propose using fusion diagrams, a technique widely employed in simulating SU($2$)-symmetric quantum many-body problems, to design new equivariant components for equivariant neural networks. This results in a diagrammatic approach to constructing novel neural network architectures. When applied to particles within a given local neighborhood, the resulting components, which we term "fusion blocks," serve as universal approximators of any continuous equivariant function defined in the neighborhood. We incorporate a fusion block into pre-existing equivariant architectures (Cormorant and MACE), leading to improved performance with fewer parameters on a range of challenging chemical problems. Furthermore, we apply group-equivariant neural networks to study non-adiabatic molecular dynamics of stilbene cis-trans isomerization. Our approach, which combines tensor networks with equivariant neural networks, suggests a potentially fruitful direction for designing more expressive equivariant neural networks.

Equivariant neural networks for robust observables

Equivariant neural networks for robust $\textit{CP}$ observables

Theory for Equivariant Quantum Neural Networks

Optimal Equivariant Architectures from the Symmetries of Matrix-Element Likelihoods

Equivariance Is Not All You Need: Characterizing the Utility of Equivariant Graph Neural Networks for Particle Physics Tasks

Equivariant Graph Neural Networks for Charged Particle Tracking

Equivariant Neural Networks for Spin Dynamics Simulations of Itinerant Magnets

Equivariant, Safe and Sensitive -- Graph Networks for New Physics

Equivariant neural networks and piecewise linear representation theory

Unifying O(3) Equivariant Neural Networks Design with Tensor-Network Formalism

Rotation-equivariant graph neural network for learning hadronic SMEFT effects

A Characterization Theorem for Equivariant Networks with Point-wise Activations

Using and Abusing Equivariance

Equivariant Neural Simulators for Stochastic Spatiotemporal Dynamics

What is an equivariant neural network?

Lie-Equivariant Quantum Graph Neural Networks

Deconstructing equivariant representations in molecular systems

Resource-efficient equivariant quantum convolutional neural networks

A rotation-equivariant graph neural network for learning hadronic SMEFT effects

Theoretical guarantees for permutation-equivariant quantum neural networks

In Search of Projectively Equivariant Networks