11
253

Equivariance Through Parameter-Sharing

Abstract

We propose to study equivariance in deep neural networks through parameter symmetries. In particular, given a group G\mathcal{G} that acts discretely on the input and output of a standard neural network layer ϕW:MN\phi_{W}: \Re^{M} \to \Re^{N}, we show that ϕW\phi_{W} is equivariant with respect to G\mathcal{G}-action iff G\mathcal{G} explains the symmetries of the network parameters WW. Inspired by this observation, we then propose two parameter-sharing schemes to induce the desirable symmetry on WW. Our procedures for tying the parameters achieve G\mathcal{G}-equivariance and, under some conditions on the action of G\mathcal{G}, they guarantee sensitivity to all other permutation groups outside G\mathcal{G}.

View on arXiv
Comments on this paper