Computer Science > Machine Learning
arXiv:2512.13927 (cs)
Abstract:Spherical equivariant graph neural networks (EGNNs) provide a principled framework for learning on three-dimensional molecular and biomolecular systems, where predictions must respect the rotational symmetries inherent in physics. These models extend traditional message-passing GNNs and Transformers by representing node and edge features as spherical tensors that transform under irreducible representations of the rotation group SO(3), ensuring that predictions change in physically meaningful ways under rotations of the input. This guide develops a complete, intuitive foundation for spherical equivariant modeling - from group representations and spherical har…
Computer Science > Machine Learning
arXiv:2512.13927 (cs)
Abstract:Spherical equivariant graph neural networks (EGNNs) provide a principled framework for learning on three-dimensional molecular and biomolecular systems, where predictions must respect the rotational symmetries inherent in physics. These models extend traditional message-passing GNNs and Transformers by representing node and edge features as spherical tensors that transform under irreducible representations of the rotation group SO(3), ensuring that predictions change in physically meaningful ways under rotations of the input. This guide develops a complete, intuitive foundation for spherical equivariant modeling - from group representations and spherical harmonics, to tensor products, Clebsch-Gordan decomposition, and the construction of SO(3)-equivariant kernels. Building on this foundation, we construct the Tensor Field Network and SE(3)-Transformer architectures and explain how they perform equivariant message-passing and attention on geometric graphs. Through clear mathematical derivations and annotated code excerpts, this guide serves as a self-contained introduction for researchers and learners seeking to understand or implement spherical EGNNs for applications in chemistry, molecular property prediction, protein structure modeling, and generative modeling.
| Comments: | This paper is a technical version of the article originally published in Alchemy Bio (99 pages, 46 figures) |
| Subjects: | Machine Learning (cs.LG); Quantitative Methods (q-bio.QM) |
| Cite as: | arXiv:2512.13927 [cs.LG] |
| (or arXiv:2512.13927v1 [cs.LG] for this version) | |
| https://doi.org/10.48550/arXiv.2512.13927 arXiv-issued DOI via DataCite (pending registration) |
Submission history
From: Sophia Tang [view email] [v1] Mon, 15 Dec 2025 22:03:09 UTC (23,399 KB)
Current browse context:
cs.LG
Change to browse by:
export BibTeX citation