Abstract:Symmetries (transformations by group actions) are present in many datasets, and leveraging them holds significant promise for improving predictions in machine learning. In this work, we aim to understand when and how deep networks can learn symmetries from data. We focus on a supervised classification paradigm where data symmetries are only partially observed during training: some classes include all transformations of a cyclic group, while others include only a subset. We ask: can deep networks generalize symmetry invariance to the partially sampled classes? In the infinite-width limit, where kernel analogies apply, we derive a neural kernel theory of symmetry learning to address this question. The group-cyclic nature of the dataset allows us to analyze the spectrum of neural kernels in the Fourier domain; here we find a simple characterization of the generalization error as a function of the interaction between class separation (signal) and class-orbit density (noise). We observe that generalization can only be successful when the local structure of the data prevails over its non-local, symmetric, structure, in the kernel space defined by the architecture. This occurs when (1) classes are sufficiently distinct and (2) class orbits are sufficiently dense. Our framework also applies to equivariant architectures (e.g., CNNs), and recovers their success in the special case where the architecture matches the inherent symmetry of the data. Empirically, our theory reproduces the generalization failure of finite-width networks (MLP, CNN, ViT) trained on partially observed versions of rotated-MNIST. We conclude that conventional networks trained with supervision lack a mechanism to learn symmetries that have not been explicitly embedded in their architecture a priori. Our framework could be extended to guide the design of architectures and training procedures able to learn symmetries from data.

SymmetryGAN: Symmetry Discovery with Deep Learning

Generative Adversarial Symmetry Discovery

Latent Space Symmetry Discovery

Robust Symmetry Detection via Riemannian Langevin Dynamics

Using unsupervised learning to detect broken symmetries, with relevance to searches for parity violation in nature. (Previously: "Stressed GANs snag desserts")

Symmetry Discovery Beyond Affine Transformations

A Generative Model of Symmetry Transformations

Learning Infinitesimal Generators of Continuous Symmetries from Data

Symmetry Discovery for Different Data Types

SYM3D: Learning Symmetric Triplanes for Better 3D-Awareness of GANs

Deep Learning Symmetries and Their Lie Groups, Algebras, and Subalgebras from First Principles

Accelerated Discovery of Machine-Learned Symmetries: Deriving the Exceptional Lie Groups G2, F4 and E6

Symmetry Strikes Back: From Single-Image Symmetry Detection to 3D Generation

On the Ability of Deep Networks to Learn Symmetries from Data: A Neural Kernel Theory

Identifying the Group-Theoretic Structure of Machine-Learned Symmetries

A Unified Framework to Enforce, Discover, and Promote Symmetry in Machine Learning

SymReg-GAN: Symmetric Image Registration with Generative Adversarial Networks

Beyond Planar Symmetry: Modeling Human Perception of Reflection and Rotation Symmetries in the Wild

Learning New Physics from Data -- a Symmetrized Approach

SymmetryNet

Learning Lie Group Symmetry Transformations with Neural Networks