NeurIPS2024

The Selective G-Bispectrum and its Inversion: Applications to G-Invariant Networks

Simon Mataigne, Johan Mathe, Sophia Sanborn, Christopher Hillar, Nina Miolane

摘要

An important problem in signal processing and deep learning is to achieve invariance to nuisance factors not relevant for the task. Since many of these factors are describable as the action of a group GG (e.g. rotations, translations, scalings), we want methods to be GG-invariant. The GG-Bispectrum extracts every characteristic of a given signal up to group action: for example, the shape of an object in an image, but not its orientation. Consequently, the GG-Bispectrum has been incorporated into deep neural network architectures as a computational primitive for GG-invarianceakin to a pooling mechanism, but with greater selectivity and robustness. However, the computational cost of the GG-Bispectrum (O(G2)\mathcal{O}(|G|^2), with G|G| the size of the group) has limited its widespread adoption. Here, we show that the GG-Bispectrum computation contains redundancies that can be reduced into a selective GG-Bispectrum with O(G)\mathcal{O}(|G|) complexity. We prove desirable mathematical properties of the selective GG-Bispectrum and demonstrate how its integration in neural networks enhances accuracy and robustness compared to traditional approaches, while enjoying considerable speeds-up compared to the full GG-Bispectrum.