Lune

ICML2026Top-tier venue

Asymptotically Fast Clebsch-Gordan Tensor Products with Vector Spherical Harmonics

YuQing Xie, Ameya Daigavane, Mit Kotak, Tess Smidt

2026Year
6Citations

Abstract

E(3)E(3)-equivariant neural networks have proven to be extremely effective in a wide range of 3D modeling tasks. A fundamental operation of such networks is the tensor product, which allows interaction between different feature types. Because this operation scales poorly, there has been considerable work towards accelerating this interaction. However, recently Xie et al. 2025 have pointed out that most speedups come from a reduction in expressivity rather than true algorithmic improvements on computing Clebsch-Gordan tensor products. A modification of Gaunt tensor product (Luo et al.) can give a true asymptotic speedup but is incomplete and misses many interactions. In this work, we provide the first complete algorithm which truly provides asymptotic benefits Clebsch-Gordan tensor products. For full CGTP, our algorithm brings runtime complexity from the naive O(L6)O(L^6) to O(L4log⁡2L)O(L^4\log^2 L), close to the lower bound of O(L4)O(L^4). We first show how generalizing fast Fourier based convolution naturally leads to the previously proposed Gaunt tensor product (Luo et al.). To remedy antisymmetry issues, we generalize from scalar signals to irrep valued signals, giving us tensor spherical harmonics. We prove a generalized Gaunt formula for the tensor harmonics. Finally, we show that we only need up to vector valued signals to recover the missing interactions of Gaunt tensor product.

Ask about this paper

Your agent reads all of it.

Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.

Questions to start from

Your agent calls

Luneget_paper_fulltext

Ask in Lune

Free to start. No credit card required.

lune papers fulltext 85007cd3-6424-49a2-ab03-6473c531b615

Builds on11

Related papers

Dusk over the sea between two cliffs drawn in fine vertical lines