Meta-learning Symmetries by Reparameterization

Allan Zhou, Tom Knowles, Chelsea Finn

Keywords: meta-learning, equivariance, convolution, symmetry

Abstract: Many successful deep learning architectures are equivariant to certain transformations in order to conserve parameters and improve generalization: most famously, convolution layers are equivariant to shifts of the input. This approach only works when practitioners know the symmetries of the task and can manually construct an architecture with the corresponding equivariances. Our goal is an approach for learning equivariances from data, without needing to design custom task-specific architectures. We present a method for learning and encoding equivariances into networks by learning corresponding parameter sharing patterns from data. Our method can provably represent equivariance-inducing parameter sharing for any finite group of symmetry transformations. Our experiments suggest that it can automatically learn to encode equivariances to common transformations used in image processing tasks.

Code Of Ethics: I acknowledge that I and all co-authors of this work have read and commit to adhering to the ICLR Code of Ethics

One-sentence Summary: A method for automatically meta-learning and encoding equivariances into neural networks.

Code: [ AllanYangZhou/metalearning-symmetries](https://github.com/AllanYangZhou/metalearning-symmetries) + [ 1 community implementation](https://paperswithcode.com/paper/?openreview=-QxT4mJdijq)

Data: [mini-Imagenet](https://paperswithcode.com/dataset/mini-imagenet)

Community Implementations: [ 2 code implementations](https://www.catalyzex.com/paper/arxiv:2007.02933/code)