LieAugmenter: Equivariant Learning by Discovering Symmetries with Learnable Augmentations

Eduardo Santos-Escriche, Ya-Wei Eileen Lin, Stefanie Jegelka

Data augmentation is a powerful mechanism in equivariant machine learning, encouraging symmetry by training networks to produce consistent outputs under transformed inputs. Yet, ef…