arXiv · 2606.10913
Conservation Laws from Data Symmetry in Neural Networks
Abstract
We explore whether intrinsic symmetries of the training data lead to conserved quantities during gradient-flow training of neural networks. Under the assumption that the loss function is analytic and non-polynomial, we prove that data symmetries generically do not induce any additional integrals of motion. For mean squared error (MSE) loss, on the other hand, there are situations in which data augmentation yields extra conserved quantities. We build a framework, utilizing \emph{tensorizable networks} to describe this phenomenon. Tensorizable networks are a family of architectures whose dependence on parameters and inputs can be separated using an intermediate representation. They include linear and polynomial networks, as well as Lightning Attention.
Explore related subjects
Keep this discovery
Jakob Galley, Vahid Shahverdi, Axel Flinth. 2026-06-09. Conservation Laws from Data Symmetry in Neural Networks. https://arxiv.org/abs/2606.10913
Cite the original work for its findings. Save a collection to share your selection of sources.