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.

Synergy and Symmetry in Deep Learning: Interactions between the Data, Model, and Inference Algorithm

Symmetry Induces Structure and Constraint of Learning

Learning in the Machine: the Symmetries of the Deep Learning Channel

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

Exploring and Exploiting the Asymmetric Valley of Deep Neural Networks

Black Boxes and Looking Glasses: Multilevel Symmetries, Reflection Planes, and Convex Optimization in Deep Networks

Symmetry Breaking in Neural Network Optimization: Insights from Input Dimension Expansion

Complexity from Adaptive-Symmetries Breaking: Global Minima in the Statistical Mechanics of Deep Neural Networks

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

Symmetries in Overparametrized Neural Networks: A Mean-Field View

Low-dimensional Intrinsic Dimension Reveals a Phase Transition in Gradient-Based Learning of Deep Neural Networks

Symphony in the Latent Space: Provably Integrating High-dimensional Techniques with Non-linear Machine Learning Models

Understanding Dynamics of Nonlinear Representation Learning and Its Application

On the hardness of learning under symmetries

The Unreasonable Effectiveness of Deep Learning in Artificial Intelligence

Git Re-Basin: Merging Models modulo Permutation Symmetries

How DNNs break the Curse of Dimensionality: Compositionality and Symmetry Learning

Symmetry Perception by Deep Networks: Inadequacy of Feed-Forward Architectures and Improvements with Recurrent Connections

SoK: On Finding Common Ground in Loss Landscapes Using Deep Model Merging Techniques

Double Descent Demystified: Identifying, Interpreting & Ablating the Sources of a Deep Learning Puzzle