On Non-Linear operators for Geometric Deep Learning
This provides theoretical justification for using point-wise non-linearities in geometric deep learning, but it is incremental as it extends known results from Euclidean spaces to manifolds.
This work tackles the problem of identifying non-linear operators for geometric deep learning that commute with diffeomorphisms on manifolds, proving that for scalar fields, only point-wise non-linearities are universal, while for vector fields, only scalar multiplication exists, indicating no universal class for neural network design.
This work studies operators mapping vector and scalar fields defined over a manifold $\mathcal{M}$, and which commute with its group of diffeomorphisms $\text{Diff}(\mathcal{M})$. We prove that in the case of scalar fields $L^p_ω(\mathcal{M,\mathbb{R}})$, those operators correspond to point-wise non-linearities, recovering and extending known results on $\mathbb{R}^d$. In the context of Neural Networks defined over $\mathcal{M}$, it indicates that point-wise non-linear operators are the only universal family that commutes with any group of symmetries, and justifies their systematic use in combination with dedicated linear operators commuting with specific symmetries. In the case of vector fields $L^p_ω(\mathcal{M},T\mathcal{M})$, we show that those operators are solely the scalar multiplication. It indicates that $\text{Diff}(\mathcal{M})$ is too rich and that there is no universal class of non-linear operators to motivate the design of Neural Networks over the symmetries of $\mathcal{M}$.