arXiv · 2505.13289
RECON: Robust symmetry discovery via Explicit Canonical Orientation Normalization
Abstract
Real world data often exhibits unknown, instance-specific symmetries that rarely exactly match a transformation group $G$ fixed a priori. Class-pose decompositions aim to create disentangled representations by factoring inputs into invariant features and a pose $g\in G$ defined relative to a training-dependent, arbitrary canonical representation. We introduce RECON, a class-pose agnostic canonical orientation normalization that corrects arbitrary canonicals via a simple right translation, yielding natural, data-aligned canonicalizations. This enables (i) unsupervised discovery of instance-specific pose distributions, (ii) detection of out-of-distribution poses and (iii) a plug-and-play test-time canonicalization layer. This layer can be attached on top of any pre-trained model to infuse group invariance, improving its performance without retraining. We validate on images and molecular ensembles, demonstrating accurate symmetry discovery, and matching or outperforming other canonicalizations in downstream classification.
Explore related subjects
Keep this discovery
Explore connections, maps & timelines
Alonso Urbano, David W. Romero, Max Zimmer, Sebastian Pokutta. 2025-05-19. RECON: Robust symmetry discovery via Explicit Canonical Orientation Normalization. https://arxiv.org/abs/2505.13289
Cite the original work for its findings. Save a collection to share your selection of sources.