We propose a computationally efficient -invariant neural network that approximates functions invariant to the action of a given permutation subgroup of the symmetric group on input data. The key element of the proposed network architecture is a new -invariant transformation module, which produces a -invariant latent representation of the input data. Theoretical considerations are supported by numerical experiments, which demonstrate the effectiveness and strong generalization properties of the proposed method in comparison to other -invariant neural networks.
View on arXiv