Equivariant Frames and the Impossibility of Continuous Canonicalization
Nadav Dym, Hannah Lawrence, Jonathan W. Siegel
TL;DR
The paper proves fundamental limits on continuous canonicalization for key group actions ($S_n$, $SO(d)$, and $O(d)$) and shows that unweighted frame-averaging often induces discontinuities. To restore robustness, it introduces weighted, robust frames that are continuous and (weakly) equivariant, and constructs practical instances for permutations and rotations, including explicit cardinality bounds. These robust frames yield invariant (and, with stable backbones, equivariant) projection operators that preserve continuity, enabling universal, continuous models for group actions on point clouds. The work connects with probabilistic/weighted-frame approaches in the literature and demonstrates, both theoretically and experimentally, that weighted frames can outperform discontinuous canonicalization while remaining computationally feasible. This provides a principled, scalable path to robust, continuous equivariant learning across common geometric groups.
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 \emph{weighted} frames, which provably preserve continuity, and demonstrate their utility by constructing efficient and continuous weighted frames for the actions of $SO(2)$, $SO(3)$, and $S_n$ on point clouds.
