Asymptotically Fast Clebsch-Gordan Tensor Products with Vector Spherical Harmonics
YuQing Xie, Ameya Daigavane, Mit Kotak, Tess Smidt
摘要
-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 to , close to the lower bound of . 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.
问问这篇 Paper
智能体会读完全文。
Lune 把这篇 Paper 索引到了每一个公式,引用它的顶会 Paper 也一样。你提问,回答直接引用原文。
它引用的顶会 Paper11
- Equivariant Diffusion for Molecule Generation in 3DEmiel Hoogeboom, Victor Garcia Satorras, Clément Vignac, Max WellingICML 2022 · 被引用 865 次
- Reducing SO(3) Convolutions to SO(2) for Efficient Equivariant GNNsSaro Passaro, C. Lawrence ZitnickICML 2023 · 被引用 157 次
- Spin-Weighted Spherical CNNsCarlos Esteves, Ameesh Makadia, Kostas DaniilidisNeurIPS 2020 · 被引用 81 次
- Equiformer: Equivariant Graph Attention Transformer for 3D Atomistic GraphsYi-Lun Liao, Tess E. SmidtICLR 2023 · 被引用 65 次
- Enabling Efficient Equivariant Operations in the Fourier Basis via Gaunt Tensor ProductsShengjie Luo, Tianlang Chen, Aditi S. KrishnapriyanICLR 2024 · 被引用 42 次
相关 Paper
- The Price of Freedom: Exploring Expressivity and Runtime Tradeoffs in Equivariant Tensor ProductsYuqing Xie, Ameya Daigavane, Mit Kotak, Tess E. SmidtICML 2025
- Lorentz Group Equivariant Neural Network for Particle PhysicsAlexander Bogatskiy, Brandon M. Anderson, Jan T. Offermann, Marwah Roussi 等ICML 2020 · 被引用 164 次
- Efficient Equivariant High-Order Crystal Tensor Prediction via Cartesian Local-Environment Many-Body CouplingDian Jin, Yancheng Yuan, Xiaoming TaoICML 2026 · 被引用 1 次
- Tensor Decomposition Networks for Fast Machine Learning Interatomic Potential ComputationsYuchao Lin, Cong Fu, Zachary Krueger, Haiyang Yu 等NeurIPS 2025 · 被引用 1 次
- GotenNet: Rethinking Efficient 3D Equivariant Graph Neural NetworksSarp Aykent, Tian XiaICLR 2025
