Continuous Symmetry Discovery and Enforcement Using Infinitesimal Generators of Multi-parameter Group Actions
Ben Shaw, Sasidhar Kunapuli, Abram Magner, Kevin R. Moon
TL;DR
This work addresses discovering continuous symmetries beyond affine transformations by estimating infinitesimal generators of multi-parameter group actions and automatically inferring the number of independent generators. It introduces isometry-focused restrictions via Killing vectors to constrain the search space and develops symmetry-enforcement methods based on invariant-feature construction and an explicit regularization term, with a robustness analysis. The approach is demonstrated on neural networks, non-affine symmetries, and infinitesimal isometries, showing improved generalization and reduced need for data augmentation. The results suggest a practical path toward leveraging symmetry in complex ML tasks with non-affine and metric-aware transformations.
Abstract
Symmetry-informed machine learning can exhibit advantages over machine learning which fails to account for symmetry. In the context of continuous symmetry detection, current state of the art experiments are largely limited to detecting affine transformations. Herein, we outline a computationally efficient framework for discovering infinitesimal generators of multi-parameter group actions which are not generally affine transformations. This framework accommodates the automatic discovery of the number of linearly independent infinitesimal generators. We build upon recent work in continuous symmetry discovery by extending to neural networks and by restricting the symmetry search space to infinitesimal isometries. We also introduce symmetry enforcement of smooth models using vector field regularization, thereby improving model generalization. The notion of vector field similarity is also generalized for non-Euclidean Riemannian metric tensors.
