REVIEW 4 cited by
Reducing SO(3) Convolutions to SO(2) for Efficient Equivariant GNNs
Not yet reviewed by Pith; the record is open.
This paper has not been read by Pith yet. Machine review is queued; the pith claim, tier, and objections will appear here once it completes.
SPECIMEN: schema-true, not a live event
T0 review · schema-true
One-sentence machine reading of the paper's core claim.
pith:XXXXXXXX · record.json · timestamp
abstract
Graph neural networks that model 3D data, such as point clouds or atoms, are typically desired to be $SO(3)$ equivariant, i.e., equivariant to 3D rotations. Unfortunately equivariant convolutions, which are a fundamental operation for equivariant networks, increase significantly in computational complexity as higher-order tensors are used. In this paper, we address this issue by reducing the $SO(3)$ convolutions or tensor products to mathematically equivalent convolutions in $SO(2)$ . This is accomplished by aligning the node embeddings' primary axis with the edge vectors, which sparsifies the tensor product and reduces the computational complexity from $O(L^6)$ to $O(L^3)$, where $L$ is the degree of the representation. We demonstrate the potential implications of this improvement by proposing the Equivariant Spherical Channel Network (eSCN), a graph neural network utilizing our novel approach to equivariant convolutions, which achieves state-of-the-art results on the large-scale OC-20 and OC-22 datasets.
Forward citations
Cited by 4 Pith papers
-
Fast contracted Clebsch--Gordan tensor products for equivariant graph neural networks
An O(L^3) algorithm computes contracted Clebsch-Gordan tensor products for equivariant ML potentials using a structured angular grid and spherical Poisson bracket to handle parity-odd terms at fixed CP rank.
-
Velocityformer: Broken-Symmetry-Matched Equivariant Graph Transformers for Cosmological Velocity Reconstruction
Velocityformer achieves 35% higher velocity correlation than linear theory by matching graph transformer inductive bias to the line-of-sight broken symmetry and conditioning on long-wavelength physics, while training ...
-
Interpretable Nanoporous Materials Design with Symmetry-Aware Networks
An equivariant transformer with periodic space sampling predicts nanoporous properties and attributes each prediction to local structural sites.
-
Hot-Ham: an accurate and efficient E(3)-equivariant machine-learning electronic structures calculation framework
Hot-Ham combines Gaunt tensor products with a local-coordinate SO(2) convolution to predict DFT Hamiltonians accurately and efficiently across several material classes.
Discussion (0). Sign in to comment.