RECON: Robust symmetry discovery via Explicit Canonical Orientation Normalization

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Urbano, Alonso, Romero, David W., Zimmer, Max, Pokutta, Sebastian
Formato: Preprint
Publicado: 2025
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866915990117285888
author Urbano, Alonso
Romero, David W.
Zimmer, Max
Pokutta, Sebastian
author_facet Urbano, Alonso
Romero, David W.
Zimmer, Max
Pokutta, Sebastian
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2505_13289
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle RECON: Robust symmetry discovery via Explicit Canonical Orientation Normalization
Urbano, Alonso
Romero, David W.
Zimmer, Max
Pokutta, Sebastian
Machine Learning
Computer Vision and Pattern Recognition
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.
title RECON: Robust symmetry discovery via Explicit Canonical Orientation Normalization
topic Machine Learning
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2505.13289