Equivariant Frames and the Impossibility of Continuous Canonicalization
Nadav Dym, Hannah Lawrence, Jonathan W. Siegel
Abstract
Canonicalization provides an architecture-agnostic method for enforcing equivariance, with generalizations such as frame-averaging recently gaining prominence as a lightweight and flexible alternative to equivariant architectures. Recent works have found an empirical benefit to using probabilistic frames instead, which learn weighted distributions over group elements. In this work, we provide strong theoretical justification for this phenomenon: for commonly-used groups, there is no efficiently computable choice of frame that preserves continuity of the function being averaged. In other words, unweighted frame-averaging can turn a smooth, non-symmetric function into a discontinuous, symmetric function. To address this fundamental robustness problem, we formally define and construct weighted frames, which provably preserve continuity, and demonstrate their utility by constructing efficient and continuous weighted frames for the actions of , , and on point clouds.
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 f3971a72-9d04-4d0d-a349-491ee1a52ff6Cited by top-tier papers26
- A Canonicalization Perspective on Invariant and Equivariant LearningGeorge Ma, Yifei Wang, Derek Lim, Stefanie Jegelka et al.NeurIPS 2024 · 38 citations
- Equivariance via Minimal Frame Averaging for More Symmetries and EfficiencyYuchao Lin, Jacob Helwig, Shurui Gui, Shuiwang JiICML 2024 · 20 citations
- Lorentz Local Canonicalization: How to make any Network Lorentz-EquivariantJonas Spinner, Luigi Favaro, Peter Lippmann, Sebastian Pitz et al.NeurIPS 2025 · 19 citations
- Weisfeiler Leman for Euclidean Equivariant Machine LearningSnir Hordan, Tal Amir, Nadav DymICML 2024 · 11 citations
- On Transferring Transferability: Towards a Theory for Size GeneralizationEitan Levin, Yuxin Ma, Mateo Díaz, Soledad VillarNeurIPS 2025 · 10 citations
Builds on20
- E(n) Equivariant Graph Neural NetworksVictor Garcia Satorras, Emiel Hoogeboom, Max WellingICML 2021 · 1,432 citations
- DiffDock: Diffusion Steps, Twists, and Turns for Molecular DockingGabriele Corso, Hannes Stärk, Bowen Jing, Regina Barzilay et al.ICLR 2023 · 331 citations
- EquiformerV2: Improved Equivariant Transformer for Scaling to Higher-Degree RepresentationsYi-Lun Liao, Brandon M. Wood, Abhishek Das, Tess E. SmidtICLR 2024 · 311 citations
- Frame Averaging for Invariant and Equivariant Network DesignOmri Puny, Matan Atzmon, Edward J. Smith, Ishan Misra et al.ICLR 2022 · 177 citations
- Understanding and Extending Subgraph GNNs by Rethinking Their SymmetriesFabrizio Frasca, Beatrice Bevilacqua, Michael M. Bronstein, Haggai MaronNeurIPS 2022 · 168 citations
Related papers
- Equivariance with Learned Canonicalization FunctionsSékou-Oumar Kaba, Arnab Kumar Mondal, Yan Zhang, Yoshua Bengio et al.ICML 2023 · 109 citations
- Beyond Canonicalization: How Tensorial Messages Improve Equivariant Message PassingPeter Lippmann, Gerrit Gerhartz, Roman Remme, Fred A. HamprechtICLR 2025
- Adaptive Canonicalization with Application to Invariant Anisotropic Geometric NetworksYa-Wei Eileen Lin, Ron LevieICLR 2026 · 4 citations
- Equivariant Adaptation of Large Pretrained ModelsArnab Kumar Mondal, Siba Smarak Panigrahi, Oumar Kaba, Sai Mudumba et al.NeurIPS 2023 · 49 citations
- Generalization Bounds for Canonicalization: A Comparative Study with Group AveragingBehrooz Tahmasebi, Stefanie JegelkaICLR 2025
