Universal Collection of Euclidean Invariants Between Pairs of Position-Orientations
摘要
Euclidean \({{\,\textrm{E}\,}}(3) := \mathbb {R}^3 \rtimes \text {O}(3)\) equivariant neural networks that employ scalar fields on position-orientation space \(\mathbb {M}_3 := \mathbb {R}^3 \times S^2\) have been effectively applied to tasks such as predicting molecular dynamics and properties. To perform equivariant convolutional-like operations in these architectures one needs Euclidean invariant kernels \(k : \mathbb {M}_3 \times \mathbb {M}_3 \rightarrow \mathbb {R}\) . In practice, a handcrafted collection of invariants is selected, and this collection is then fed into multilayer perceptrons to parametrize the kernels. We rigorously describe an optimal collection of \(4\) smooth scalar invariants on the whole of \(\mathbb {M}_3 \times \mathbb {M}_3\) . With optimal we mean that the collection is independent and universal, meaning that all invariants are pertinent, and any invariant kernel is a function of them. We evaluate two collections of invariants, one universal and one not, using the PONITA neural network architecture. Our experiments show that using a collection of invariants that is universal positively impacts the accuracy of PONITA significantly.