A General Theory of Equivariant CNNs on Homogeneous Spaces
Machine Learning
2020-01-10 v2 Artificial Intelligence
Computational Geometry
Computer Vision and Pattern Recognition
Machine Learning
Abstract
We present a general theory of Group equivariant Convolutional Neural Networks (G-CNNs) on homogeneous spaces such as Euclidean space and the sphere. Feature maps in these networks represent fields on a homogeneous base space, and layers are equivariant maps between spaces of fields. The theory enables a systematic classification of all existing G-CNNs in terms of their symmetry group, base space, and field type. We also consider a fundamental question: what is the most general kind of equivariant linear map between feature spaces (fields) of given types? Following Mackey, we show that such maps correspond one-to-one with convolutions using equivariant kernels, and characterize the space of such kernels.
Keywords
Cite
@article{arxiv.1811.02017,
title = {A General Theory of Equivariant CNNs on Homogeneous Spaces},
author = {Taco Cohen and Mario Geiger and Maurice Weiler},
journal= {arXiv preprint arXiv:1811.02017},
year = {2020}
}