LieAugmenter: Equivariant Learning by Discovering Symmetries with Learnable Augmentations
arXiv, 2025
Data augmentation for geometric domains typically requires underlying symmetries to be specified a priori, which can limit generalization in cases where the symmetries are unknown or approximate. To address this, we introduce LieAugmenter, an end-to-end framework that discovers task-relevant continuous symmetries through learnable augmentations parameterized by Lie groups.