Equivariance Through Parameter-Sharing
Ravanbakhsh, Siamak, Schneider, Jeff, Poczos, Barnabas
We propose to study equivariance in deep neural networks through parameter symmetries. In particular, given a group $\mathcal{G}$ that acts discretely on the input and output of a standard neural network layer $\phi_{W}: \Re^{M} \to \Re^{N}$, we show that $\phi_{W}$ is equivariant with respect to $\mathcal{G}$-action iff $\mathcal{G}$ explains the symmetries of the network parameters $W$. Inspired by this observation, we then propose two parameter-sharing schemes to induce the desirable symmetry on $W$. Our procedures for tying the parameters achieve $\mathcal{G}$-equivariance and, under some conditions on the action of $\mathcal{G}$, they guarantee sensitivity to all other permutation groups outside $\mathcal{G}$.
Jun-13-2017
- Country:
- North America > United States
- Illinois > Cook County
- Chicago (0.04)
- Ohio (0.04)
- Pennsylvania > Allegheny County
- Pittsburgh (0.14)
- Illinois > Cook County
- Oceania > Australia
- New South Wales > Sydney (0.04)
- North America > United States
- Genre:
- Research Report (0.50)
- Technology: