arXiv · 2410.01452
Ensembles provably learn equivariance through data augmentation
Abstract
Recently, it was proved that group equivariance emerges in ensembles of neural networks as the result of full augmentation in the limit of infinitely wide neural networks (neural tangent kernel limit). In this paper, we extend this result significantly. We provide a proof that this emergence does not depend on the neural tangent kernel limit at all. We also consider stochastic settings, and furthermore general architectures. For the latter, we provide a simple sufficient condition on the relation between the architecture and the action of the group for our results to hold. We validate our findings through simple numeric experiments.
Explore related subjects
Keep this discovery
Oskar Nordenfors, Axel Flinth. 2024-10-02. Ensembles provably learn equivariance through data augmentation. https://arxiv.org/abs/2410.01452
Cite the original work for its findings. Save a collection to share your selection of sources.