Reducing Symmetry Increase in Equivariant Neural Networks
Ning Lin, Jiacheng Cen, Anyi Li, Wenbing Huang, Hao Sun
Abstract
Equivariant Neural Networks (ENNs) have empowered numerous applications in scientific fields. Despite their remarkable capacity for representing geometric structures, ENNs suffer from degraded expressivity when processing symmetric inputs: the output representations are invariant to transformations that extend beyond the input's symmetries. The mathematical essence of this phenomenon is that a symmetric input, after being processed by an equivariant map, experiences an increase in symmetry. While prior research has documented symmetry increase in specific cases, a rigorous understanding of its underlying causes and general reduction strategies remains lacking. In this paper, we provide a detailed and in-depth characterization of symmetry increase together with a principled framework for its reduction: (i) For any given feature space and input symmetry group, we prove that the increased symmetry admits an infimum determined by the structure of the feature space; (ii) Building on this foundation, we develop a computable algorithm to derive this infimum, and propose practical guidelines for feature design to prevent harmful symmetry increases. (iii) Under standard regularity assumptions, we demonstrate that for most equivariant maps, our guidelines effectively reduce symmetry increase. To complement our theoretical findings, we provide visualizations and experiments on both synthetic datasets and the real-world QM9 dataset. The results validate our theoretical predictions.
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.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext d4c4a0f6-d69e-4c40-b51b-17b5e10ba1adCited by top-tier papers1
Ask how each one uses itBuilds on14
- E(n) Equivariant Graph Neural NetworksVictor Garcia Satorras, Emiel Hoogeboom, Max WellingICML 2021 · 1,432 citations
- Frame Averaging for Invariant and Equivariant Network DesignOmri Puny, Matan Atzmon, Edward J. Smith, Ishan Misra et al.ICLR 2022 · 177 citations
- On the Expressive Power of Geometric Graph Neural NetworksChaitanya K. Joshi, Cristian Bodnar, Simon V. Mathis, Taco Cohen et al.ICML 2023 · 125 citations
- Equivariant Spatio-Temporal Attentive Graph Networks to Simulate Physical DynamicsLiming Wu, Zhichao Hou, Jirui Yuan, Yu Rong et al.NeurIPS 2023 · 34 citations
- Are High-Degree Representations Really Unnecessary in Equivariant Graph Neural Networks?Jiacheng Cen, Anyi Li, Ning Lin, Yuxiang Ren et al.NeurIPS 2024 · 31 citations
Related papers
- Equivariant Adaptation of Large Pretrained ModelsArnab Kumar Mondal, Siba Smarak Panigrahi, Oumar Kaba, Sai Mudumba et al.NeurIPS 2023 · 49 citations
- FAENet: Frame Averaging Equivariant GNN for Materials ModelingAlexandre Duval, Victor Schmidt, Alex Hernández-García, Santiago Miret et al.ICML 2023 · 93 citations
- Equivariance with Learned Canonicalization FunctionsSékou-Oumar Kaba, Arnab Kumar Mondal, Yan Zhang, Yoshua Bengio et al.ICML 2023 · 109 citations
- Approximation-Generalization Trade-offs under (Approximate) Group EquivarianceMircea Petrache, Shubhendu TrivediNeurIPS 2023 · 54 citations
- A Canonicalization Perspective on Invariant and Equivariant LearningGeorge Ma, Yifei Wang, Derek Lim, Stefanie Jegelka et al.NeurIPS 2024 · 38 citations
