arXiv · 2410.03989
Symmetry From Scratch: Group Equivariance as a Supervised Learning Task
Abstract
In machine learning datasets with symmetries, the paradigm for backward compatibility with symmetry-breaking has been to relax equivariant architectural constraints, engineering extra weights to differentiate symmetries of interest. However, this process becomes increasingly over-engineered as models are geared towards specific symmetries/asymmetries hardwired of a particular set of equivariant basis functions. In this work, we introduce symmetry-cloning, a method for inducing equivariance in machine learning models. We show that general machine learning architectures (i.e., MLPs) can learn symmetries directly as a supervised learning task from group equivariant architectures and retain/break the learned symmetry for downstream tasks. This simple formulation enables machine learning models with group-agnostic architectures to capture the inductive bias of group-equivariant architectures.
Explore related subjects
Keep this discovery
Explore connections, maps & timelines
Haozhe Huang, Leo Kaixuan Cheng, Kaiwen Chen, Alán Aspuru-Guzik. 2024-10-05. Symmetry From Scratch: Group Equivariance as a Supervised Learning Task. https://arxiv.org/abs/2410.03989
Cite the original work for its findings. Save a collection to share your selection of sources.