English

Fast contracted Clebsch--Gordan tensor products for equivariant graph neural networks

Computational Physics 2026-05-15 v1 Materials Science Chemical Physics

Abstract

We present an O(L3)\mathcal{O}(L^3) algorithm for evaluating contracted Clebsch--Gordan tensor products in O(3)\mathrm{O}(3)-equivariant machine learning potentials at fixed Canonical Polyadic (CP) rank. Mapping the angular integral to a structured Gauss--Legendre and Fourier tensor-product grid decouples the radial channel contractions from the angular transforms. The antisymmetric parity-odd Clebsch--Gordan channels, unreachable by the symmetric pointwise product on a scalar S2S^2 grid, are recovered through the surface-curl pairing r^[S2A×S2B]\hat r \cdot [\nabla_{S^2} A \times \nabla_{S^2} B], the spherical Poisson bracket, which supplies the L=1L=1 angular momentum on the grid while preserving rotational equivariance. The construction extends to parity-aware equivariant message passing in atomic-cluster-expansion-style architectures and is verified by direct numerical quadrature. The full uncontracted Clebsch--Gordan tensor product remains subject to the O(L4)\mathcal{O}(L^4) output-size lower bound. A benchmark shows wall-clock scaling empirically as L2L^2 across the practical lmaxl_{\max} range. For the on-site contraction this is pre-asymptotic, giving way to L3L^3 at large lmaxl_{\max}. For message passing it is structural and the runtime is memory-bandwidth bound on L2L^2-sized grid tensors.

Keywords

Cite

@article{arxiv.2605.15073,
  title  = {Fast contracted Clebsch--Gordan tensor products for equivariant graph neural networks},
  author = {Anton Bochkarev and Yury Lysogorskiy and Ralf Drautz},
  journal= {arXiv preprint arXiv:2605.15073},
  year   = {2026}
}